Browse Source

Fix reshape bug.

tags/v1.2.0
unknown 3 years ago
parent
commit
637416941e
1 changed files with 3 additions and 4 deletions
  1. +3
    -4
      ge/generator/ge_generator.cc

+ 3
- 4
ge/generator/ge_generator.cc View File

@@ -627,12 +627,11 @@ namespace {
bool is_dynamic = false; bool is_dynamic = false;
for (const auto &node : graph->GetDirectNode()) { for (const auto &node : graph->GetDirectNode()) {
GE_CHECK_NOTNULL(node); GE_CHECK_NOTNULL(node);
if (node->GetType() == DATA || node->GetType() == CONSTANT || node->GetType() == CONSTANTOP ||
node->GetType() == NETOUTPUT) {
continue;
}
auto op_desc = node->GetOpDesc(); auto op_desc = node->GetOpDesc();
GE_CHECK_NOTNULL(op_desc); GE_CHECK_NOTNULL(op_desc);
if (op_desc->GetOpEngineName() != kAIcoreEngine) {
continue;
}
if (AttrUtils::HasAttr(op_desc, kAttrSupportDynamicShape)) { if (AttrUtils::HasAttr(op_desc, kAttrSupportDynamicShape)) {
is_dynamic = true; is_dynamic = true;
(void) AttrUtils::GetBool(op_desc, kAttrSupportDynamicShape, support_dynamic); (void) AttrUtils::GetBool(op_desc, kAttrSupportDynamicShape, support_dynamic);


Loading…
Cancel
Save