From 4f4846bd3dfffa1b49b42011d01818d06e87cb99 Mon Sep 17 00:00:00 2001 From: Andrew Luo Date: Tue, 21 Sep 2021 12:13:15 -0700 Subject: [PATCH 1/2] recreate things --- 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", From e35f201769df736d33badb3890b322c63b087d82 Mon Sep 17 00:00:00 2001 From: Andrew Luo Date: Tue, 21 Sep 2021 13:43:58 -0700 Subject: [PATCH 2/2] jostle