|
|
|
@@ -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<string, string> &parser_params; |
|
|
|
std::map<string, string> 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<ge::ComputeGraph>(graph_name); |
|
|
|
GE_CHECK_NOTNULL(compute_graph); |
|
|
|
|
|
|
|
|