Browse Source

fix bug of strided slice

tags/v1.2.0-rc1
fuzhiye 4 years ago
parent
commit
6647e0e6ac
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      mindspore/lite/src/runtime/kernel/arm/base/strided_slice.cc

+ 3
- 1
mindspore/lite/src/runtime/kernel/arm/base/strided_slice.cc View File

@@ -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];
}


Loading…
Cancel
Save