From db230d8460935731d0a3acaff3d0ab5f31306fb5 Mon Sep 17 00:00:00 2001 From: zhaoxinxin Date: Wed, 24 Feb 2021 09:58:18 +0800 Subject: [PATCH] modified: ge/graph/passes/no_use_reshape_remove_pass.cc --- ge/graph/passes/no_use_reshape_remove_pass.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ge/graph/passes/no_use_reshape_remove_pass.cc b/ge/graph/passes/no_use_reshape_remove_pass.cc index c2b8bdad..1da939c6 100644 --- 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;