diff options
Diffstat (limited to 'compiler/enco/frontend/tflite/src/GraphBuilderRegistry.h')
| -rw-r--r-- | compiler/enco/frontend/tflite/src/GraphBuilderRegistry.h | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/compiler/enco/frontend/tflite/src/GraphBuilderRegistry.h b/compiler/enco/frontend/tflite/src/GraphBuilderRegistry.h index 1ae882e89..ca4f74fc5 100644 --- a/compiler/enco/frontend/tflite/src/GraphBuilderRegistry.h +++ b/compiler/enco/frontend/tflite/src/GraphBuilderRegistry.h @@ -29,11 +29,11 @@ #include "Op/Div.h" #include <schema_generated.h> -#include <stdex/Memory.h> +#include <memory> #include <map> -using stdex::make_unique; +using std::make_unique; namespace tflimport { @@ -68,7 +68,7 @@ private: // add GraphBuilder for each tflite operation. _builder_map[tflite::BuiltinOperator_CONV_2D] = make_unique<Conv2DGraphBuilder>(); _builder_map[tflite::BuiltinOperator_DEPTHWISE_CONV_2D] = - make_unique<DepthwiseConv2DGraphBuilder>(); + make_unique<DepthwiseConv2DGraphBuilder>(); _builder_map[tflite::BuiltinOperator_AVERAGE_POOL_2D] = make_unique<AvgPool2DGraphBuilder>(); _builder_map[tflite::BuiltinOperator_MAX_POOL_2D] = make_unique<MaxPool2DGraphBuilder>(); _builder_map[tflite::BuiltinOperator_CONCATENATION] = make_unique<ConcatenationGraphBuilder>(); |