Browse Source

modified: ge/graph/passes/no_use_reshape_remove_pass.cc

tags/v1.2.0
zhaoxinxin 4 years ago
parent
commit
db230d8460
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      ge/graph/passes/no_use_reshape_remove_pass.cc

+ 2
- 1
ge/graph/passes/no_use_reshape_remove_pass.cc View File

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


Loading…
Cancel
Save