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