From cf52dbb46d794743f332c9789a2b1757de5232e7 Mon Sep 17 00:00:00 2001
From: tao_yunhao <taoyunhao@huawei.com>
Date: Thu, 17 Sep 2020 09:18:36 +0800
Subject: [PATCH] fix bug of op space_to_depth

---
 mindspore/lite/src/runtime/kernel/arm/fp32/space_to_depth.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mindspore/lite/src/runtime/kernel/arm/fp32/space_to_depth.cc b/mindspore/lite/src/runtime/kernel/arm/fp32/space_to_depth.cc
index d1dc8bafe3..896418fec5 100644
--- a/mindspore/lite/src/runtime/kernel/arm/fp32/space_to_depth.cc
+++ b/mindspore/lite/src/runtime/kernel/arm/fp32/space_to_depth.cc
@@ -50,7 +50,7 @@ int SpaceToDepthCPUKernel::ReSize() {
     return RET_FORMAT_ERR;
   }
 
-  num_unit_ = static_cast<int>(in_tensors_[0]->shape().at(kNHWC_H));
+  num_unit_ = static_cast<int>(out_tensors_[0]->shape().at(kNHWC_H));
   thread_h_num_ = MSMIN(op_parameter_->thread_num_, num_unit_);
   thread_h_stride_ = UP_DIV(num_unit_, thread_h_num_);
   return RET_OK;