From feb46f13e15e45549b3f433c04eb7f8c282fb108 Mon Sep 17 00:00:00 2001 From: l00444296 Date: Mon, 2 Nov 2020 19:27:11 +0800 Subject: [PATCH] Description:aclgrphParse IR interface Team:HISI_SW Feature or Bugfix:Feature --- parser/caffe/caffe_parser.cc | 4 ++-- parser/tensorflow/tensorflow_parser.cc | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/parser/caffe/caffe_parser.cc b/parser/caffe/caffe_parser.cc index 24a3862..223dada 100644 --- a/parser/caffe/caffe_parser.cc +++ b/parser/caffe/caffe_parser.cc @@ -108,7 +108,7 @@ graphStatus aclgrphParseCaffe(const char *model_file, const char *weights_file, return ret; } GELOGI("Weights parse success. graph: %s", graph.GetName().c_str()); - std::map &parser_params; + std::map parser_params; if (acl_graph_parse_util.SetOutputNodeInfo(graph, parser_params) != ge::SUCCESS) { GELOGE(ret, "Set graph %s default output node failed.", graph.GetName().c_str()); return ge::FAILED; @@ -134,7 +134,7 @@ graphStatus aclgrphParseCaffe(const char *model_file, const char *weights_file, return ge::FAILED; } // Create an empty computegraph - string graph_name = output_name.empty() ? "tmpGraph", output_name; + string graph_name = output_name.empty() ? "tmpGraph" : output_name; ge::ComputeGraphPtr compute_graph = ge::parser::MakeShared(graph_name); GE_CHECK_NOTNULL(compute_graph); diff --git a/parser/tensorflow/tensorflow_parser.cc b/parser/tensorflow/tensorflow_parser.cc index 59c4973..473b007 100644 --- a/parser/tensorflow/tensorflow_parser.cc +++ b/parser/tensorflow/tensorflow_parser.cc @@ -113,7 +113,7 @@ graphStatus aclgrphParseTensorFlow(const char *model_file, ge::Graph &graph) { return ge::FAILED; } - std::map &parser_params; + std::map parser_params; if (acl_graph_parse_util.SetOutputNodeInfo(graph, parser_params) != ge::SUCCESS) { GELOGE(ret, "Set graph %s default output node failed.", graph.GetName().c_str()); return ge::FAILED; @@ -140,7 +140,7 @@ graphStatus aclgrphParseTensorFlow(const char *model_file, return ge::FAILED; } // Create an empty computegraph - string graph_name = output_name.empty() ? "tmpGraph", output_name; + string graph_name = output_name.empty() ? "tmpGraph" : output_name; ge::ComputeGraphPtr compute_graph = ge::parser::MakeShared(graph_name); GE_CHECK_NOTNULL(compute_graph);