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

!48132 bugfix for StridedSlice in changed batch scene 

Merge pull request !48132 from zhangyanhui/personal_mas
This commit is contained in:
i-robot 2023-01-30 01:29:18 +00:00 committed by Gitee
parent 2b05fd48cb 1b06812430
commit 09457b7fb3
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
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++) {
inner_ *= in_shape[i];
}
parallel_on_split_axis_ = false;
parallel_on_outer_ = false;
outer_ == 1 ? (parallel_on_split_axis_ = true) : (parallel_on_outer_ = true);
if (UpdateThreadNumPass(TC_TYPE(PrimitiveType_StridedSlice, parallel_on_outer_), 1, 1,