diff --git a/src/ge/graph/build/graph_builder.cc b/src/ge/graph/build/graph_builder.cc index 27d0b13f..6c069483 100644 --- a/src/ge/graph/build/graph_builder.cc +++ b/src/ge/graph/build/graph_builder.cc @@ -113,6 +113,7 @@ Status GraphBuilder::UpdateParentNodeOutputSize(const ge::ComputeGraphPtr &graph parent_node_ptr->GetName().c_str()); return SUCCESS; } + for (const auto &node_ptr : graph->GetDirectNode()) { if (node_ptr->GetType() != NETOUTPUT) { continue;