Removed ONNX transformations for Less/GreaterOrEqual (openvinotoolkit…

…#21984)
ilya-lavrenov · Jan 8, 2024 · dcf87ef · dcf87ef
1 parent 8c09df0
commit dcf87ef
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 7 deletions.
diff --git a/src/frontends/onnx/frontend/src/core/transform.hpp b/src/frontends/onnx/frontend/src/core/transform.hpp
@@ -12,8 +12,6 @@ namespace transform {
 
 static const std::vector<std::string> onnx_functions_to_expand = {"Bernoulli",
                                                                   "Celu",
-                                                                  "GreaterOrEqual",
-                                                                  "LessOrEqual",
                                                                   "NegativeLogLikelihoodLoss",
                                                                   "SoftmaxCrossEntropyLoss",
                                                                   "LayerNormalization"};

diff --git a/src/frontends/onnx/frontend/src/ops_bridge.cpp b/src/frontends/onnx/frontend/src/ops_bridge.cpp
@@ -403,8 +403,8 @@ OperatorsBridge::OperatorsBridge() {
     REGISTER_OPERATOR("GlobalLpPool", 1, global_lp_pool);
     REGISTER_OPERATOR("GlobalMaxPool", 1, global_max_pool);
     REGISTER_OPERATOR("Greater", 1, greater);
-    REGISTER_OPERATOR("Greater_Or_Equal", 1, greater_or_equal);
-    REGISTER_OPERATOR("Greater_Or_Equal", 16, greater_or_equal);
+    REGISTER_OPERATOR("GreaterOrEqual", 1, greater_or_equal);
+    REGISTER_OPERATOR("GreaterOrEqual", 16, greater_or_equal);
     REGISTER_OPERATOR("GridSample", 1, grid_sample);
     REGISTER_OPERATOR("GroupNormalization", 1, group_normalization);
     REGISTER_OPERATOR("GRU", 1, gru);

diff --git a/src/frontends/onnx/tests/models/greater_or_equal_float.prototxt b/src/frontends/onnx/tests/models/greater_or_equal_float.prototxt
@@ -5,7 +5,7 @@ graph {
     input: "A"
     input: "B"
     output: "C"
-    op_type: "Greater_Or_Equal"
+    op_type: "GreaterOrEqual"
   }
   name: "test_greater_or_equal_float"
   input {

diff --git a/src/frontends/onnx/tests/models/greater_or_equal_int.prototxt b/src/frontends/onnx/tests/models/greater_or_equal_int.prototxt
@@ -5,7 +5,7 @@ graph {
     input: "A"
     input: "B"
     output: "C"
-    op_type: "Greater_Or_Equal"
+    op_type: "GreaterOrEqual"
   }
   name: "test_greater_or_equal_int"
   input {

diff --git a/src/frontends/onnx/tests/onnx_transformations.cpp b/src/frontends/onnx/tests/onnx_transformations.cpp
@@ -72,7 +72,9 @@ OPENVINO_TEST(onnx_transformations, expand_function_greater_or_equal) {
                                             "greater_or_equal_expanded.onnx"});
 
     const auto result = compare_onnx_models(editor.model_string(), ref_model, after_func_expand_name_comp);
-    EXPECT_TRUE(result.is_ok) << result.error_message;
+
+    // After operation translation was implemented - check it doesn't apply
+    EXPECT_FALSE(result.is_ok) << result.error_message;
 }
 
 // Disabled, ticket: #81976