Browse Source

Bugfix:fix error of get null subgraph

tags/v1.3.0
lichun 3 years ago
parent
commit
c99ad051b9
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      ge/hybrid/model/hybrid_model_builder.cc

+ 1
- 0
ge/hybrid/model/hybrid_model_builder.cc View File

@@ -601,6 +601,7 @@ Status HybridModelBuilder::MergeNetOutputNode(ComputeGraph &graph) {

Status HybridModelBuilder::UnfoldSubgraphs(ComputeGraphPtr &root_graph, ComputeGraphPtr &merged_graph) {
merged_graph = MakeShared<ComputeGraph>("MergedGraph");
merged_graph->SetGraphUnknownFlag(root_graph->GetGraphUnknownFlag());
for (const auto &node : root_graph->GetDirectNode()) {
GE_CHECK_NOTNULL(node);
auto op_desc = node->GetOpDesc();


Loading…
Cancel
Save