p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix bug of strided slice

This commit is contained in:
fuzhiye 2021-03-18 10:12:50 +08:00
parent 4a1b9e4a12
commit 6647e0e6ac
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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