Browse Source

Feature: reset shape of dynamic single op

pull/565/head
l00444296 4 years ago
parent
commit
976d6a6b76
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/graph/passes/dynamic_single_op_reset_shape_pass.cc

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

@@ -39,7 +39,7 @@ Status DynamicSingleOpResetShapePass::Run(ComputeGraphPtr graph) {

auto op_desc = node->GetOpDesc();
std::vector<int64_t> dynamic_shape_dims = {-2};
ge::Shape dynamic_shape(dynamic_shape_dims);
GeShape dynamic_shape(dynamic_shape_dims);
for (size_t i = 0; i < op_desc->GetAllInputsDesc().size(); i++) {
auto input_desc = op_desc->MutableInputDesc(static_cast<uint32_t>(i));
GE_CHECK_NOTNULL(input_desc);


Loading…
Cancel
Save