Browse Source

Add root_graph output_edges.

tags/v1.2.0
unknown 3 years ago
parent
commit
6b2bf715a2
1 changed files with 2 additions and 3 deletions
  1. +2
    -3
      ge/hybrid/model/hybrid_model_builder.cc

+ 2
- 3
ge/hybrid/model/hybrid_model_builder.cc View File

@@ -738,6 +738,7 @@ Status HybridModelBuilder::BuildOutputMapping(GraphItem &graph_item,
graph_item.output_index_mapping_.emplace_back(p_index); graph_item.output_index_mapping_.emplace_back(p_index);
} }
} }

return SUCCESS; return SUCCESS;
} }


@@ -1899,9 +1900,7 @@ Status HybridModelBuilder::LoadDynamicSubgraph(ComputeGraph &graph, bool is_root
data_nodes.emplace_back(node_item); data_nodes.emplace_back(node_item);
} else if (op_type == NETOUTPUT) { } else if (op_type == NETOUTPUT) {
graph_item->output_node_ = node_item; 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)); GE_CHK_STATUS_RET_NOLOG(CreateProfilingNodeBefore(*graph_item, node));
graph_item->node_items_.emplace_back(node_item); graph_item->node_items_.emplace_back(node_item);


Loading…
Cancel
Save