From 6b2bf715a290481998255bb7f2e7649098f1047c Mon Sep 17 00:00:00 2001 From: unknown Date: Fri, 5 Feb 2021 14:51:25 +0800 Subject: [PATCH] Add root_graph output_edges. --- ge/hybrid/model/hybrid_model_builder.cc | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/ge/hybrid/model/hybrid_model_builder.cc b/ge/hybrid/model/hybrid_model_builder.cc index 8383347c..d2862553 100755 --- a/ge/hybrid/model/hybrid_model_builder.cc +++ b/ge/hybrid/model/hybrid_model_builder.cc @@ -738,6 +738,7 @@ Status HybridModelBuilder::BuildOutputMapping(GraphItem &graph_item, graph_item.output_index_mapping_.emplace_back(p_index); } } + return SUCCESS; } @@ -1899,9 +1900,7 @@ Status HybridModelBuilder::LoadDynamicSubgraph(ComputeGraph &graph, bool is_root data_nodes.emplace_back(node_item); } else if (op_type == NETOUTPUT) { graph_item->output_node_ = node_item; - if (!is_root_graph) { - GE_CHK_STATUS_RET_NOLOG(BuildOutputMapping(*graph_item, *node_item)); - } + GE_CHK_STATUS_RET_NOLOG(BuildOutputMapping(*graph_item, *node_item, is_root_graph)); } GE_CHK_STATUS_RET_NOLOG(CreateProfilingNodeBefore(*graph_item, node)); graph_item->node_items_.emplace_back(node_item);