Col_lin
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!13514 [MS][LITE]fix bug of strided slice 

From: @fuzhiye
Reviewed-by: @zhang_xue_tong,@zhanghaibo5
Signed-off-by: @zhang_xue_tong
This commit is contained in:
mindspore-ci-bot 2021-03-18 11:16:28 +08:00 committed by Gitee
parent 34b16e6a64 6647e0e6ac
commit ad1658b928
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];
}