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

bugfix for StridedSlice in changed batch scene

This commit is contained in:
zhangyanhui 2023-01-29 14:54:36 +08:00
parent 34ba528c42
commit 142f0292d5
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/lite/src/litert/kernel/cpu/base/strided_slice.cc
View File

@ -53,6 +53,8 @@ void StridedSliceCPUKernel::InitFastRunParam() {
for (size_t i = static_cast<size_t>(split_axis_ + 1); i < in_shape.size(); i++) { for (size_t i = static_cast<size_t>(split_axis_ + 1); i < in_shape.size(); i++) {
inner_ *= in_shape[i]; inner_ *= in_shape[i];
} }
parallel_on_split_axis_ = false;
parallel_on_outer_ = false;
outer_ == 1 ? (parallel_on_split_axis_ = true) : (parallel_on_outer_ = true); outer_ == 1 ? (parallel_on_split_axis_ = true) : (parallel_on_outer_ = true);
if (UpdateThreadNumPass(TC_TYPE(PrimitiveType_StridedSlice, parallel_on_outer_), 1, 1, if (UpdateThreadNumPass(TC_TYPE(PrimitiveType_StridedSlice, parallel_on_outer_), 1, 1,