Skip to content

Commit 41b15af

Browse files
authored
Merge branch 'main' into pow
2 parents f605a93 + ab3511a commit 41b15af

File tree

2 files changed

+19
-9
lines changed

2 files changed

+19
-9
lines changed

python/tvm/relay/frontend/tflite.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1482,11 +1482,7 @@ def convert_less(self, op):
14821482

14831483
def convert_less_equal(self, op):
14841484
"""Convert TFLite LESS_EQUAL"""
1485-
if self.is_quantized(op):
1486-
raise tvm.error.OpNotImplemented(
1487-
"TFlite quantized LESS_EQUAL operator is not supported yet."
1488-
)
1489-
return self._convert_elemwise(_op.less_equal, op)
1485+
return self._convert_elemwise(_op.less_equal, op, self.is_quantized(op), comparison_op=True)
14901486

14911487
def convert_equal(self, op):
14921488
"""Convert TFLite EQUAL"""

tests/python/frontend/tflite/test_forward.py

Lines changed: 18 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2722,9 +2722,17 @@ def _test_less(data, fused_activation_function=None, quantized=False, qnn_op=Non
27222722
# ----------
27232723

27242724

2725-
def _test_less_equal(data):
2725+
def _test_less_equal(data, fused_activation_function=None, quantized=False, qnn_op=None):
27262726
"""One iteration of less_equal"""
2727-
return _test_elemwise(math_ops.less_equal, data)
2727+
return _test_elemwise(
2728+
math_ops.less_equal,
2729+
data,
2730+
fused_activation_function,
2731+
quantized,
2732+
qnn_op,
2733+
same_qnn_params=True,
2734+
comparison_op=True,
2735+
)
27282736

27292737

27302738
#######################################################################
@@ -2865,7 +2873,8 @@ def _test_elemwise_qnn_out_range(qnn_op):
28652873
_test_less: (-150, 150),
28662874
_test_floor_mod: (-150, 150),
28672875
_test_not_equal: (-150, 150),
2868-
_test_pow: (-5e3, 5e3),
2876+
_test_pow: (-8, 8),
2877+
_test_less_equal: (-150, 150),
28692878
_test_greater_equal: (-150, 150),
28702879
}
28712880

@@ -2907,6 +2916,7 @@ def test_all_elemwise():
29072916
_test_forward_elemwise(_test_less)
29082917
_test_forward_elemwise_quantized(_test_less)
29092918
_test_forward_elemwise(_test_less_equal)
2919+
_test_forward_elemwise_quantized(_test_less_equal)
29102920
_test_forward_elemwise(_test_equal)
29112921
_test_forward_elemwise_quantized(_test_equal)
29122922
_test_forward_elemwise(_test_not_equal)
@@ -2915,7 +2925,11 @@ def test_all_elemwise():
29152925
_test_forward_elemwise(_test_floor_divide)
29162926
_test_forward_elemwise_quantized(_test_floor_divide)
29172927
_test_forward_elemwise(_test_floor_mod)
2918-
_test_forward_elemwise_quantized(_test_floor_mod)
2928+
# This test of quantized floor mod is currently disabled due
2929+
# to flaky CI failures in main, failing approximately 45% of
2930+
# the time.
2931+
#
2932+
# _test_forward_elemwise_quantized(_test_floor_mod)
29192933

29202934

29212935
#######################################################################

0 commit comments

Comments
 (0)