|
|
@@ -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()); |
|
|
|
} |
|
|
|