@@ -2847,25 +2847,15 @@ def _test_forward_elemwise(testop):
28472847def _test_forward_elemwise_quantized (testop , dtype = np .uint8 ):
28482848 type_info = np .iinfo (dtype )
28492849 _min , _max = type_info .min , type_info .max
2850- if testop .__name__ == _test_pow .__name__ :
2851- testop (
2852- [
2853- np .array (np .random .uniform (- 3 , 3 , (3 , 6 )), dtype = dtype ),
2854- np .array (np .random .uniform (- 3 , 3 , (3 , 6 )), dtype = dtype ),
2855- ],
2856- quantized = True ,
2857- qnn_op = testop ,
2858- )
28592850 # np.uint8 0-255
2860- else :
2861- testop (
2862- [
2863- np .array (np .random .uniform (_min , _max , (3 , 6 )), dtype = dtype ),
2864- np .array (np .random .uniform (_min , _max , (3 , 6 )), dtype = dtype ),
2865- ],
2866- quantized = True ,
2867- qnn_op = testop ,
2868- )
2851+ testop (
2852+ [
2853+ np .array (np .random .uniform (_min , _max , (3 , 6 )), dtype = dtype ),
2854+ np .array (np .random .uniform (_min , _max , (3 , 6 )), dtype = dtype ),
2855+ ],
2856+ quantized = True ,
2857+ qnn_op = testop ,
2858+ )
28692859
28702860
28712861def _test_elemwise_qnn_out_range (qnn_op ):
@@ -2884,7 +2874,7 @@ def _test_elemwise_qnn_out_range(qnn_op):
28842874 _test_less : (- 150 , 150 ),
28852875 _test_floor_mod : (- 150 , 150 ),
28862876 _test_not_equal : (- 150 , 150 ),
2887- _test_pow : (- 2 , 2 ),
2877+ _test_pow : (- 26 , 26 ),
28882878 _test_less_equal : (- 150 , 150 ),
28892879 _test_greater_equal : (- 150 , 150 ),
28902880 }
0 commit comments