IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[CPU] Fixed default ellipsis in StridedSlice (#10713)

Browse Source
This commit is contained in:
Alexandra Sidorova 2022-03-16 14:09:42 +03:00 committed by GitHub
parent 49fb48e744
commit 68452e97e2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/plugins/intel_cpu/src/nodes/strided_slice.cpp
View File

@ -119,10 +119,12 @@ MKLDNNStridedSliceNode::MKLDNNStridedSliceNode(const std::shared_ptr<ov::Node>&
attrs.shrinkAxisMask = createMask(ss->get_shrink_axis_mask());
auto origEllipsisMask = ss->get_ellipsis_mask();
bool isEllipsis = false;
for (const auto &o : origEllipsisMask) {
isEllipsis = isEllipsis || o != 0;
attrs.ellipsisMask.push_back(o);
}
if (attrs.ellipsisMask.size() == 0) {
if (attrs.ellipsisMask.size() == 0 || !isEllipsis) {
for (size_t i = attrs.ellipsisMask.size(); i < nDims; ++i) attrs.ellipsisMask.push_back(0);
}
} else {

2
src/tests/functional/plugin/cpu/shared_tests_instances/single_layer_tests/strided_slice.cpp
View File

@ -66,6 +66,8 @@ std::vector<StridedSliceSpecificParams> ss_only_test_cases = {
{ 1, 0 }, { 1, 0 }, { }, { }, { 1, 0 } },
StridedSliceSpecificParams{ { 20, 10, 5 }, { 0, 0 }, { 0, -1 }, { 1, 1 },
{ 1, 0 }, { 1, 0 }, { }, { }, { 1, 0 } },
StridedSliceSpecificParams{ { 20, 10, 5 }, { 0, 0 }, { 0, -1 }, { 1, 1 },
{ 1, 0 }, { 1, 0 }, { 0, 0 }, { 0, 0 }, { 0, 0 } },
StridedSliceSpecificParams{ { 1, 12, 100, 1, 1 }, { 0, -1, 0, 0 }, { 0, 0, 0, 0 }, { 1, 1, 1, 1 },
{ 1, 0, 1, 0 }, { 1, 0, 1, 0 }, { }, { 0, 1, 0, 1 }, {} },
StridedSliceSpecificParams{ { 2, 2, 2, 2 }, { 0, 0, 0, 0 }, { 2, 2, 2, 2 }, { 1, 1, 1, 1 },