[Transformations] Added If operation to NMS path propagation for igno…

…re negative indices in Gather
openvinotoolkit · Mar 14, 2024 · c75f3f0 · c75f3f0
1 parent f5259bd
commit c75f3f0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/...rmations/src/transformations/common_optimizations/convert_nms_gather_path_to_unsigned.cpp b/...rmations/src/transformations/common_optimizations/convert_nms_gather_path_to_unsigned.cpp
@@ -10,6 +10,7 @@
 #include "openvino/core/rt_info.hpp"
 #include "openvino/op/concat.hpp"
 #include "openvino/op/convert.hpp"
+#include "openvino/op/if.hpp"
 #include "openvino/op/non_max_suppression.hpp"
 #include "openvino/op/reshape.hpp"
 #include "openvino/op/slice.hpp"
@@ -60,7 +61,8 @@ class PropagateNMSPath : public pass::MatcherPass {
                                                ov::op::v1::VariadicSplit,
                                                op::util::GatherBase,
                                                ov::op::v0::Concat,
-                                               ov::op::v0::Convert>();
+                                               ov::op::v0::Convert,
+                                               ov::op::v8::If>();
         matcher_pass_callback callback = [=](pattern::Matcher& m) {
             auto node = m.get_match_root();
             const auto& inputs = node->input_values();