Browse Source

fix space to depth

tags/v0.7.0-beta
sunsuodong 6 years ago
parent
commit
fa6800c798
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

@@ -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) {


Loading…
Cancel
Save