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

Fixed squeeze condition

Browse Source
This commit is contained in:
Mikhail Ryzhov 2023-03-22 15:29:59 +01:00
parent d105cfcc68
commit 4fd047deec
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/plugins/intel_gna/src/transformations/utils/transformation_helper.cpp
View File

@ -128,7 +128,7 @@ ov::Shape SqueezeShape(const ov::Shape& shape) {
auto start_it = std::find_if(shape.begin(), shape.end(), comp);
auto end_it = std::find_if(shape.rbegin(), shape.rend(), comp).base();
if (start_it == shape.end() || end_it == shape.end() || start_it < end_it) {
if (start_it == shape.end() || end_it == shape.end()) {
return ov::Shape(shape.begin(), shape.end());
}
return ov::Shape(start_it, end_it);