Browse Source

Pre Merge pull request !621 from 刘江涛/ge_dev

pull/621/MERGE
刘江涛 Gitee 3 years ago
parent
commit
85ec54ce3b
No known key found for this signature in database GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 4 additions and 2 deletions
  1. +4
    -2
      parser/tensorflow/tensorflow_parser.cc

+ 4
- 2
parser/tensorflow/tensorflow_parser.cc View File

@@ -101,7 +101,8 @@ graphStatus aclgrphParseTensorFlow(const char *model_file, ge::Graph &graph) {
}

// Create an empty computegraph
ge::ComputeGraphPtr compute_graph = ge::parser::MakeShared<ge::ComputeGraph>("tmpGraph");
ge::ComputeGraphPtr compute_graph = ge::parser::MakeShared<ge::ComputeGraph>("tmpGraph" +
std::to_string(ge::parser::GetCurrentTimestamp()));
if (compute_graph == nullptr) {
REPORT_CALL_ERROR("E19999", "New ComputeGraph failed");
GELOGE(FAILED, "Create ComputeGraph fail.");
@@ -154,7 +155,8 @@ graphStatus aclgrphParseTensorFlow(const char *model_file, const std::map<Ascend
return ge::FAILED;
}
// Create an empty computegraph
string graph_name = output_name.empty() ? "tmpGraph" : output_name;
string graph_name = output_name.empty() ? ("tmpGraph" +
std::to_string(ge::parser::GetCurrentTimestamp())) : output_name;
ge::ComputeGraphPtr compute_graph = ge::parser::MakeShared<ge::ComputeGraph>(graph_name);
if (compute_graph == nullptr) {
REPORT_CALL_ERROR("E19999", "New ComputeGraph failed");


Loading…
Cancel
Save