diff --git a/mindspore/lite/src/runtime/kernel/arm/fp32/space_to_batch.cc b/mindspore/lite/src/runtime/kernel/arm/fp32/space_to_batch.cc
index 647372e6e3..5bf1e4fa0f 100644
--- a/mindspore/lite/src/runtime/kernel/arm/fp32/space_to_batch.cc
+++ b/mindspore/lite/src/runtime/kernel/arm/fp32/space_to_batch.cc
@@ -89,7 +89,7 @@ int SpaceToBatchCPUKernel::ReSize() {
   padded_in_shape_[2] = in_shape[2] + param->paddings_[2] + param->paddings_[3];
   auto num_elements_padded = EnumElement(padded_in_shape_.data(), in_shape.size());
   auto output_shape = out_tensors_[0]->shape();
-  auto pedding_h_size = output_shape[2] * output_shape[3] * sizeof(float);
+  auto pedding_h_size = padded_in_shape_[2] * output_shape[3] * sizeof(float);
   pedding_h_data_ = reinterpret_cast<float *>(context_->allocator->Malloc(pedding_h_size));
   if (pedding_h_data_ == nullptr) {
     MS_LOG(ERROR) << "malloc pedding h data fail!";
diff --git a/mindspore/lite/test/models_tflite.cfg b/mindspore/lite/test/models_tflite.cfg
index 2ce1231bcf..7ced66b177 100644
--- a/mindspore/lite/test/models_tflite.cfg
+++ b/mindspore/lite/test/models_tflite.cfg
@@ -89,6 +89,7 @@ hiai_cv_labelDetectorModel_v2.tflite
 #hiai_cv_labelDetectorModel_v3.tflite
 hiai_cv_labelDetectorModel_v4.tflite
 hiai_dress_detect.tflite
+hiai_cv_saliencyDetectorModel.tflite
 hiai_frozen_inference_graph.tflite
 hiai_ghostnet.tflite
 hiai_iMaxDN_RGB.tflite