Browse Source

!4613 [MS][LITE[Develop]]fix space to depth

Merge pull request !4613 from sunsuodong/fix_space_to_depth
tags/v0.7.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
233e90fca1
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/lite/src/populate_parameter.cc

+ 1
- 1
mindspore/lite/src/populate_parameter.cc View File

@@ -1051,7 +1051,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) {


Loading…
Cancel
Save