From 637416941ecb0b90e557e3567c4892c03c87496b Mon Sep 17 00:00:00 2001
From: unknown <zhaozhixuan2@hisilicon.com>
Date: Thu, 18 Feb 2021 21:11:12 +0800
Subject: [PATCH] Fix reshape bug.

---
 ge/generator/ge_generator.cc | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/ge/generator/ge_generator.cc b/ge/generator/ge_generator.cc
index e9c2e6ab..0d484fbf 100644
--- a/ge/generator/ge_generator.cc
+++ b/ge/generator/ge_generator.cc
@@ -627,12 +627,11 @@ namespace {
     bool is_dynamic = false;
     for (const auto &node : graph->GetDirectNode()) {
       GE_CHECK_NOTNULL(node);
-      if (node->GetType() == DATA || node->GetType() == CONSTANT || node->GetType() == CONSTANTOP ||
-          node->GetType() == NETOUTPUT) {
-        continue;
-      }
       auto op_desc = node->GetOpDesc();
       GE_CHECK_NOTNULL(op_desc);
+      if (op_desc->GetOpEngineName() != kAIcoreEngine) {
+        continue;
+      }
       if (AttrUtils::HasAttr(op_desc, kAttrSupportDynamicShape)) {
         is_dynamic = true;
         (void) AttrUtils::GetBool(op_desc, kAttrSupportDynamicShape, support_dynamic);