|
|
@ -556,6 +556,11 @@ Status NetOutputPass::AddNetOutputNodeToGraph(const ge::ComputeGraphPtr &graph,
|
|
|
|
output_node = graph->AddNode(net_output_desc);
|
|
|
|
output_node = graph->AddNode(net_output_desc);
|
|
|
|
GE_CHK_STATUS_RET(AddCtrlEdgesBetweenLeafAndNetOutput(graph, output_node),
|
|
|
|
GE_CHK_STATUS_RET(AddCtrlEdgesBetweenLeafAndNetOutput(graph, output_node),
|
|
|
|
"add ctrl edge between leaf and netoutput failed");
|
|
|
|
"add ctrl edge between leaf and netoutput failed");
|
|
|
|
|
|
|
|
GE_IF_BOOL_EXEC(!ge::AttrUtils::SetInt(output_node->GetOpDesc(), ATTR_NAME_TRUE_BRANCH_STREAM, 0),
|
|
|
|
|
|
|
|
REPORT_CALL_ERROR("E19999", "Set Attr:%s to op:%s(%s) failed", ATTR_NAME_TRUE_BRANCH_STREAM.c_str(),
|
|
|
|
|
|
|
|
output_node->GetName().c_str(), output_node->GetType().c_str());
|
|
|
|
|
|
|
|
GELOGE(INTERNAL_ERROR, "set ATTR_NAME_TRUE_BRANCH_STREAM failed");
|
|
|
|
|
|
|
|
return INTERNAL_ERROR);
|
|
|
|
return SUCCESS;
|
|
|
|
return SUCCESS;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
GELOGI("[NETOUTPUT PASS] Output node size:%lu.", output_nodes_info.size());
|
|
|
|
GELOGI("[NETOUTPUT PASS] Output node size:%lu.", output_nodes_info.size());
|
|
|
|