[ONNX] LessOrEqual and GreaterOrEqual ops (#9066)

* recreate things * jostle
apache · Sep 22, 2021 · 26b2067 · 26b2067
1 parent cd15b79
commit 26b2067
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py
@@ -3986,7 +3986,9 @@ def _get_convert_map(opset):
         "Elu": Elu.get_converter(opset),
         "Exp": Renamer("exp"),
         "Greater": Renamer("greater"),
+        "GreaterOrEqual": Renamer("greater_equal"),
         "Less": Renamer("less"),
+        "LessOrEqual": Renamer("less_equal"),
         "Log": Renamer("log"),
         "Acos": Renamer("acos"),
         "Acosh": Renamer("acosh"),

diff --git a/tests/python/frontend/onnx/test_forward.py b/tests/python/frontend/onnx/test_forward.py
@@ -1998,8 +1998,12 @@ def verify_binary_ops(op, x, y, out_type="float32"):
     verify_binary_ops("Sum", x, z)
     verify_binary_ops("Greater", x, y, "bool")
     verify_binary_ops("Greater", x, z, "bool")
+    verify_binary_ops("GreaterOrEqual", x, y, "bool")
+    verify_binary_ops("GreaterOrEqual", x, z, "bool")
     verify_binary_ops("Less", x, y, "bool")
     verify_binary_ops("Less", x, z, "bool")
+    verify_binary_ops("LessOrEqual", x, y, "bool")
+    verify_binary_ops("LessOrEqual", x, z, "bool")
     verify_binary_ops("Equal", x, y, "bool")
     verify_binary_ops("Equal", x, z, "bool")
 
@@ -4877,11 +4881,7 @@ def verify_eyelike(indata):
     "test_dropout_default_mask",
     "test_dropout_default_mask_ratio",
     "test_dropout_default_ratio",
-    "test_greater_equal",
-    "test_greater_equal_bcast",
     "test_if_seq",
-    "test_less_equal",
-    "test_less_equal_bcast",
     "test_logsoftmax_axis_0_expanded",
     "test_logsoftmax_axis_1_expanded",
     "test_logsoftmax_axis_2_expanded",