From 80e9e632073a6756c9147ed6464e7a254b919dcf Mon Sep 17 00:00:00 2001 From: 13291271729 Date: Tue, 30 Aug 2022 16:04:04 +0800 Subject: [PATCH] clean code --- parser/tensorflow/graph_functiondef.h | 2 +- parser/tensorflow/graph_optimizer.cc | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/parser/tensorflow/graph_functiondef.h b/parser/tensorflow/graph_functiondef.h index 44cd48a..081a022 100644 --- a/parser/tensorflow/graph_functiondef.h +++ b/parser/tensorflow/graph_functiondef.h @@ -28,7 +28,6 @@ #include "proto/tensorflow/graph.pb.h" #include "register/register_error_codes.h" -namespace ge { using domi::tensorflow::AttrValue; using domi::tensorflow::AttrValue_ListValue; using domi::tensorflow::DataType; @@ -40,6 +39,7 @@ using std::string; using std::to_string; using std::vector; +namespace ge { class GraphToFunctionDef { public: static domi::Status RecordArg(ge::ComputeGraphPtr graph, diff --git a/parser/tensorflow/graph_optimizer.cc b/parser/tensorflow/graph_optimizer.cc index 3346364..e9c8799 100644 --- a/parser/tensorflow/graph_optimizer.cc +++ b/parser/tensorflow/graph_optimizer.cc @@ -202,9 +202,9 @@ Status ParserGraphOptimizer::UpdateGraph(vector &nodes) { "insert node to sub_graph failed."); GE_CHK_STATUS_RET(LinkInnerAnchor(node_map), "Link inner anchor failed."); - std::unique_ptr node_def = std::make_unique(); // tensorflow NodeDef + std::unique_ptr node_def(new (std::nothrow) NodeDef()); // tensorflow NodeDef GE_CHECK_NOTNULL(node_def); - std::unique_ptr func_def_lib = std::make_unique(); + std::unique_ptr func_def_lib(new (std::nothrow) FunctionDefLibrary()); GE_CHECK_NOTNULL(func_def_lib); // convert graph to FunctionDef if (nodes.size() == 0) {