Browse Source

Feature: reset shape of dynamic single op

pull/563/head
l00444296 4 years ago
parent
commit
80ee77be05
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      ge/graph/passes/dynamic_single_op_reset_shape_pass.cc
  2. +1
    -1
      metadef

+ 1
- 1
ge/graph/passes/dynamic_single_op_reset_shape_pass.cc View File

@@ -43,7 +43,7 @@ Status DynamicSingleOpResetShapePass::Run(ComputeGraphPtr graph) {
for (size_t i = 0; i < op_desc->GetAllInputsDesc().size(); i++) {
const auto &input_desc = op_desc->MutableInputDesc(static_cast<uint32_t>(i));
GE_CHECK_NOTNULL(input_desc);
input_desc.SetShape(dynamic_shape);
input_desc->SetShape(dynamic_shape);
}
GELOGD("Reset dynamic aicpu node [%s] shape success!", node->GetName().c_str());
}


+ 1
- 1
metadef

@@ -1 +1 @@
Subproject commit 5b9a7f84a4347f8816d492aa51f2414ccf8a0744
Subproject commit 0aee977581117ca5c8df5d5ccdf7333c6ea33834

Loading…
Cancel
Save