diff options
Diffstat (limited to 'compiler/moco/import/src/Nodes/Mean.cpp')
-rw-r--r-- | compiler/moco/import/src/Nodes/Mean.cpp | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/compiler/moco/import/src/Nodes/Mean.cpp b/compiler/moco/import/src/Nodes/Mean.cpp index 30fb0f1f7..3f559bc41 100644 --- a/compiler/moco/import/src/Nodes/Mean.cpp +++ b/compiler/moco/import/src/Nodes/Mean.cpp @@ -19,9 +19,10 @@ #include <moco/IR/Nodes/TFMean.h> #include <loco.h> -#include <stdex/Memory.h> #include <plier/tf/Convert.h> +#include <memory> + namespace { using namespace moco; @@ -34,7 +35,7 @@ class MeanGraphUpdate final : public GraphUpdate public: MeanGraphUpdate(TFMean *node, const TensorName &&input_name, const TensorName &&reduction_indices_name) - : _node(node), _input_name(input_name), _reduction_indices_name(reduction_indices_name) + : _node(node), _input_name(input_name), _reduction_indices_name(reduction_indices_name) { // DO NOTHING } @@ -91,8 +92,8 @@ void MeanGraphBuilder::build(const tensorflow::NodeDef &node, GraphBuilderContex TensorName output_name(node.name(), 0); tensor_names->enroll(output_name, tf_mean); - auto update = stdex::make_unique<MeanGraphUpdate>(tf_mean, TensorName(node.input(0)), - TensorName(node.input(1))); + auto update = std::make_unique<MeanGraphUpdate>(tf_mean, TensorName(node.input(0)), + TensorName(node.input(1))); updates->enroll(std::move(update)); } |