From fa6800c798a7e37055a1faa9a34ce75a4df2c6ae Mon Sep 17 00:00:00 2001
From: sunsuodong <sunsuodong@huawei.com>
Date: Mon, 17 Aug 2020 20:40:00 +0800
Subject: [PATCH] fix space to depth

---
 mindspore/lite/src/populate_parameter.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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