!6395 fix calculate stride bug

Merge pull request !6395 from zhaozhenlong/lite/issue/mirror_pad_calc_stride_fix
4 years ago · 2ea3f22937
parent 1663af7591 b491055a70
commit 2ea3f22937
1 changed files with 1 additions and 2 deletions
--- a/mindspore/lite/src/runtime/kernel/arm/fp32/pad.cc
+++ b/mindspore/lite/src/runtime/kernel/arm/fp32/pad.cc
@ -201,10 +201,9 @@ int PadCPUKernel::CopyPaddingFromInput() {
 }

 void PadCPUKernel::CalculateStrides() {
-  auto input_shape = in_tensors_.at(0)->shape();
  pad_param_->in_strides[DEFAULT_PAD_NDIMS - 1] = 1;
  for (auto i = DEFAULT_PAD_NDIMS - 2; i >= 0; --i) {
-    pad_param_->in_strides[i] = input_shape[i + 1] * pad_param_->in_strides[i + 1];
+    pad_param_->in_strides[i] = in_[i + 1] * pad_param_->in_strides[i + 1];
  }
  for (auto i = 0; i < DEFAULT_PAD_NDIMS; ++i) {
    out_[i] = in_[i] + pad_param_->paddings_[i * 2] + pad_param_->paddings_[i * 2 + 1];