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

…re negative indices in Gather
openvinotoolkit · Mar 14, 2024 · 95040e3 · 95040e3
1 parent f5259bd
commit 95040e3
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
@@ -19,6 +19,7 @@
 #include "openvino/op/util/broadcast_base.hpp"
 #include "openvino/op/util/gather_base.hpp"
 #include "openvino/op/variadic_split.hpp"
+#include "openvino/op/if.hpp"
 #include "openvino/pass/pattern/op/wrap_type.hpp"
 #include "transformations/rt_info/nms_selected_indices.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();