Pre Merge pull request !1481 from 薛鹏/master

pull/1481/MERGE
薛鹏 4 years ago committed by Gitee
commit 7ebd34bb30

@ -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());

Loading…
Cancel
Save