xref: /aosp_15_r20/external/XNNPACK/test/transpose-normalization.cc (revision 4bdc94577ba0e567308109d787f7fec7b531ce36)
1*4bdc9457SAndroid Build Coastguard Worker #include <gtest/gtest.h>
2*4bdc9457SAndroid Build Coastguard Worker 
3*4bdc9457SAndroid Build Coastguard Worker #include <xnnpack.h>
4*4bdc9457SAndroid Build Coastguard Worker #include <xnnpack/normalization.h>
5*4bdc9457SAndroid Build Coastguard Worker 
6*4bdc9457SAndroid Build Coastguard Worker #include "transpose-normalization-tester.h"
7*4bdc9457SAndroid Build Coastguard Worker 
TEST(TRANSPOSE_NORMALIZATION_TEST,input_stride_fold_0_1)8*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, input_stride_fold_0_1) {
9*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
10*4bdc9457SAndroid Build Coastguard Worker       .num_dims(4)
11*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
12*4bdc9457SAndroid Build Coastguard Worker       .perm({0,1,3,2})
13*4bdc9457SAndroid Build Coastguard Worker       .shape({5,4,3,2})
14*4bdc9457SAndroid Build Coastguard Worker       .input_stride({24,6,2,1})
15*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({20,3,2})
16*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({0,2,1})
17*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(3)
18*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(4)
19*4bdc9457SAndroid Build Coastguard Worker       .expected_input_stride({24,8,4})
20*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
21*4bdc9457SAndroid Build Coastguard Worker       .Test();
22*4bdc9457SAndroid Build Coastguard Worker }
23*4bdc9457SAndroid Build Coastguard Worker 
TEST(TRANSPOSE_NORMALIZATION_TEST,output_stride_fold_0_1)24*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, output_stride_fold_0_1) {
25*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
26*4bdc9457SAndroid Build Coastguard Worker       .num_dims(4)
27*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
28*4bdc9457SAndroid Build Coastguard Worker       .perm({0,1,3,2})
29*4bdc9457SAndroid Build Coastguard Worker       .shape({5,4,3,2})
30*4bdc9457SAndroid Build Coastguard Worker       .output_stride({24,6,3,1})
31*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({20,3,2})
32*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({0,2,1})
33*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(3)
34*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(4)
35*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
36*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
37*4bdc9457SAndroid Build Coastguard Worker       .Test();
38*4bdc9457SAndroid Build Coastguard Worker }
39*4bdc9457SAndroid Build Coastguard Worker 
TEST(TRANSPOSE_NORMALIZATION_TEST,output_stride)40*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, output_stride) {
41*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
42*4bdc9457SAndroid Build Coastguard Worker       .num_dims(6)
43*4bdc9457SAndroid Build Coastguard Worker       .element_size(1)
44*4bdc9457SAndroid Build Coastguard Worker       .perm({1,0,3,2,5,4})
45*4bdc9457SAndroid Build Coastguard Worker       .shape({5,4,3,2,6,7})
46*4bdc9457SAndroid Build Coastguard Worker       .output_stride({1260,252,126,42,6,1})
47*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({5,4,3,2,6,7})
48*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({1,0,3,2,5,4})
49*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(6)
50*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(1)
51*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
52*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
53*4bdc9457SAndroid Build Coastguard Worker       .Test();
54*4bdc9457SAndroid Build Coastguard Worker }
55*4bdc9457SAndroid Build Coastguard Worker 
TEST(TRANSPOSE_NORMALIZATION_TEST,output_stride_fold_2_3)56*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, output_stride_fold_2_3) {
57*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
58*4bdc9457SAndroid Build Coastguard Worker       .num_dims(6)
59*4bdc9457SAndroid Build Coastguard Worker       .element_size(1)
60*4bdc9457SAndroid Build Coastguard Worker       .perm({1,0,2,3,5,4})
61*4bdc9457SAndroid Build Coastguard Worker       .shape({5,4,3,2,6,7})
62*4bdc9457SAndroid Build Coastguard Worker       .output_stride({1260,252,84,42,6,1})
63*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({5,4,6,6,7})
64*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({1,0,2,4,3})
65*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(5)
66*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(1)
67*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
68*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
69*4bdc9457SAndroid Build Coastguard Worker       .Test();
70*4bdc9457SAndroid Build Coastguard Worker }
71*4bdc9457SAndroid Build Coastguard Worker 
TEST(TRANSPOSE_NORMALIZATION_TEST,output_stride_no_fold_2_3)72*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, output_stride_no_fold_2_3) {
73*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
74*4bdc9457SAndroid Build Coastguard Worker       .num_dims(6)
75*4bdc9457SAndroid Build Coastguard Worker       .element_size(1)
76*4bdc9457SAndroid Build Coastguard Worker       .perm({1,0,2,3,5,4})
77*4bdc9457SAndroid Build Coastguard Worker       .shape({5,4,3,2,6,7})
78*4bdc9457SAndroid Build Coastguard Worker       .output_stride({1275,255,85,42,6,1})
79*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({5,4,3,2,6,7})
80*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({1,0,2,3,5,4})
81*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(6)
82*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(1)
83*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
84*4bdc9457SAndroid Build Coastguard Worker       .expected_output_stride({1275,255,85,42,6,1})
85*4bdc9457SAndroid Build Coastguard Worker       .Test();
86*4bdc9457SAndroid Build Coastguard Worker }
87*4bdc9457SAndroid Build Coastguard Worker 
TEST(TRANSPOSE_NORMALIZATION_TEST,output_stride_fold_2_3_with_large_strides)88*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, output_stride_fold_2_3_with_large_strides) {
89*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
90*4bdc9457SAndroid Build Coastguard Worker       .num_dims(6)
91*4bdc9457SAndroid Build Coastguard Worker       .element_size(1)
92*4bdc9457SAndroid Build Coastguard Worker       .perm({1,0,2,3,5,4})
93*4bdc9457SAndroid Build Coastguard Worker       .shape({5,4,3,2,6,7})
94*4bdc9457SAndroid Build Coastguard Worker       .output_stride({1290,258,86,43,6,1})
95*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({5,4,6,6,7})
96*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({1,0,2,4,3})
97*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(5)
98*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(1)
99*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
100*4bdc9457SAndroid Build Coastguard Worker       .expected_output_stride({1290,258,43,6,1})
101*4bdc9457SAndroid Build Coastguard Worker       .Test();
102*4bdc9457SAndroid Build Coastguard Worker }
103*4bdc9457SAndroid Build Coastguard Worker 
TEST(TRANSPOSE_NORMALIZATION_TEST,output_stride_fold_last_dim)104*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, output_stride_fold_last_dim) {
105*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
106*4bdc9457SAndroid Build Coastguard Worker       .num_dims(6)
107*4bdc9457SAndroid Build Coastguard Worker       .element_size(1)
108*4bdc9457SAndroid Build Coastguard Worker       .perm({1,0,2,4,3,5})
109*4bdc9457SAndroid Build Coastguard Worker       .shape({5,4,3,2,6,7})
110*4bdc9457SAndroid Build Coastguard Worker       .output_stride({1260,252,84,14,7,1})
111*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({5,4,3,2,6})
112*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({1,0,2,4,3})
113*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(5)
114*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(7)
115*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
116*4bdc9457SAndroid Build Coastguard Worker       .expected_output_stride({1260,252,84,14,7})
117*4bdc9457SAndroid Build Coastguard Worker       .Test();
118*4bdc9457SAndroid Build Coastguard Worker }
119*4bdc9457SAndroid Build Coastguard Worker 
TEST(TRANSPOSE_NORMALIZATION_TEST,output_stride_remove_dim_size_1)120*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, output_stride_remove_dim_size_1) {
121*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
122*4bdc9457SAndroid Build Coastguard Worker       .num_dims(6)
123*4bdc9457SAndroid Build Coastguard Worker       .element_size(1)
124*4bdc9457SAndroid Build Coastguard Worker       .perm({1,0,3,2,5,4})
125*4bdc9457SAndroid Build Coastguard Worker       .shape({5,4,1,2,6,7})
126*4bdc9457SAndroid Build Coastguard Worker       .output_stride({420,84,42,42,6,1})
127*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({5,4,2,6,7})
128*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({1,0,2,4,3})
129*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(5)
130*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(1)
131*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
132*4bdc9457SAndroid Build Coastguard Worker       .expected_output_stride({420,84,42,6,1})
133*4bdc9457SAndroid Build Coastguard Worker       .Test();
134*4bdc9457SAndroid Build Coastguard Worker }
135*4bdc9457SAndroid Build Coastguard Worker 
TEST(TRANSPOSE_NORMALIZATION_TEST,output_stride_no_remove_dim_size_1)136*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, output_stride_no_remove_dim_size_1) {
137*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
138*4bdc9457SAndroid Build Coastguard Worker       .num_dims(6)
139*4bdc9457SAndroid Build Coastguard Worker       .element_size(1)
140*4bdc9457SAndroid Build Coastguard Worker       .perm({1,0,3,2,5,4})
141*4bdc9457SAndroid Build Coastguard Worker       .shape({5,4,1,2,6,7})
142*4bdc9457SAndroid Build Coastguard Worker       .output_stride({420,86,43,42,6,1})
143*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({5,4,1,2,6,7})
144*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({1,0,3,2,5,4})
145*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(6)
146*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(1)
147*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
148*4bdc9457SAndroid Build Coastguard Worker       .expected_output_stride({420,86,43,42,6,1})
149*4bdc9457SAndroid Build Coastguard Worker       .Test();
150*4bdc9457SAndroid Build Coastguard Worker }
151*4bdc9457SAndroid Build Coastguard Worker 
TEST(TRANSPOSE_NORMALIZATION_TEST,output_stride_remove_dim_size_1_4bytes)152*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, output_stride_remove_dim_size_1_4bytes) {
153*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
154*4bdc9457SAndroid Build Coastguard Worker       .num_dims(6)
155*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
156*4bdc9457SAndroid Build Coastguard Worker       .perm({1,0,3,2,5,4})
157*4bdc9457SAndroid Build Coastguard Worker       .shape({5,4,1,2,6,7})
158*4bdc9457SAndroid Build Coastguard Worker       .output_stride({420,84,42,42,6,1})
159*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({5,4,2,6,7})
160*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({1,0,2,4,3})
161*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(5)
162*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(4)
163*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
164*4bdc9457SAndroid Build Coastguard Worker       .expected_output_stride({1680,336,168,24,4})
165*4bdc9457SAndroid Build Coastguard Worker       .Test();
166*4bdc9457SAndroid Build Coastguard Worker }
167*4bdc9457SAndroid Build Coastguard Worker 
TEST(TRANSPOSE_NORMALIZATION_TEST,input_output_stride_remove_dim_size_1)168*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, input_output_stride_remove_dim_size_1) {
169*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
170*4bdc9457SAndroid Build Coastguard Worker       .num_dims(6)
171*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
172*4bdc9457SAndroid Build Coastguard Worker       .perm({1,0,3,2,5,4})
173*4bdc9457SAndroid Build Coastguard Worker       .shape({5,4,1,2,6,7})
174*4bdc9457SAndroid Build Coastguard Worker       .input_stride({336,84,84,42,7,1})
175*4bdc9457SAndroid Build Coastguard Worker       .output_stride({420,84,42,42,6,1})
176*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({5,4,2,6,7})
177*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({1,0,2,4,3})
178*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(5)
179*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(4)
180*4bdc9457SAndroid Build Coastguard Worker       .expected_input_stride({1344,336,168,28,4})
181*4bdc9457SAndroid Build Coastguard Worker       .expected_output_stride({1680,336,168,24,4})
182*4bdc9457SAndroid Build Coastguard Worker       .Test();
183*4bdc9457SAndroid Build Coastguard Worker }
184*4bdc9457SAndroid Build Coastguard Worker 
TEST(TRANSPOSE_NORMALIZATION_TEST,input_output_stride_remove_fold_large_element)185*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, input_output_stride_remove_fold_large_element) {
186*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
187*4bdc9457SAndroid Build Coastguard Worker       .num_dims(6)
188*4bdc9457SAndroid Build Coastguard Worker       .element_size(1)
189*4bdc9457SAndroid Build Coastguard Worker       .perm({1,0,3,2,5,4})
190*4bdc9457SAndroid Build Coastguard Worker       .shape({5,4,1,2,1,7})
191*4bdc9457SAndroid Build Coastguard Worker       .input_stride({56,14,14,7,7,1})
192*4bdc9457SAndroid Build Coastguard Worker       .output_stride({70,14,7,7,1,1})
193*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({5,4})
194*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({1,0})
195*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(2)
196*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(14)
197*4bdc9457SAndroid Build Coastguard Worker       .expected_input_stride({56,14})
198*4bdc9457SAndroid Build Coastguard Worker       .expected_output_stride({70,14})
199*4bdc9457SAndroid Build Coastguard Worker       .Test();
200*4bdc9457SAndroid Build Coastguard Worker }
201*4bdc9457SAndroid Build Coastguard Worker 
TEST(TRANSPOSE_NORMALIZATION_TEST,input_output_stride_no_remove_dim_size_1)202*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, input_output_stride_no_remove_dim_size_1) {
203*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
204*4bdc9457SAndroid Build Coastguard Worker       .num_dims(6)
205*4bdc9457SAndroid Build Coastguard Worker       .element_size(1)
206*4bdc9457SAndroid Build Coastguard Worker       .perm({1,0,3,2,5,4})
207*4bdc9457SAndroid Build Coastguard Worker       .shape({5,4,1,2,1,7})
208*4bdc9457SAndroid Build Coastguard Worker       .input_stride({70,15,14,7,7,1})
209*4bdc9457SAndroid Build Coastguard Worker       .output_stride({140,28,14,14,2,1})
210*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({5,4,1,2,1,7})
211*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({1,0,3,2,5,4})
212*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(6)
213*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(1)
214*4bdc9457SAndroid Build Coastguard Worker       .expected_input_stride({70,15,14,7,7,1})
215*4bdc9457SAndroid Build Coastguard Worker       .expected_output_stride({140,28,14,14,2,1})
216*4bdc9457SAndroid Build Coastguard Worker       .Test();
217*4bdc9457SAndroid Build Coastguard Worker }
218*4bdc9457SAndroid Build Coastguard Worker 
TEST(TRANSPOSE_NORMALIZATION_TEST,input_output_stride_no_remove_dim_1_no_fold)219*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, input_output_stride_no_remove_dim_1_no_fold) {
220*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
221*4bdc9457SAndroid Build Coastguard Worker       .num_dims(6)
222*4bdc9457SAndroid Build Coastguard Worker       .element_size(1)
223*4bdc9457SAndroid Build Coastguard Worker       .perm({5,4,1,2,0,3})
224*4bdc9457SAndroid Build Coastguard Worker       .shape({4,9,7,2,1,6})
225*4bdc9457SAndroid Build Coastguard Worker       .input_stride({882,98,14,7,6,1})
226*4bdc9457SAndroid Build Coastguard Worker       .output_stride({505,505,56,8,2,1})
227*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({4,9,7,2,1,6})
228*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({5,4,1,2,0,3})
229*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(6)
230*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(1)
231*4bdc9457SAndroid Build Coastguard Worker       .expected_input_stride({882,98,14,7,6,1})
232*4bdc9457SAndroid Build Coastguard Worker       .expected_output_stride({505,505,56,8,2,1})
233*4bdc9457SAndroid Build Coastguard Worker       .Test();
234*4bdc9457SAndroid Build Coastguard Worker }
235*4bdc9457SAndroid Build Coastguard Worker 
TEST(TRANSPOSE_NORMALIZATION_TEST,input_stride)236*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, input_stride) {
237*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
238*4bdc9457SAndroid Build Coastguard Worker       .num_dims(4)
239*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
240*4bdc9457SAndroid Build Coastguard Worker       .perm({3,1,2,0})
241*4bdc9457SAndroid Build Coastguard Worker       .shape({5,4,3,2})
242*4bdc9457SAndroid Build Coastguard Worker       .input_stride({24,6,2,1})
243*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({5,12,2})
244*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({2,1,0})
245*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(3)
246*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(4)
247*4bdc9457SAndroid Build Coastguard Worker       .expected_input_stride({96,8,4})
248*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
249*4bdc9457SAndroid Build Coastguard Worker       .Test();
250*4bdc9457SAndroid Build Coastguard Worker }
251*4bdc9457SAndroid Build Coastguard Worker 
TEST(TRANSPOSE_NORMALIZATION_TEST,input_stride_remove_dim_1)252*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, input_stride_remove_dim_1) {
253*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
254*4bdc9457SAndroid Build Coastguard Worker       .num_dims(4)
255*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
256*4bdc9457SAndroid Build Coastguard Worker       .perm({3,2,1,0})
257*4bdc9457SAndroid Build Coastguard Worker       .shape({5,4,1,2})
258*4bdc9457SAndroid Build Coastguard Worker       .input_stride({8,2,2,1})
259*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({5,4,2})
260*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({2,1,0})
261*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(3)
262*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(4)
263*4bdc9457SAndroid Build Coastguard Worker       .expected_input_stride({32,8,4})
264*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
265*4bdc9457SAndroid Build Coastguard Worker       .Test();
266*4bdc9457SAndroid Build Coastguard Worker }
267*4bdc9457SAndroid Build Coastguard Worker 
TEST(TRANSPOSE_NORMALIZATION_TEST,input_stride_elem_size_1)268*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, input_stride_elem_size_1) {
269*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
270*4bdc9457SAndroid Build Coastguard Worker       .num_dims(4)
271*4bdc9457SAndroid Build Coastguard Worker       .element_size(1)
272*4bdc9457SAndroid Build Coastguard Worker       .perm({3,2,1,0})
273*4bdc9457SAndroid Build Coastguard Worker       .shape({5,4,2,2})
274*4bdc9457SAndroid Build Coastguard Worker       .input_stride({16,4,2,1})
275*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({5,4,2,2})
276*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({3,2,1,0})
277*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(4)
278*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(1)
279*4bdc9457SAndroid Build Coastguard Worker       .expected_input_stride({16,4,2,1})
280*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
281*4bdc9457SAndroid Build Coastguard Worker       .Test();
282*4bdc9457SAndroid Build Coastguard Worker }
283*4bdc9457SAndroid Build Coastguard Worker 
TEST(TRANSPOSE_NORMALIZATION_TEST,input_stride_strided)284*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, input_stride_strided) {
285*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
286*4bdc9457SAndroid Build Coastguard Worker       .num_dims(5)
287*4bdc9457SAndroid Build Coastguard Worker       .element_size(1)
288*4bdc9457SAndroid Build Coastguard Worker       .perm({3,1,2,4,0})
289*4bdc9457SAndroid Build Coastguard Worker       .shape({5,4,2,2,3})
290*4bdc9457SAndroid Build Coastguard Worker       .input_stride({96,24,6,3,1})
291*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({5,4,2,2,3})
292*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({3,1,2,4,0})
293*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(5)
294*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(1)
295*4bdc9457SAndroid Build Coastguard Worker       .expected_input_stride({96,24,6,3,1})
296*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
297*4bdc9457SAndroid Build Coastguard Worker       .Test();
298*4bdc9457SAndroid Build Coastguard Worker }
299*4bdc9457SAndroid Build Coastguard Worker 
TEST(TRANSPOSE_NORMALIZATION_TEST,input_stride_strided_size_1_dims)300*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, input_stride_strided_size_1_dims) {
301*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
302*4bdc9457SAndroid Build Coastguard Worker       .num_dims(6)
303*4bdc9457SAndroid Build Coastguard Worker       .element_size(1)
304*4bdc9457SAndroid Build Coastguard Worker       .perm({2,1,3,5,4,0})
305*4bdc9457SAndroid Build Coastguard Worker       .shape({5,4,2,1,1,7})
306*4bdc9457SAndroid Build Coastguard Worker       .input_stride({112,28,14,14,7,1})
307*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({5,4,2,1,7})
308*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({2,1,4,3,0})
309*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(5)
310*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(1)
311*4bdc9457SAndroid Build Coastguard Worker       .expected_input_stride({112,28,14,7,1})
312*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
313*4bdc9457SAndroid Build Coastguard Worker       .Test();
314*4bdc9457SAndroid Build Coastguard Worker }
315*4bdc9457SAndroid Build Coastguard Worker 
TEST(TRANSPOSE_NORMALIZATION_TEST,input_stride_size_1_dims_flatten)316*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, input_stride_size_1_dims_flatten) {
317*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
318*4bdc9457SAndroid Build Coastguard Worker       .num_dims(6)
319*4bdc9457SAndroid Build Coastguard Worker       .element_size(1)
320*4bdc9457SAndroid Build Coastguard Worker       .perm({3,4,5,2,1,0})
321*4bdc9457SAndroid Build Coastguard Worker       .shape({1,4,1,3,5,7})
322*4bdc9457SAndroid Build Coastguard Worker       .input_stride({420,105,105,35,7,1})
323*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({4,105})
324*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({1,0})
325*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(2)
326*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(1)
327*4bdc9457SAndroid Build Coastguard Worker       .expected_input_stride({105,1})
328*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
329*4bdc9457SAndroid Build Coastguard Worker       .Test();
330*4bdc9457SAndroid Build Coastguard Worker }
331*4bdc9457SAndroid Build Coastguard Worker 
TEST(TRANSPOSE_NORMALIZATION_TEST,input_stride_size_1_dims_flatten_strided)332*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, input_stride_size_1_dims_flatten_strided) {
333*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
334*4bdc9457SAndroid Build Coastguard Worker       .num_dims(6)
335*4bdc9457SAndroid Build Coastguard Worker       .element_size(1)
336*4bdc9457SAndroid Build Coastguard Worker       .perm({3,4,5,2,1,0})
337*4bdc9457SAndroid Build Coastguard Worker       .shape({1,4,1,3,5,7})
338*4bdc9457SAndroid Build Coastguard Worker       .input_stride({660,165,165,55,11,1})
339*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({4,15,7})
340*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({1,2,0})
341*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(3)
342*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(1)
343*4bdc9457SAndroid Build Coastguard Worker       .expected_input_stride({165,11,1})
344*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
345*4bdc9457SAndroid Build Coastguard Worker       .Test();
346*4bdc9457SAndroid Build Coastguard Worker }
347*4bdc9457SAndroid Build Coastguard Worker 
TEST(TRANSPOSE_NORMALIZATION_TEST,input_stride_size_1_dims_flatten_strided_copy)348*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, input_stride_size_1_dims_flatten_strided_copy) {
349*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
350*4bdc9457SAndroid Build Coastguard Worker       .num_dims(6)
351*4bdc9457SAndroid Build Coastguard Worker       .element_size(1)
352*4bdc9457SAndroid Build Coastguard Worker       .perm({1,0,3,4,2,5})
353*4bdc9457SAndroid Build Coastguard Worker       .shape({1,4,1,3,5,7})
354*4bdc9457SAndroid Build Coastguard Worker       .input_stride({668,167,115,35,7,1})
355*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({4,1,15})
356*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({0,2,1})
357*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(3)
358*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(7)
359*4bdc9457SAndroid Build Coastguard Worker       .expected_input_stride({167,115,7})
360*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
361*4bdc9457SAndroid Build Coastguard Worker       .Test();
362*4bdc9457SAndroid Build Coastguard Worker }
363*4bdc9457SAndroid Build Coastguard Worker 
TEST(TRANSPOSE_NORMALIZATION_TEST,input_stride_size_1_dims_flatten_last_dim_strided_copy)364*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, input_stride_size_1_dims_flatten_last_dim_strided_copy) {
365*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
366*4bdc9457SAndroid Build Coastguard Worker       .num_dims(6)
367*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
368*4bdc9457SAndroid Build Coastguard Worker       .perm({0,1,3,2,4,5})
369*4bdc9457SAndroid Build Coastguard Worker       .shape({2,4,1,3,5,7})
370*4bdc9457SAndroid Build Coastguard Worker       .input_stride({700,160,150,36,7,1})
371*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({2,4,1,3,1})
372*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({0,1,3,2,4})
373*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(5)
374*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(140)
375*4bdc9457SAndroid Build Coastguard Worker       .expected_input_stride({2800,640,600,144,140})
376*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
377*4bdc9457SAndroid Build Coastguard Worker       .Test();
378*4bdc9457SAndroid Build Coastguard Worker }
379*4bdc9457SAndroid Build Coastguard Worker 
TEST(TRANSPOSE_NORMALIZATION_TEST,input_output_stride_size_1_dims_flatten_last_dim_strided_copy)380*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, input_output_stride_size_1_dims_flatten_last_dim_strided_copy) {
381*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
382*4bdc9457SAndroid Build Coastguard Worker       .num_dims(6)
383*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
384*4bdc9457SAndroid Build Coastguard Worker       .perm({0,1,3,2,4,5})
385*4bdc9457SAndroid Build Coastguard Worker       .shape({2,4,1,3,5,7})
386*4bdc9457SAndroid Build Coastguard Worker       .input_stride({700,160,150,36,7,1})
387*4bdc9457SAndroid Build Coastguard Worker       .output_stride({420,105,35,35,7,1})
388*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({2,4,1,3,1})
389*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({0,1,3,2,4})
390*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(5)
391*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(140)
392*4bdc9457SAndroid Build Coastguard Worker       .expected_input_stride({2800,640,600,144,140})
393*4bdc9457SAndroid Build Coastguard Worker       .expected_output_stride({1680,420,140,140,140})
394*4bdc9457SAndroid Build Coastguard Worker       .Test();
395*4bdc9457SAndroid Build Coastguard Worker }
396*4bdc9457SAndroid Build Coastguard Worker 
TEST(TRANSPOSE_NORMALIZATION_TEST,input_output_stride_flatten_last_dim_strided_copy)397*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, input_output_stride_flatten_last_dim_strided_copy) {
398*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
399*4bdc9457SAndroid Build Coastguard Worker       .num_dims(5)
400*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
401*4bdc9457SAndroid Build Coastguard Worker       .perm({0,3,2,1,4})
402*4bdc9457SAndroid Build Coastguard Worker       .shape({4,2,3,5,7})
403*4bdc9457SAndroid Build Coastguard Worker       .output_stride({240,48,16,8,1})
404*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({4,2,3,5,1})
405*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({0,3,2,1,4})
406*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(5)
407*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(28)
408*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
409*4bdc9457SAndroid Build Coastguard Worker       .expected_output_stride({960,192,64,32,28})
410*4bdc9457SAndroid Build Coastguard Worker       .Test();
411*4bdc9457SAndroid Build Coastguard Worker }
412*4bdc9457SAndroid Build Coastguard Worker 
TEST(TRANSPOSE_NORMALIZATION_TEST,adjacent_1_dims)413*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, adjacent_1_dims) {
414*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
415*4bdc9457SAndroid Build Coastguard Worker       .num_dims(4)
416*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
417*4bdc9457SAndroid Build Coastguard Worker       .perm({0,1,3,2})
418*4bdc9457SAndroid Build Coastguard Worker       .shape({1,1,60,2400})
419*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({60,2400})
420*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({1,0})
421*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(2)
422*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(4)
423*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
424*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
425*4bdc9457SAndroid Build Coastguard Worker       .Test();
426*4bdc9457SAndroid Build Coastguard Worker }
427*4bdc9457SAndroid Build Coastguard Worker 
428*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, 1D) {
429*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
430*4bdc9457SAndroid Build Coastguard Worker       .num_dims(1)
431*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
432*4bdc9457SAndroid Build Coastguard Worker       .perm({0})
433*4bdc9457SAndroid Build Coastguard Worker       .shape({37})
434*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({1})
435*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({0})
436*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(1)
437*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(37*4)
438*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
439*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
440*4bdc9457SAndroid Build Coastguard Worker       .Test();
441*4bdc9457SAndroid Build Coastguard Worker }
442*4bdc9457SAndroid Build Coastguard Worker 
443*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, 2D_flatten_1D) {
444*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
445*4bdc9457SAndroid Build Coastguard Worker       .num_dims(2)
446*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
447*4bdc9457SAndroid Build Coastguard Worker       .perm({0,1})
448*4bdc9457SAndroid Build Coastguard Worker       .shape({37,19})
449*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({1})
450*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({0})
451*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(1)
452*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(37*19*4)
453*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
454*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
455*4bdc9457SAndroid Build Coastguard Worker       .Test();
456*4bdc9457SAndroid Build Coastguard Worker }
457*4bdc9457SAndroid Build Coastguard Worker 
458*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, 2D_flatten_2D) {
459*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
460*4bdc9457SAndroid Build Coastguard Worker       .num_dims(2)
461*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
462*4bdc9457SAndroid Build Coastguard Worker       .perm({1,0})
463*4bdc9457SAndroid Build Coastguard Worker       .shape({23,17})
464*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({23,17})
465*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({1,0})
466*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(2)
467*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(4)
468*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
469*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
470*4bdc9457SAndroid Build Coastguard Worker       .Test();
471*4bdc9457SAndroid Build Coastguard Worker }
472*4bdc9457SAndroid Build Coastguard Worker 
473*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, 3D_redundant_dim) {
474*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
475*4bdc9457SAndroid Build Coastguard Worker       .num_dims(3)
476*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
477*4bdc9457SAndroid Build Coastguard Worker       .perm({0,2,1})
478*4bdc9457SAndroid Build Coastguard Worker       .shape({2,1,3})
479*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({1})
480*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({0})
481*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(1)
482*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(24)
483*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
484*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
485*4bdc9457SAndroid Build Coastguard Worker       .Test();
486*4bdc9457SAndroid Build Coastguard Worker }
487*4bdc9457SAndroid Build Coastguard Worker 
488*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, 3D_all_ones) {
489*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
490*4bdc9457SAndroid Build Coastguard Worker       .num_dims(3)
491*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
492*4bdc9457SAndroid Build Coastguard Worker       .perm({2,1,0})
493*4bdc9457SAndroid Build Coastguard Worker       .shape({1,1,1})
494*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({1})
495*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({0})
496*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(1)
497*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(4)
498*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
499*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
500*4bdc9457SAndroid Build Coastguard Worker       .Test();
501*4bdc9457SAndroid Build Coastguard Worker }
502*4bdc9457SAndroid Build Coastguard Worker 
503*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, 3D_flatten_1D) {
504*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
505*4bdc9457SAndroid Build Coastguard Worker       .num_dims(3)
506*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
507*4bdc9457SAndroid Build Coastguard Worker       .perm({0,1,2})
508*4bdc9457SAndroid Build Coastguard Worker       .shape({101,13,7})
509*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({1})
510*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({0})
511*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(1)
512*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(101*13*7*4)
513*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
514*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
515*4bdc9457SAndroid Build Coastguard Worker       .Test();
516*4bdc9457SAndroid Build Coastguard Worker }
517*4bdc9457SAndroid Build Coastguard Worker 
518*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, 3D_flatten_2D) {
519*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
520*4bdc9457SAndroid Build Coastguard Worker       .num_dims(3)
521*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
522*4bdc9457SAndroid Build Coastguard Worker       .perm({2,0,1})
523*4bdc9457SAndroid Build Coastguard Worker       .shape({101,13,7})
524*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({101*13,7})
525*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({1,0})
526*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(2)
527*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(4)
528*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
529*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
530*4bdc9457SAndroid Build Coastguard Worker       .Test();
531*4bdc9457SAndroid Build Coastguard Worker }
532*4bdc9457SAndroid Build Coastguard Worker 
533*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, 3D_flatten_element_size_2D) {
534*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
535*4bdc9457SAndroid Build Coastguard Worker       .num_dims(3)
536*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
537*4bdc9457SAndroid Build Coastguard Worker       .perm({1,0,2})
538*4bdc9457SAndroid Build Coastguard Worker       .shape({101,13,7})
539*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({101,13})
540*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({1,0})
541*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(2)
542*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(7*4)
543*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
544*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
545*4bdc9457SAndroid Build Coastguard Worker       .Test();
546*4bdc9457SAndroid Build Coastguard Worker }
547*4bdc9457SAndroid Build Coastguard Worker 
548*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, 3D_flatten_3D) {
549*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
550*4bdc9457SAndroid Build Coastguard Worker       .num_dims(3)
551*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
552*4bdc9457SAndroid Build Coastguard Worker       .perm({2,1,0})
553*4bdc9457SAndroid Build Coastguard Worker       .shape({101,13,7})
554*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({101,13,7})
555*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({2,1,0})
556*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(3)
557*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(4)
558*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
559*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
560*4bdc9457SAndroid Build Coastguard Worker       .Test();
561*4bdc9457SAndroid Build Coastguard Worker }
562*4bdc9457SAndroid Build Coastguard Worker 
563*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, 3D_redundant_dim_first) {
564*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
565*4bdc9457SAndroid Build Coastguard Worker       .num_dims(3)
566*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
567*4bdc9457SAndroid Build Coastguard Worker       .perm({0,2,1})
568*4bdc9457SAndroid Build Coastguard Worker       .shape({1,19,13})
569*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({19,13})
570*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({1,0})
571*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(2)
572*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(4)
573*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
574*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
575*4bdc9457SAndroid Build Coastguard Worker       .Test();
576*4bdc9457SAndroid Build Coastguard Worker }
577*4bdc9457SAndroid Build Coastguard Worker 
578*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, 4D_to_1D) {
579*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
580*4bdc9457SAndroid Build Coastguard Worker       .num_dims(4)
581*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
582*4bdc9457SAndroid Build Coastguard Worker       .perm({0, 2, 3, 1})
583*4bdc9457SAndroid Build Coastguard Worker       .shape({2, 2, 1, 1})
584*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({1})
585*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({0})
586*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(1)
587*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(16)
588*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
589*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
590*4bdc9457SAndroid Build Coastguard Worker       .Test();
591*4bdc9457SAndroid Build Coastguard Worker }
592*4bdc9457SAndroid Build Coastguard Worker 
593*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, 4D_flatten_element_size_2D) {
594*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
595*4bdc9457SAndroid Build Coastguard Worker       .num_dims(4)
596*4bdc9457SAndroid Build Coastguard Worker       .element_size(1)
597*4bdc9457SAndroid Build Coastguard Worker       .perm({1,0,2,3})
598*4bdc9457SAndroid Build Coastguard Worker       .shape({101,13,7,19})
599*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({101,13})
600*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({1,0})
601*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(2)
602*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(1*7*19)
603*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
604*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
605*4bdc9457SAndroid Build Coastguard Worker       .Test();
606*4bdc9457SAndroid Build Coastguard Worker }
607*4bdc9457SAndroid Build Coastguard Worker 
608*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, 4D_flatten_3D) {
609*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
610*4bdc9457SAndroid Build Coastguard Worker       .num_dims(4)
611*4bdc9457SAndroid Build Coastguard Worker       .element_size(2)
612*4bdc9457SAndroid Build Coastguard Worker       .perm({0,3,1,2})
613*4bdc9457SAndroid Build Coastguard Worker       .shape({19,31,41,7})
614*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({19,31*41,7})
615*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({0,2,1})
616*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(3)
617*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(2)
618*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
619*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
620*4bdc9457SAndroid Build Coastguard Worker       .Test();
621*4bdc9457SAndroid Build Coastguard Worker }
622*4bdc9457SAndroid Build Coastguard Worker 
623*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, 5D_double_flatten_3D) {
624*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
625*4bdc9457SAndroid Build Coastguard Worker       .num_dims(5)
626*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
627*4bdc9457SAndroid Build Coastguard Worker       .perm({4,2,3,0,1})
628*4bdc9457SAndroid Build Coastguard Worker       .shape({19,13,31,41,7})
629*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({19*13,31*41,7})
630*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({2,1,0})
631*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(3)
632*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(4)
633*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
634*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
635*4bdc9457SAndroid Build Coastguard Worker       .Test();
636*4bdc9457SAndroid Build Coastguard Worker }
637*4bdc9457SAndroid Build Coastguard Worker 
638*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, 5D_flatten_3D) {
639*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
640*4bdc9457SAndroid Build Coastguard Worker       .num_dims(5)
641*4bdc9457SAndroid Build Coastguard Worker       .element_size(2)
642*4bdc9457SAndroid Build Coastguard Worker       .perm({4,3,0,1,2})
643*4bdc9457SAndroid Build Coastguard Worker       .shape({19,13,31,41,7})
644*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({19*13*31,41,7})
645*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({2,1,0})
646*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(3)
647*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(2)
648*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
649*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
650*4bdc9457SAndroid Build Coastguard Worker       .Test();
651*4bdc9457SAndroid Build Coastguard Worker }
652*4bdc9457SAndroid Build Coastguard Worker 
653*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, 5D_flatten_4D) {
654*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
655*4bdc9457SAndroid Build Coastguard Worker       .num_dims(5)
656*4bdc9457SAndroid Build Coastguard Worker       .element_size(2)
657*4bdc9457SAndroid Build Coastguard Worker       .perm({4,3,1,2,0})
658*4bdc9457SAndroid Build Coastguard Worker       .shape({19,13,31,41,7})
659*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({19,13*31,41,7})
660*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({3,2,1,0})
661*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(4)
662*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(2)
663*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
664*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
665*4bdc9457SAndroid Build Coastguard Worker       .Test();
666*4bdc9457SAndroid Build Coastguard Worker }
667*4bdc9457SAndroid Build Coastguard Worker 
668*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, 5D_flatten_2D) {
669*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
670*4bdc9457SAndroid Build Coastguard Worker       .num_dims(6)
671*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
672*4bdc9457SAndroid Build Coastguard Worker       .perm({4,5,0,1,2,3})
673*4bdc9457SAndroid Build Coastguard Worker       .shape({53,19,13,31,41,7})
674*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({53*19*13*31,41*7})
675*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({1,0})
676*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(2)
677*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(4)
678*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
679*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
680*4bdc9457SAndroid Build Coastguard Worker       .Test();
681*4bdc9457SAndroid Build Coastguard Worker }
682*4bdc9457SAndroid Build Coastguard Worker 
683*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, 6D_flatten_3D) {
684*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
685*4bdc9457SAndroid Build Coastguard Worker       .num_dims(6)
686*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
687*4bdc9457SAndroid Build Coastguard Worker       .perm({0,1,2,3,5,4})
688*4bdc9457SAndroid Build Coastguard Worker       .shape({53,19,13,31,41,7})
689*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({53*19*13*31,41,7})
690*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({0,2,1})
691*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(3)
692*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(4)
693*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
694*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
695*4bdc9457SAndroid Build Coastguard Worker       .Test();
696*4bdc9457SAndroid Build Coastguard Worker }
697*4bdc9457SAndroid Build Coastguard Worker 
698*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, 6D_double_flatten_3D) {
699*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
700*4bdc9457SAndroid Build Coastguard Worker       .num_dims(6)
701*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
702*4bdc9457SAndroid Build Coastguard Worker       .perm({0,3,1,2,4,5})
703*4bdc9457SAndroid Build Coastguard Worker       .shape({53,19,13,31,41,7})
704*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({53,19*13,31})
705*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({0,2,1})
706*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(3)
707*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(4*41*7)
708*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
709*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
710*4bdc9457SAndroid Build Coastguard Worker       .Test();
711*4bdc9457SAndroid Build Coastguard Worker }
712*4bdc9457SAndroid Build Coastguard Worker 
713*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, 6D_double_flatten_4D) {
714*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
715*4bdc9457SAndroid Build Coastguard Worker       .num_dims(6)
716*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
717*4bdc9457SAndroid Build Coastguard Worker       .perm({4,5,3,1,2,0})
718*4bdc9457SAndroid Build Coastguard Worker       .shape({53,19,13,31,41,7})
719*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({53,19*13,31,41*7})
720*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({3,2,1,0})
721*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(4)
722*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(4)
723*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
724*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
725*4bdc9457SAndroid Build Coastguard Worker       .Test();
726*4bdc9457SAndroid Build Coastguard Worker }
727*4bdc9457SAndroid Build Coastguard Worker 
728*4bdc9457SAndroid Build Coastguard Worker TEST(TRANSPOSE_NORMALIZATION_TEST, 6D_flatten_ones) {
729*4bdc9457SAndroid Build Coastguard Worker     TransposeNormalizationTester()
730*4bdc9457SAndroid Build Coastguard Worker       .num_dims(6)
731*4bdc9457SAndroid Build Coastguard Worker       .element_size(4)
732*4bdc9457SAndroid Build Coastguard Worker       .perm({5,4,3,2,1,0})
733*4bdc9457SAndroid Build Coastguard Worker       .shape({23,1,1,1,17,13})
734*4bdc9457SAndroid Build Coastguard Worker       .expected_shape({23,17,13})
735*4bdc9457SAndroid Build Coastguard Worker       .expected_perm({2,1,0})
736*4bdc9457SAndroid Build Coastguard Worker       .expected_dims(3)
737*4bdc9457SAndroid Build Coastguard Worker       .expected_element_size(4)
738*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_input_stride()
739*4bdc9457SAndroid Build Coastguard Worker       .calculate_expected_output_stride()
740*4bdc9457SAndroid Build Coastguard Worker       .Test();
741*4bdc9457SAndroid Build Coastguard Worker }
742