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

Fix begin and end mask in StridedSlice in builder::opset1::flatten (#9837)

Browse Source
This commit is contained in:
Mateusz Tabaka 2022-01-25 13:04:36 +01:00 committed by GitHub
parent 94d403a1eb
commit fa605a2760
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/core/builder/src/builder/reshape.cpp
View File

@ -114,8 +114,8 @@ shared_ptr<Node> builder::opset1::flatten(const Output<Node>& value, int axis) {
make_shared<ngraph::opset1::StridedSlice>(value_shape,
ngraph::opset1::Constant::create(element::i64, {1}, {0}),
axis_node,
vector<int64_t>{},
vector<int64_t>{});
vector<int64_t>{0},
vector<int64_t>{0});
const auto first_part_dims_length =
make_shared<ngraph::opset1::ReduceProd>(first_part_dims,
ngraph::opset1::Constant::create(element::i64, {}, {0}),

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

@ -79,9 +79,9 @@ std::vector<StridedSliceSpecificParams> ss_only_test_cases = {
StridedSliceSpecificParams{ { 2, 2, 2, 2 }, { 1, 1, 1, 1 }, { 2, 2, 2, 2 }, { 1, 1, 1, 1 },
{ 0, 0, 0, 0}, { 0, 0, 0, 0}, {}, {}, {} },
StridedSliceSpecificParams{ { 1, 2, 6, 4 }, { 0, 0, 4, 0 }, { 1, 2, 6, 4 }, { 1, 1, 1, 1 },
{}, {}, {}, {}, {} },
{ 0, 0, 0, 0 }, { 0, 0, 0, 0 }, {}, {}, {} },
StridedSliceSpecificParams{ { 1, 2, 6, 4 }, { 0, 0, -3, 0 }, { 1, 2, 6, 4 }, { 1, 1, 1, 1 },
{}, {}, {}, {}, {} },
{ 0, 0, 0, 0 }, { 0, 0, 0, 0 }, {}, {}, {} },
StridedSliceSpecificParams{ { 1, 2, 6, 4 }, { 0, 0, 4, 0 }, { 1, 2, 6, 4 }, { 1, 1, 1, 1 },
{ 1, 1, 0, 1}, { 1, 1, 1, 1}, {}, {}, {} },
StridedSliceSpecificParams{ { 10, 2, 2, 2 }, { 0, 0, 0, 0 }, { 0, 0, 0, 0 }, { 2, 1, 1, 1 },