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

[Core] Fix leftovers from PR13581 (#13693)

Browse Source 
* [Core] Fix leftovers from PR13581

Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com>

* Revert incorrect change for shrink axis

Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com>
This commit is contained in:
Roman Kazantsev 2022-10-28 23:15:27 +03:00 committed by GitHub
parent 04f09ba0c0
commit 22de883e6e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/core/shape_inference/include/strided_slice_shape_inference.hpp
View File

@ -82,9 +82,7 @@ void shape_infer(const StridedSlice* op,
auto number_axes = number_elements_in_1d(op, begin_shape);
auto end_number_axes = number_elements_in_1d(op, end_shape);
if (number_axes != -1 && end_number_axes != -1) {
NODE_VALIDATION_CHECK(op,
number_axes == end_number_axes,
"Lower bounds and Upper bounds need to have same number of values");
NODE_VALIDATION_CHECK(op, number_axes == end_number_axes, "Begin and end need to have same number of values.");
} else if (end_number_axes != -1) {
number_axes = end_number_axes;
}
@ -92,7 +90,7 @@ void shape_infer(const StridedSlice* op,
if (number_axes != -1 && strides_number_axes != -1) {
NODE_VALIDATION_CHECK(op,
number_axes == strides_number_axes,
"Stride needs to have same number of values as Lower and Upper bounds");
"Stride needs to have same number of values as begin and end.");
} else if (strides_number_axes != -1) {
number_axes = strides_number_axes;
}