|
|
|
@@ -556,6 +556,11 @@ Status NetOutputPass::AddNetOutputNodeToGraph(const ge::ComputeGraphPtr &graph, |
|
|
|
output_node = graph->AddNode(net_output_desc); |
|
|
|
GE_CHK_STATUS_RET(AddCtrlEdgesBetweenLeafAndNetOutput(graph, output_node), |
|
|
|
"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; |
|
|
|
} |
|
|
|
GELOGI("[NETOUTPUT PASS] Output node size:%lu.", output_nodes_info.size()); |
|
|
|
|