modified: ge/graph/passes/no_use_reshape_remove_pass.cc

4 years ago · db230d8460
parent dcbaf26680
commit db230d8460
1 changed files with 2 additions and 1 deletions
--- a/ge/graph/passes/no_use_reshape_remove_pass.cc
+++ b/ge/graph/passes/no_use_reshape_remove_pass.cc
@ -95,7 +95,8 @@ Status NoUseReshapeRemovePass::TryRemoveConstShapeInput(ge::NodePtr &reshape_nod
  if (shape_input_anchor == nullptr) {
    return SUCCESS;
  }
-  auto shape_input = shape_input_anchor->GetOwnerNode();
+  GE_CHECK_NOTNULL(shape_input_anchor->GetPeerOutAnchor());
+  auto shape_input = shape_input_anchor->GetPeerOutAnchor()->GetOwnerNode();
  GE_CHECK_NOTNULL(shape_input);
  if (shape_input->GetType() != CONSTANT && shape_input->GetType() != CONSTANTOP) {
    return SUCCESS;