p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix space to depth

This commit is contained in:
sunsuodong 2020-08-17 20:40:00 +08:00
parent c102771fad
commit fa6800c798
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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