Skip to content

Commit 5921ba7

Browse files
authored
Update test_forward.py
1 parent 9e885c1 commit 5921ba7

File tree

1 file changed

+9
-19
lines changed

1 file changed

+9
-19
lines changed

tests/python/frontend/tflite/test_forward.py

Lines changed: 9 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -2847,25 +2847,15 @@ def _test_forward_elemwise(testop):
28472847
def _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

28712861
def _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

Comments
 (0)