diff --git a/mindspore/lite/src/populate_parameter.cc b/mindspore/lite/src/populate_parameter.cc index f31d50603c..f6f7363559 100644 --- a/mindspore/lite/src/populate_parameter.cc +++ b/mindspore/lite/src/populate_parameter.cc @@ -1038,7 +1038,7 @@ OpParameter *PopulateSpaceToDepthParameter(const lite::Primitive *primitive) { return nullptr; } space_depth_param->op_parameter_.type_ = primitive->Type(); - auto param = primitive->Value()->value_as_DepthToSpace(); + auto param = primitive->Value()->value_as_SpaceToDepth(); space_depth_param->op_parameter_.type_ = primitive->Type(); space_depth_param->block_size_ = param->blockSize(); if (param->format() != schema::Format_NHWC) {