From d9916526deb062129c8692499e33ac6d7ba1deae Mon Sep 17 00:00:00 2001
From: lichun <lichun30@hisilicon.com>
Date: Sat, 26 Jun 2021 17:04:34 +0800
Subject: [PATCH] ReshapeRemovePass_not_remove_reshape_in_dynamic_subgraph

---
 ge/graph/passes/reshape_remove_pass.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ge/graph/passes/reshape_remove_pass.cc b/ge/graph/passes/reshape_remove_pass.cc
index 856fda71..7d79a3c3 100755
--- a/ge/graph/passes/reshape_remove_pass.cc
+++ b/ge/graph/passes/reshape_remove_pass.cc
@@ -47,7 +47,7 @@ Status ReshapeRemovePass::Run(NodePtr &node) {
     case kReshapeType: {
       bool is_in_unknown_shape_graph = false;
       bool forced_unknown = false;
-      for (const auto &node : graph->GetDirectNode()) {
+      for (const auto &node : node->GetOwnerComputeGraph()->GetDirectNode()) {
         GE_CHK_GRAPH_STATUS_RET(ge::NodeUtils::GetNodeUnknownShapeStatus(*node, is_in_unknown_shape_graph),
                                 "[Get][ShapeStatus] of node[%s] failed!", node->GetName().c_str());
         if (is_in_unknown_shape_graph) {