@@ -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