|
|
@@ -41,7 +41,7 @@ Status DynamicSingleOpResetShapePass::Run(ComputeGraphPtr graph) { |
|
|
|
std::vector<int64_t> dynamic_shape_dims = {-2}; |
|
|
|
ge::Shape dynamic_shape(dynamic_shape_dims); |
|
|
|
for (size_t i = 0; i < op_desc->GetAllInputsDesc().size(); i++) { |
|
|
|
const auto &input_desc = op_desc->MutableInputDesc(static_cast<uint32_t>(i)); |
|
|
|
auto input_desc = op_desc->MutableInputDesc(static_cast<uint32_t>(i)); |
|
|
|
GE_CHECK_NOTNULL(input_desc); |
|
|
|
input_desc->SetShape(dynamic_shape); |
|
|
|
} |
|
|
|