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);