From 26b206702559d72acb141378553873e44eb09394 Mon Sep 17 00:00:00 2001 From: AndrewZhaoLuo Date: Tue, 21 Sep 2021 22:15:20 -0700 Subject: [PATCH] [ONNX] LessOrEqual and GreaterOrEqual ops (#9066) * recreate things * jostle --- python/tvm/relay/frontend/onnx.py | 2 ++ tests/python/frontend/onnx/test_forward.py | 8 ++++---- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py index 4d48f5796aca..5b5076a4997c 100644 --- 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 index 91d3911da530..abae6dd30de6 100644 --- 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",