fix bug that all subgraph is unknown and netoutput format is not nd bug

pull/1249/head
wxl 4 years ago
parent 64c4bbd8e7
commit 08ecc484fd

@ -555,6 +555,8 @@ void NetOutputPass::AddInOutForNetOutputOp(const ComputeGraphPtr &graph, OpDescP
return; return;
} }
ge::GeTensorDesc out_desc = src_node->GetOpDesc()->GetOutputDesc(src_index); ge::GeTensorDesc out_desc = src_node->GetOpDesc()->GetOutputDesc(src_index);
out_desc.SetFormat(FORMAT_ND);
out_desc.SetOriginFormat(FORMAT_ND);
GE_IF_BOOL_EXEC(net_output_desc->AddInputDesc(out_desc) != SUCCESS, GELOGW("add input desc failed"); return ); GE_IF_BOOL_EXEC(net_output_desc->AddInputDesc(out_desc) != SUCCESS, GELOGW("add input desc failed"); return );
is_input_const.push_back(PassUtils::IsConstant(src_node)); is_input_const.push_back(PassUtils::IsConstant(src_node));
++iter; ++iter;

Loading…
Cancel
Save