diff options
Diffstat (limited to 'compiler/exo/src/TFLite/TFLExporterUtils.cpp')
| -rw-r--r-- | compiler/exo/src/TFLite/TFLExporterUtils.cpp | 13 |
1 files changed, 6 insertions, 7 deletions
diff --git a/compiler/exo/src/TFLite/TFLExporterUtils.cpp b/compiler/exo/src/TFLite/TFLExporterUtils.cpp index d35afc9aa..daec03c40 100644 --- a/compiler/exo/src/TFLite/TFLExporterUtils.cpp +++ b/compiler/exo/src/TFLite/TFLExporterUtils.cpp @@ -78,13 +78,13 @@ tflite::Padding getOpPadding(const loco::Padding2D *pad, const loco::Stride<2> * // // NOTE input and output 'feature' map are shape of NHWC bool same_padding_criterion_1 = - (static_cast<uint32_t>(ofm._dims[1]) == (ifm._dims[1] - 1) / stride->vertical() + 1) && - (static_cast<uint32_t>(ofm._dims[2]) == (ifm._dims[2] - 1) / stride->horizontal() + 1); + (static_cast<uint32_t>(ofm._dims[1]) == (ifm._dims[1] - 1) / stride->vertical() + 1) && + (static_cast<uint32_t>(ofm._dims[2]) == (ifm._dims[2] - 1) / stride->horizontal() + 1); // For same padding, rear padding is same or bigger than front padding by at most 1 bool same_padding_criterion_2 = - (pad->top() <= pad->bottom()) && (pad->bottom() <= pad->top() + 1) && - (pad->left() <= pad->right()) && (pad->right() <= pad->left() + 1); + (pad->top() <= pad->bottom()) && (pad->bottom() <= pad->top() + 1) && + (pad->left() <= pad->right()) && (pad->right() <= pad->left() + 1); if (same_padding_criterion_1 && same_padding_criterion_2) return tflite::Padding_SAME; @@ -120,8 +120,7 @@ void registerGraphIOName(loco::Graph *graph, SerializedModelData &gd) } } -#include <stdex/Memory.h> - +#include <memory> #include <cassert> namespace @@ -147,7 +146,7 @@ private: void set_tensor_index(loco::Node *node, const TFLTensorIndex &tensor_id) { assert(node->annot<TFLTensorIndexAnnotation>() == nullptr); - node->annot(stdex::make_unique<TFLTensorIndexAnnotation>(tensor_id)); + node->annot(std::make_unique<TFLTensorIndexAnnotation>(tensor_id)); } TFLTensorIndex get_tensor_index(loco::Node *node) |