fix bug of strided slice

4 years ago · 6647e0e6ac
parent 4a1b9e4a12
commit 6647e0e6ac
1 changed files with 3 additions and 1 deletions
--- 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];
  }