From c75f3f0c7ae030c6e634b3894fdcc5e451f49f0b Mon Sep 17 00:00:00 2001 From: Lyamin-Roman Date: Thu, 14 Mar 2024 11:20:55 +0900 Subject: [PATCH] [Transformations] Added If operation to NMS path propagation for ignore negative indices in Gather --- .../convert_nms_gather_path_to_unsigned.cpp | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/common/transformations/src/transformations/common_optimizations/convert_nms_gather_path_to_unsigned.cpp b/src/common/transformations/src/transformations/common_optimizations/convert_nms_gather_path_to_unsigned.cpp index a59232ed2e461e..a4f42063bcb1da 100644 --- a/src/common/transformations/src/transformations/common_optimizations/convert_nms_gather_path_to_unsigned.cpp +++ b/src/common/transformations/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();