Browse Source

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

tags/v1.2.0
wxl 3 years ago
parent
commit
64c4bbd8e7
1 changed files with 0 additions and 2 deletions
  1. +0
    -2
      ge/graph/passes/net_output_pass.cc

+ 0
- 2
ge/graph/passes/net_output_pass.cc View File

@@ -202,8 +202,6 @@ Status NetOutputPass::UpdateNetOutputDesc(const ge::NodePtr &net_output) {
GE_CHECK_NOTNULL(src_op_desc); GE_CHECK_NOTNULL(src_op_desc);
uint32_t peer_index = static_cast<uint32_t>(in_anchor->GetPeerOutAnchor()->GetIdx()); uint32_t peer_index = static_cast<uint32_t>(in_anchor->GetPeerOutAnchor()->GetIdx());
ge::GeTensorDesc output_in_desc = src_op_desc->GetOutputDesc(peer_index); 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) { if (net_output_desc->UpdateInputDesc(index, output_in_desc) != GRAPH_SUCCESS) {
GELOGE(INTERNAL_ERROR, "Update input desc failed, index:%u.", index); GELOGE(INTERNAL_ERROR, "Update input desc failed, index:%u.", index);
return INTERNAL_ERROR; return INTERNAL_ERROR;


Loading…
Cancel
Save