From 6647e0e6acb62a88ef61d40accf1bcd061db03fe Mon Sep 17 00:00:00 2001 From: fuzhiye Date: Thu, 18 Mar 2021 10:12:50 +0800 Subject: [PATCH] fix bug of strided slice --- mindspore/lite/src/runtime/kernel/arm/base/strided_slice.cc | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/mindspore/lite/src/runtime/kernel/arm/base/strided_slice.cc b/mindspore/lite/src/runtime/kernel/arm/base/strided_slice.cc index f5dfdfd239..a7baa6dcb1 100644 --- a/mindspore/lite/src/runtime/kernel/arm/base/strided_slice.cc +++ b/mindspore/lite/src/runtime/kernel/arm/base/strided_slice.cc @@ -37,7 +37,9 @@ int StridedSliceCPUKernel::Init() { void StridedSliceCPUKernel::InitFastRunParam() { auto in_shape = in_tensors_.front()->shape(); auto out_shape = out_tensors_.front()->shape(); - // cal inner, outer + // reset && cal inner, outer + outer_ = 1; + inner_ = 1; for (int i = 0; i < split_axis_; ++i) { outer_ *= in_shape[i]; }