From 0c03f3a75a13cf55bd7c89a877e1a76a7dd43f60 Mon Sep 17 00:00:00 2001 From: Tlopex <68688494+tlopex@users.noreply.github.com> Date: Tue, 19 Sep 2023 23:24:13 +0800 Subject: [PATCH 1/2] Update tflite.py --- python/tvm/relay/frontend/tflite.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/python/tvm/relay/frontend/tflite.py b/python/tvm/relay/frontend/tflite.py index 6728263c7f0e..cd073c33a2e1 100644 --- a/python/tvm/relay/frontend/tflite.py +++ b/python/tvm/relay/frontend/tflite.py @@ -1486,11 +1486,7 @@ def convert_less(self, op): def convert_less_equal(self, op): """Convert TFLite LESS_EQUAL""" - if self.is_quantized(op): - raise tvm.error.OpNotImplemented( - "TFlite quantized LESS_EQUAL operator is not supported yet." - ) - return self._convert_elemwise(_op.less_equal, op) + return self._convert_elemwise(_op.less_equal, op, self.is_quantized(op), comparison_op=True) def convert_equal(self, op): """Convert TFLite EQUAL""" From ae74fae566ccbef48f2cc8680c48ceb8b243ac61 Mon Sep 17 00:00:00 2001 From: Tlopex <68688494+tlopex@users.noreply.github.com> Date: Tue, 19 Sep 2023 23:31:26 +0800 Subject: [PATCH 2/2] add tests to quantized less_equal --- tests/python/frontend/tflite/test_forward.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/tests/python/frontend/tflite/test_forward.py b/tests/python/frontend/tflite/test_forward.py index ab89de6301b2..384816226fd0 100644 --- a/tests/python/frontend/tflite/test_forward.py +++ b/tests/python/frontend/tflite/test_forward.py @@ -2707,9 +2707,17 @@ def _test_less(data, fused_activation_function=None, quantized=False, qnn_op=Non # ---------- -def _test_less_equal(data): +def _test_less_equal(data, fused_activation_function=None, quantized=False, qnn_op=None): """One iteration of less_equal""" - return _test_elemwise(math_ops.less_equal, data) + return _test_elemwise( + math_ops.less_equal, + data, + fused_activation_function, + quantized, + qnn_op, + same_qnn_params=True, + comparison_op=True, + ) ####################################################################### @@ -2850,6 +2858,7 @@ def _test_elemwise_qnn_out_range(qnn_op): _test_less: (-150, 150), _test_floor_mod: (-150, 150), _test_not_equal: (-150, 150), + _test_less_equal: (-150, 150), } return qnn_out_range[qnn_op] @@ -2888,6 +2897,7 @@ def test_all_elemwise(): _test_forward_elemwise(_test_less) _test_forward_elemwise_quantized(_test_less) _test_forward_elemwise(_test_less_equal) + _test_forward_elemwise_quantized(_test_less_equal) _test_forward_elemwise(_test_equal) _test_forward_elemwise_quantized(_test_equal) _test_forward_elemwise(_test_not_equal)