From 848236b21c0f492f141354f852d0ac23d8b45d63 Mon Sep 17 00:00:00 2001
From: zhaoxinxin <zhaoxinxin1@huawei.com>
Date: Mon, 15 Mar 2021 20:47:19 +0800
Subject: [PATCH] 	modified:   ge/graph/passes/infershape_pass.cc

---
 ge/graph/passes/infershape_pass.cc | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/ge/graph/passes/infershape_pass.cc b/ge/graph/passes/infershape_pass.cc
index fb18204c..a54a15c1 100755
--- a/ge/graph/passes/infershape_pass.cc
+++ b/ge/graph/passes/infershape_pass.cc
@@ -42,9 +42,9 @@ Status InferShapePass::Run(NodePtr &node) {
     GELOGE(GE_GRAPH_INFERSHAPE_FAILED, "infershape failed. node: %s", node->GetName().c_str());
     return GE_GRAPH_INFERSHAPE_FAILED;
   }
-  if (node->GetType() == WHILE) {
-    bool need_repass = false;
-    AttrUtils::GetBool(node->GetOpDesc(), "need_infer_again_", need_repass);
+  bool need_repass = false;
+  auto has_attr = AttrUtils::GetBool(node->GetOpDesc(), "need_infer_again_", need_repass);
+  if (has_attr) {
     if (!OptionExists(kOptimizeAfterSubGraph)) {
       return SUCCESS;
     }