forked from mindspore-Ecosystem/mindspore
!6395 fix calculate stride bug
Merge pull request !6395 from zhaozhenlong/lite/issue/mirror_pad_calc_stride_fix
This commit is contained in:
commit
2ea3f22937
|
@ -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];
|
||||
|
|
Loading…
Reference in New Issue