From 64c4bbd8e7675e6034e80b450163a4e8c73754ee Mon Sep 17 00:00:00 2001 From: wxl Date: Sat, 13 Mar 2021 15:40:46 +0800 Subject: [PATCH] fix bug that all subgraph is unknown and netoutput format is not nd bug --- ge/graph/passes/net_output_pass.cc | 2 -- 1 file changed, 2 deletions(-) diff --git a/ge/graph/passes/net_output_pass.cc b/ge/graph/passes/net_output_pass.cc index 37de2af9..c553607f 100644 --- a/ge/graph/passes/net_output_pass.cc +++ b/ge/graph/passes/net_output_pass.cc @@ -202,8 +202,6 @@ Status NetOutputPass::UpdateNetOutputDesc(const ge::NodePtr &net_output) { GE_CHECK_NOTNULL(src_op_desc); uint32_t peer_index = static_cast(in_anchor->GetPeerOutAnchor()->GetIdx()); ge::GeTensorDesc output_in_desc = src_op_desc->GetOutputDesc(peer_index); - output_in_desc.SetFormat(FORMAT_ND); - output_in_desc.SetOriginFormat(FORMAT_ND); if (net_output_desc->UpdateInputDesc(index, output_in_desc) != GRAPH_SUCCESS) { GELOGE(INTERNAL_ERROR, "Update input desc failed, index:%u.", index); return INTERNAL_ERROR;