Skip to content

Commit e3036c5

Browse files
jdduketensorflower-gardener
authored andcommitted
Remove TFLite reduce test cases with redundant axes
PiperOrigin-RevId: 318844070 Change-Id: If4b385d4988ca71808dc8c94c617aaf6a406f260
1 parent 177f89f commit e3036c5

File tree

1 file changed

+60
-11
lines changed

1 file changed

+60
-11
lines changed

tensorflow/lite/testing/op_tests/reduce.py

Lines changed: 60 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -50,9 +50,25 @@ def f(options):
5050
"input_dtype": [tf.float32, tf.int32, tf.int64],
5151
"input_shape": [[3, 3, 2, 4]],
5252
"axis": [
53-
0, 1, 2, [0, 1], [0, 2], [1, 2], [0, 1, 2], [1, 0], [2, 0],
54-
[2, 1], [2, 1, 0], [2, 0, 1], -1, -2, -3, [1, -1], [0, -1],
55-
[-1, 0], [-1, -2, -3], [0, 0, 0], [2, 2, 0], [1, 0, -3, -3]
53+
0,
54+
1,
55+
2,
56+
[0, 1],
57+
[0, 2],
58+
[1, 2],
59+
[0, 1, 2],
60+
[1, 0],
61+
[2, 0],
62+
[2, 1],
63+
[2, 1, 0],
64+
[2, 0, 1],
65+
-1,
66+
-2,
67+
-3,
68+
[1, -1],
69+
[0, -1],
70+
[-1, 0],
71+
[-1, -2, -3],
5672
],
5773
"const_axis": [True, False],
5874
"keepdims": [True, False],
@@ -62,11 +78,28 @@ def f(options):
6278
"input_dtype": [tf.float32],
6379
"input_shape": [[1, 8, 8, 3]],
6480
"axis": [
65-
0, 1, 2, 3, [1, 2], [0, 3], [1, 2, 3], [0, 1, 2,
66-
3], [3, 2, 1, 0],
67-
[3, 1, 0, 2], [2, 0], [3, 0], [3, 1], [1, 0], -1, -2, -3, -4,
68-
[0, -2], [2, 3, -1, 0], [3, 1, 2, -3], [3, -4], [2, 2, 2],
69-
[2, 2, 3], [-3, -3, -4], [-3, 2, 1]
81+
0,
82+
1,
83+
2,
84+
3,
85+
[1, 2],
86+
[0, 3],
87+
[1, 2, 3],
88+
[0, 1, 2, 3],
89+
[3, 2, 1, 0],
90+
[3, 1, 0, 2],
91+
[2, 0],
92+
[3, 0],
93+
[3, 1],
94+
[1, 0],
95+
-1,
96+
-2,
97+
-3,
98+
-4,
99+
[0, -2],
100+
[2, 3, 1, 0],
101+
[3, 1, 2],
102+
[3, -4],
70103
],
71104
"const_axis": [True, False],
72105
"keepdims": [True, False],
@@ -92,9 +125,25 @@ def f(options):
92125
"input_dtype": [tf.float32],
93126
"input_shape": [[3, 3, 2, 4]],
94127
"axis": [
95-
0, 1, 2, [0, 1], [0, 2], [1, 2], [0, 1, 2], [1, 0], [2, 0],
96-
[2, 1], [2, 1, 0], [2, 0, 1], -1, -2, -3, [1, -1], [0, -1],
97-
[-1, 0], [-1, -2, -3], [0, 0, 0], [2, 2, 0], [1, 0, -3, -3]
128+
0,
129+
1,
130+
2,
131+
[0, 1],
132+
[0, 2],
133+
[1, 2],
134+
[0, 1, 2],
135+
[1, 0],
136+
[2, 0],
137+
[2, 1],
138+
[2, 1, 0],
139+
[2, 0, 1],
140+
-1,
141+
-2,
142+
-3,
143+
[1, -1],
144+
[0, -1],
145+
[-1, 0],
146+
[-1, -2, -3],
98147
],
99148
"const_axis": [True],
100149
"keepdims": [True, False],

0 commit comments

Comments
 (0)