1*4bdc9457SAndroid Build Coastguard Worker // Copyright 2019 Google LLC
2*4bdc9457SAndroid Build Coastguard Worker //
3*4bdc9457SAndroid Build Coastguard Worker // This source code is licensed under the BSD-style license found in the
4*4bdc9457SAndroid Build Coastguard Worker // LICENSE file in the root directory of this source tree.
5*4bdc9457SAndroid Build Coastguard Worker
6*4bdc9457SAndroid Build Coastguard Worker #include <gtest/gtest.h>
7*4bdc9457SAndroid Build Coastguard Worker
8*4bdc9457SAndroid Build Coastguard Worker #include <xnnpack/params.h>
9*4bdc9457SAndroid Build Coastguard Worker
10*4bdc9457SAndroid Build Coastguard Worker #include "deconvolution-operator-tester.h"
11*4bdc9457SAndroid Build Coastguard Worker
12*4bdc9457SAndroid Build Coastguard Worker
13*4bdc9457SAndroid Build Coastguard Worker constexpr size_t kUnstridedInputHeight = 8;
14*4bdc9457SAndroid Build Coastguard Worker constexpr size_t kUnstridedInputWidth = 7;
15*4bdc9457SAndroid Build Coastguard Worker constexpr size_t kStridedInputHeight = 6;
16*4bdc9457SAndroid Build Coastguard Worker constexpr size_t kStridedInputWidth = 5;
17*4bdc9457SAndroid Build Coastguard Worker
18*4bdc9457SAndroid Build Coastguard Worker
19*4bdc9457SAndroid Build Coastguard Worker /**************************** Future GEMM path ****************************/
20*4bdc9457SAndroid Build Coastguard Worker
21*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 1x1) {
22*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
23*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
24*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
25*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
26*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
27*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
28*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
29*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
30*4bdc9457SAndroid Build Coastguard Worker }
31*4bdc9457SAndroid Build Coastguard Worker
32*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 1x1_varying_input_width) {
33*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
34*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
35*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
36*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
37*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
38*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
39*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
40*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
41*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
42*4bdc9457SAndroid Build Coastguard Worker }
43*4bdc9457SAndroid Build Coastguard Worker }
44*4bdc9457SAndroid Build Coastguard Worker
45*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 1x1_varying_input_height) {
46*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
47*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
48*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
49*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
50*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
51*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
52*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
53*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
54*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
55*4bdc9457SAndroid Build Coastguard Worker }
56*4bdc9457SAndroid Build Coastguard Worker }
57*4bdc9457SAndroid Build Coastguard Worker
58*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 1x1_varying_input_channels) {
59*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
60*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
61*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
62*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
63*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
64*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
65*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
66*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
67*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
68*4bdc9457SAndroid Build Coastguard Worker }
69*4bdc9457SAndroid Build Coastguard Worker }
70*4bdc9457SAndroid Build Coastguard Worker
71*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 1x1_varying_output_channels) {
72*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
73*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qs8.gemm.nr * 2; output_channels *= 2) {
74*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
75*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
76*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
77*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
78*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
79*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
80*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
81*4bdc9457SAndroid Build Coastguard Worker }
82*4bdc9457SAndroid Build Coastguard Worker }
83*4bdc9457SAndroid Build Coastguard Worker
84*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 1x1_with_input_stride) {
85*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
86*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
87*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
88*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
89*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
90*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
91*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
92*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
93*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
94*4bdc9457SAndroid Build Coastguard Worker }
95*4bdc9457SAndroid Build Coastguard Worker
96*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 1x1_with_output_stride) {
97*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
98*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
99*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
100*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
101*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
102*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
103*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qs8.gemm.nr * 2 + 13)
104*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
105*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
106*4bdc9457SAndroid Build Coastguard Worker }
107*4bdc9457SAndroid Build Coastguard Worker
108*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 1x1_with_qmin) {
109*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
110*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
111*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
112*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
113*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
114*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
115*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
116*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
117*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
118*4bdc9457SAndroid Build Coastguard Worker }
119*4bdc9457SAndroid Build Coastguard Worker
120*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 1x1_with_qmax) {
121*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
122*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
123*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
124*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
125*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
126*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
127*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
128*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
129*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
130*4bdc9457SAndroid Build Coastguard Worker }
131*4bdc9457SAndroid Build Coastguard Worker
132*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 1x1_without_bias) {
133*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
134*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
135*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
136*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
137*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
138*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
139*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
140*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
141*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
142*4bdc9457SAndroid Build Coastguard Worker }
143*4bdc9457SAndroid Build Coastguard Worker
144*4bdc9457SAndroid Build Coastguard Worker /**************************** Future GEMM path, grouped ****************************/
145*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_1x1)146*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_1x1) {
147*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
148*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
149*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
150*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
151*4bdc9457SAndroid Build Coastguard Worker .groups(2)
152*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
153*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
154*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
155*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
156*4bdc9457SAndroid Build Coastguard Worker }
157*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_1x1_varying_input_width)158*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_1x1_varying_input_width) {
159*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
160*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
161*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
162*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
163*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
164*4bdc9457SAndroid Build Coastguard Worker .groups(2)
165*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
166*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
167*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
168*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
169*4bdc9457SAndroid Build Coastguard Worker }
170*4bdc9457SAndroid Build Coastguard Worker }
171*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_1x1_varying_input_height)172*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_1x1_varying_input_height) {
173*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
174*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
175*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
176*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
177*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
178*4bdc9457SAndroid Build Coastguard Worker .groups(2)
179*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
180*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
181*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
182*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
183*4bdc9457SAndroid Build Coastguard Worker }
184*4bdc9457SAndroid Build Coastguard Worker }
185*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_1x1_varying_input_channels)186*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_1x1_varying_input_channels) {
187*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
188*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
189*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
190*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
191*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
192*4bdc9457SAndroid Build Coastguard Worker .groups(2)
193*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
194*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
195*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
196*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
197*4bdc9457SAndroid Build Coastguard Worker }
198*4bdc9457SAndroid Build Coastguard Worker }
199*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_1x1_varying_output_channels)200*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_1x1_varying_output_channels) {
201*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
202*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qs8.gemm.nr * 2; output_channels *= 2) {
203*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
204*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
205*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
206*4bdc9457SAndroid Build Coastguard Worker .groups(2)
207*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
208*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
209*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
210*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
211*4bdc9457SAndroid Build Coastguard Worker }
212*4bdc9457SAndroid Build Coastguard Worker }
213*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_1x1_with_input_stride)214*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_1x1_with_input_stride) {
215*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
216*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
217*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
218*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
219*4bdc9457SAndroid Build Coastguard Worker .groups(2)
220*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
221*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
222*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(47)
223*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
224*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
225*4bdc9457SAndroid Build Coastguard Worker }
226*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_1x1_with_output_stride)227*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_1x1_with_output_stride) {
228*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
229*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
230*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
231*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
232*4bdc9457SAndroid Build Coastguard Worker .groups(2)
233*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
234*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr + 3)
235*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qs8.gemm.nr * 2 + 13)
236*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
237*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
238*4bdc9457SAndroid Build Coastguard Worker }
239*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_1x1_with_qmin)240*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_1x1_with_qmin) {
241*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
242*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
243*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
244*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
245*4bdc9457SAndroid Build Coastguard Worker .groups(2)
246*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
247*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
248*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
249*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
250*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
251*4bdc9457SAndroid Build Coastguard Worker }
252*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_1x1_with_qmax)253*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_1x1_with_qmax) {
254*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
255*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
256*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
257*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
258*4bdc9457SAndroid Build Coastguard Worker .groups(2)
259*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
260*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
261*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
262*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
263*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
264*4bdc9457SAndroid Build Coastguard Worker }
265*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_1x1_without_bias)266*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_1x1_without_bias) {
267*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
268*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
269*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
270*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
271*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
272*4bdc9457SAndroid Build Coastguard Worker .groups(2)
273*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
274*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
275*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
276*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
277*4bdc9457SAndroid Build Coastguard Worker }
278*4bdc9457SAndroid Build Coastguard Worker
279*4bdc9457SAndroid Build Coastguard Worker /**************************** Future GEMM path, batched ****************************/
280*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_1x1)281*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_1x1) {
282*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
283*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
284*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
285*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
286*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
287*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
288*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
289*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
290*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
291*4bdc9457SAndroid Build Coastguard Worker }
292*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_1x1_varying_input_width)293*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_1x1_varying_input_width) {
294*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
295*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
296*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
297*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
298*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
299*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
300*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
301*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
302*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
303*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
304*4bdc9457SAndroid Build Coastguard Worker }
305*4bdc9457SAndroid Build Coastguard Worker }
306*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_1x1_varying_input_height)307*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_1x1_varying_input_height) {
308*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
309*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
310*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
311*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
312*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
313*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
314*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
315*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
316*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
317*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
318*4bdc9457SAndroid Build Coastguard Worker }
319*4bdc9457SAndroid Build Coastguard Worker }
320*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_1x1_varying_input_channels)321*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_1x1_varying_input_channels) {
322*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
323*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
324*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
325*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
326*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
327*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
328*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
329*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
330*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
331*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
332*4bdc9457SAndroid Build Coastguard Worker }
333*4bdc9457SAndroid Build Coastguard Worker }
334*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_1x1_varying_output_channels)335*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_1x1_varying_output_channels) {
336*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
337*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qs8.gemm.nr * 2; output_channels *= 2) {
338*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
339*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
340*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
341*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
342*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
343*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
344*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
345*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
346*4bdc9457SAndroid Build Coastguard Worker }
347*4bdc9457SAndroid Build Coastguard Worker }
348*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_1x1_with_input_stride)349*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_1x1_with_input_stride) {
350*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
351*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
352*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
353*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
354*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
355*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
356*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
357*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
358*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
359*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
360*4bdc9457SAndroid Build Coastguard Worker }
361*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_1x1_with_output_stride)362*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_1x1_with_output_stride) {
363*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
364*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
365*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
366*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
367*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
368*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
369*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
370*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qs8.gemm.nr * 2 + 13)
371*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
372*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
373*4bdc9457SAndroid Build Coastguard Worker }
374*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_1x1_with_qmin)375*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_1x1_with_qmin) {
376*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
377*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
378*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
379*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
380*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
381*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
382*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
383*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
384*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
385*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
386*4bdc9457SAndroid Build Coastguard Worker }
387*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_1x1_with_qmax)388*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_1x1_with_qmax) {
389*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
390*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
391*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
392*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
393*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
394*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
395*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
396*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
397*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
398*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
399*4bdc9457SAndroid Build Coastguard Worker }
400*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_1x1_without_bias)401*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_1x1_without_bias) {
402*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
403*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
404*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
405*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
406*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
407*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
408*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
409*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
410*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
411*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
412*4bdc9457SAndroid Build Coastguard Worker }
413*4bdc9457SAndroid Build Coastguard Worker
414*4bdc9457SAndroid Build Coastguard Worker /**************************** Future GEMM path, batched, grouped ****************************/
415*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_1x1)416*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_1x1) {
417*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
418*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
419*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
420*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
421*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
422*4bdc9457SAndroid Build Coastguard Worker .groups(2)
423*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
424*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
425*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
426*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
427*4bdc9457SAndroid Build Coastguard Worker }
428*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_1x1_varying_input_width)429*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_1x1_varying_input_width) {
430*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
431*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
432*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
433*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
434*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
435*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
436*4bdc9457SAndroid Build Coastguard Worker .groups(2)
437*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
438*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
439*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
440*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
441*4bdc9457SAndroid Build Coastguard Worker }
442*4bdc9457SAndroid Build Coastguard Worker }
443*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_1x1_varying_input_height)444*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_1x1_varying_input_height) {
445*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
446*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
447*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
448*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
449*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
450*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
451*4bdc9457SAndroid Build Coastguard Worker .groups(2)
452*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
453*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
454*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
455*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
456*4bdc9457SAndroid Build Coastguard Worker }
457*4bdc9457SAndroid Build Coastguard Worker }
458*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_1x1_varying_input_channels)459*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_1x1_varying_input_channels) {
460*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
461*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
462*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
463*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
464*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
465*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
466*4bdc9457SAndroid Build Coastguard Worker .groups(2)
467*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
468*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
469*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
470*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
471*4bdc9457SAndroid Build Coastguard Worker }
472*4bdc9457SAndroid Build Coastguard Worker }
473*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_1x1_varying_output_channels)474*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_1x1_varying_output_channels) {
475*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
476*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qs8.gemm.nr * 2; output_channels *= 2) {
477*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
478*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
479*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
480*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
481*4bdc9457SAndroid Build Coastguard Worker .groups(2)
482*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
483*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
484*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
485*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
486*4bdc9457SAndroid Build Coastguard Worker }
487*4bdc9457SAndroid Build Coastguard Worker }
488*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_1x1_with_input_stride)489*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_1x1_with_input_stride) {
490*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
491*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
492*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
493*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
494*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
495*4bdc9457SAndroid Build Coastguard Worker .groups(2)
496*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
497*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
498*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(47)
499*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
500*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
501*4bdc9457SAndroid Build Coastguard Worker }
502*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_1x1_with_output_stride)503*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_1x1_with_output_stride) {
504*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
505*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
506*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
507*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
508*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
509*4bdc9457SAndroid Build Coastguard Worker .groups(2)
510*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
511*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr + 3)
512*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qs8.gemm.nr * 2 + 13)
513*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
514*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
515*4bdc9457SAndroid Build Coastguard Worker }
516*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_1x1_with_qmin)517*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_1x1_with_qmin) {
518*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
519*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
520*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
521*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
522*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
523*4bdc9457SAndroid Build Coastguard Worker .groups(2)
524*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
525*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
526*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
527*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
528*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
529*4bdc9457SAndroid Build Coastguard Worker }
530*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_1x1_with_qmax)531*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_1x1_with_qmax) {
532*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
533*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
534*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
535*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
536*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
537*4bdc9457SAndroid Build Coastguard Worker .groups(2)
538*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
539*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
540*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
541*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
542*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
543*4bdc9457SAndroid Build Coastguard Worker }
544*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_1x1_without_bias)545*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_1x1_without_bias) {
546*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
547*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
548*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
549*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
550*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
551*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
552*4bdc9457SAndroid Build Coastguard Worker .groups(2)
553*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
554*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
555*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
556*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
557*4bdc9457SAndroid Build Coastguard Worker }
558*4bdc9457SAndroid Build Coastguard Worker
559*4bdc9457SAndroid Build Coastguard Worker /**************************** CONV path ****************************/
560*4bdc9457SAndroid Build Coastguard Worker
561*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3) {
562*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
563*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
564*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
565*4bdc9457SAndroid Build Coastguard Worker .padding(1)
566*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
567*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
568*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
569*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
570*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
571*4bdc9457SAndroid Build Coastguard Worker }
572*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,Kx3)573*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, Kx3) {
574*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
575*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 1; kernel_height <= 4; kernel_height *= 2) {
576*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
577*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
578*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
579*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
580*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
581*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
582*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
583*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
584*4bdc9457SAndroid Build Coastguard Worker }
585*4bdc9457SAndroid Build Coastguard Worker }
586*4bdc9457SAndroid Build Coastguard Worker
587*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3xK) {
588*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
589*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 1; kernel_width <= 4; kernel_width *= 2) {
590*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
591*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
592*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
593*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
594*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
595*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
596*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
597*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
598*4bdc9457SAndroid Build Coastguard Worker }
599*4bdc9457SAndroid Build Coastguard Worker }
600*4bdc9457SAndroid Build Coastguard Worker
601*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3_varying_height_padding) {
602*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
603*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
604*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
605*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
606*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
607*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
608*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
609*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
610*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
611*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
612*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
613*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
614*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
615*4bdc9457SAndroid Build Coastguard Worker }
616*4bdc9457SAndroid Build Coastguard Worker }
617*4bdc9457SAndroid Build Coastguard Worker }
618*4bdc9457SAndroid Build Coastguard Worker
619*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3_varying_width_padding) {
620*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
621*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
622*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
623*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
624*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
625*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
626*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
627*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
628*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
629*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
630*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
631*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
632*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
633*4bdc9457SAndroid Build Coastguard Worker }
634*4bdc9457SAndroid Build Coastguard Worker }
635*4bdc9457SAndroid Build Coastguard Worker }
636*4bdc9457SAndroid Build Coastguard Worker
637*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3_varying_height_adjustment) {
638*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
639*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 1; adjustment_height <= 2; adjustment_height++) {
640*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
641*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
642*4bdc9457SAndroid Build Coastguard Worker .padding(1)
643*4bdc9457SAndroid Build Coastguard Worker .stride_height(adjustment_height + 1)
644*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
645*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
646*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
647*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
648*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
649*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
650*4bdc9457SAndroid Build Coastguard Worker }
651*4bdc9457SAndroid Build Coastguard Worker }
652*4bdc9457SAndroid Build Coastguard Worker
653*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3_varying_width_adjustment) {
654*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
655*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 1; adjustment_width <= 2; adjustment_width++) {
656*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
657*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
658*4bdc9457SAndroid Build Coastguard Worker .padding(1)
659*4bdc9457SAndroid Build Coastguard Worker .stride_width(adjustment_width + 1)
660*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
661*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
662*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
663*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
664*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
665*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
666*4bdc9457SAndroid Build Coastguard Worker }
667*4bdc9457SAndroid Build Coastguard Worker }
668*4bdc9457SAndroid Build Coastguard Worker
669*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3_varying_input_height) {
670*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
671*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
672*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
673*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
674*4bdc9457SAndroid Build Coastguard Worker .padding(1)
675*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
676*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
677*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
678*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
679*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
680*4bdc9457SAndroid Build Coastguard Worker }
681*4bdc9457SAndroid Build Coastguard Worker }
682*4bdc9457SAndroid Build Coastguard Worker
683*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3_varying_input_width) {
684*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
685*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
686*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
687*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
688*4bdc9457SAndroid Build Coastguard Worker .padding(1)
689*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
690*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
691*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
692*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
693*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
694*4bdc9457SAndroid Build Coastguard Worker }
695*4bdc9457SAndroid Build Coastguard Worker }
696*4bdc9457SAndroid Build Coastguard Worker
697*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3_varying_input_channels) {
698*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
699*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
700*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
701*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
702*4bdc9457SAndroid Build Coastguard Worker .padding(1)
703*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
704*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
705*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
706*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
707*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
708*4bdc9457SAndroid Build Coastguard Worker }
709*4bdc9457SAndroid Build Coastguard Worker }
710*4bdc9457SAndroid Build Coastguard Worker
711*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3_varying_output_channels) {
712*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
713*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qs8.gemm.nr * 2; output_channels *= 2) {
714*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
715*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
716*4bdc9457SAndroid Build Coastguard Worker .padding(1)
717*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
718*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
719*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
720*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
721*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
722*4bdc9457SAndroid Build Coastguard Worker }
723*4bdc9457SAndroid Build Coastguard Worker }
724*4bdc9457SAndroid Build Coastguard Worker
725*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3_with_height_dilation) {
726*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
727*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_height = 2; dilation_height <= 3; dilation_height++) {
728*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
729*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
730*4bdc9457SAndroid Build Coastguard Worker .padding(1)
731*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
732*4bdc9457SAndroid Build Coastguard Worker .dilation_height(dilation_height)
733*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
734*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
735*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
736*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
737*4bdc9457SAndroid Build Coastguard Worker }
738*4bdc9457SAndroid Build Coastguard Worker }
739*4bdc9457SAndroid Build Coastguard Worker
740*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3_with_width_dilation) {
741*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
742*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) {
743*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
744*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
745*4bdc9457SAndroid Build Coastguard Worker .padding(1)
746*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
747*4bdc9457SAndroid Build Coastguard Worker .dilation_width(dilation_width)
748*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
749*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
750*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
751*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
752*4bdc9457SAndroid Build Coastguard Worker }
753*4bdc9457SAndroid Build Coastguard Worker }
754*4bdc9457SAndroid Build Coastguard Worker
755*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3_with_height_dilation_and_stride) {
756*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
757*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
758*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
759*4bdc9457SAndroid Build Coastguard Worker .padding(1)
760*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
761*4bdc9457SAndroid Build Coastguard Worker .dilation_height(3)
762*4bdc9457SAndroid Build Coastguard Worker .stride_height(2)
763*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
764*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
765*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
766*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
767*4bdc9457SAndroid Build Coastguard Worker }
768*4bdc9457SAndroid Build Coastguard Worker
769*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3_with_width_dilation_and_stride) {
770*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
771*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
772*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
773*4bdc9457SAndroid Build Coastguard Worker .padding(1)
774*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
775*4bdc9457SAndroid Build Coastguard Worker .dilation_width(3)
776*4bdc9457SAndroid Build Coastguard Worker .stride_width(2)
777*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
778*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
779*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
780*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
781*4bdc9457SAndroid Build Coastguard Worker }
782*4bdc9457SAndroid Build Coastguard Worker
783*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3_with_input_stride) {
784*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
785*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
786*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
787*4bdc9457SAndroid Build Coastguard Worker .padding(1)
788*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
789*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
790*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
791*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
792*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
793*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
794*4bdc9457SAndroid Build Coastguard Worker }
795*4bdc9457SAndroid Build Coastguard Worker
796*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3_with_output_stride) {
797*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
798*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
799*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
800*4bdc9457SAndroid Build Coastguard Worker .padding(1)
801*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
802*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
803*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
804*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qs8.gemm.nr * 2 + 13)
805*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
806*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
807*4bdc9457SAndroid Build Coastguard Worker }
808*4bdc9457SAndroid Build Coastguard Worker
809*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3_with_qmin) {
810*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
811*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
812*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
813*4bdc9457SAndroid Build Coastguard Worker .padding(1)
814*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
815*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
816*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
817*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
818*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
819*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
820*4bdc9457SAndroid Build Coastguard Worker }
821*4bdc9457SAndroid Build Coastguard Worker
822*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3_with_qmax) {
823*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
824*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
825*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
826*4bdc9457SAndroid Build Coastguard Worker .padding(1)
827*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
828*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
829*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
830*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
831*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
832*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
833*4bdc9457SAndroid Build Coastguard Worker }
834*4bdc9457SAndroid Build Coastguard Worker
835*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3_without_bias) {
836*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
837*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
838*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
839*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
840*4bdc9457SAndroid Build Coastguard Worker .padding(1)
841*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
842*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
843*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
844*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
845*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
846*4bdc9457SAndroid Build Coastguard Worker }
847*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,weights_cache_3x3)848*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, weights_cache_3x3) {
849*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
850*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
851*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
852*4bdc9457SAndroid Build Coastguard Worker .padding(1)
853*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
854*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
855*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
856*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
857*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
858*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
859*4bdc9457SAndroid Build Coastguard Worker }
860*4bdc9457SAndroid Build Coastguard Worker
861*4bdc9457SAndroid Build Coastguard Worker /**************************** CONV path, grouped ****************************/
862*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3)863*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3) {
864*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
865*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
866*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
867*4bdc9457SAndroid Build Coastguard Worker .padding(1)
868*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
869*4bdc9457SAndroid Build Coastguard Worker .groups(2)
870*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
871*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
872*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
873*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
874*4bdc9457SAndroid Build Coastguard Worker }
875*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_Kx3)876*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_Kx3) {
877*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
878*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 1; kernel_height <= 4; kernel_height *= 2) {
879*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
880*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
881*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
882*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
883*4bdc9457SAndroid Build Coastguard Worker .groups(2)
884*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
885*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
886*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
887*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
888*4bdc9457SAndroid Build Coastguard Worker }
889*4bdc9457SAndroid Build Coastguard Worker }
890*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3xK)891*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3xK) {
892*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
893*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 1; kernel_width <= 4; kernel_width *= 2) {
894*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
895*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
896*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
897*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
898*4bdc9457SAndroid Build Coastguard Worker .groups(2)
899*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
900*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
901*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
902*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
903*4bdc9457SAndroid Build Coastguard Worker }
904*4bdc9457SAndroid Build Coastguard Worker }
905*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3_varying_height_padding)906*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3_varying_height_padding) {
907*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
908*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
909*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
910*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
911*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
912*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
913*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
914*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
915*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
916*4bdc9457SAndroid Build Coastguard Worker .groups(2)
917*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
918*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
919*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
920*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
921*4bdc9457SAndroid Build Coastguard Worker }
922*4bdc9457SAndroid Build Coastguard Worker }
923*4bdc9457SAndroid Build Coastguard Worker }
924*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3_varying_width_padding)925*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3_varying_width_padding) {
926*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
927*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
928*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
929*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
930*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
931*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
932*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
933*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
934*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
935*4bdc9457SAndroid Build Coastguard Worker .groups(2)
936*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
937*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
938*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
939*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
940*4bdc9457SAndroid Build Coastguard Worker }
941*4bdc9457SAndroid Build Coastguard Worker }
942*4bdc9457SAndroid Build Coastguard Worker }
943*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3_varying_height_adjustment)944*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3_varying_height_adjustment) {
945*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
946*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 1; adjustment_height <= 2; adjustment_height++) {
947*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
948*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
949*4bdc9457SAndroid Build Coastguard Worker .padding(1)
950*4bdc9457SAndroid Build Coastguard Worker .stride_height(adjustment_height + 1)
951*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
952*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
953*4bdc9457SAndroid Build Coastguard Worker .groups(2)
954*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
955*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
956*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
957*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
958*4bdc9457SAndroid Build Coastguard Worker }
959*4bdc9457SAndroid Build Coastguard Worker }
960*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3_varying_width_adjustment)961*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3_varying_width_adjustment) {
962*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
963*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 1; adjustment_width <= 2; adjustment_width++) {
964*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
965*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
966*4bdc9457SAndroid Build Coastguard Worker .padding(1)
967*4bdc9457SAndroid Build Coastguard Worker .stride_width(adjustment_width + 1)
968*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
969*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
970*4bdc9457SAndroid Build Coastguard Worker .groups(2)
971*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
972*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
973*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
974*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
975*4bdc9457SAndroid Build Coastguard Worker }
976*4bdc9457SAndroid Build Coastguard Worker }
977*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3_varying_input_height)978*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3_varying_input_height) {
979*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
980*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
981*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
982*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
983*4bdc9457SAndroid Build Coastguard Worker .padding(1)
984*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
985*4bdc9457SAndroid Build Coastguard Worker .groups(2)
986*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
987*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
988*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
989*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
990*4bdc9457SAndroid Build Coastguard Worker }
991*4bdc9457SAndroid Build Coastguard Worker }
992*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3_varying_input_width)993*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3_varying_input_width) {
994*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
995*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
996*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
997*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
998*4bdc9457SAndroid Build Coastguard Worker .padding(1)
999*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1000*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1001*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
1002*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1003*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
1004*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1005*4bdc9457SAndroid Build Coastguard Worker }
1006*4bdc9457SAndroid Build Coastguard Worker }
1007*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3_varying_input_channels)1008*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3_varying_input_channels) {
1009*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1010*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
1011*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1012*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1013*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1014*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1015*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1016*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
1017*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1018*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
1019*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1020*4bdc9457SAndroid Build Coastguard Worker }
1021*4bdc9457SAndroid Build Coastguard Worker }
1022*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3_varying_output_channels)1023*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3_varying_output_channels) {
1024*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1025*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qs8.gemm.nr * 2; output_channels *= 2) {
1026*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1027*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1028*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1029*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1030*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1031*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1032*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
1033*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
1034*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1035*4bdc9457SAndroid Build Coastguard Worker }
1036*4bdc9457SAndroid Build Coastguard Worker }
1037*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3_with_height_dilation)1038*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3_with_height_dilation) {
1039*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1040*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_height = 2; dilation_height <= 3; dilation_height++) {
1041*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1042*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1043*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1044*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1045*4bdc9457SAndroid Build Coastguard Worker .dilation_height(dilation_height)
1046*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1047*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1048*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1049*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1050*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1051*4bdc9457SAndroid Build Coastguard Worker }
1052*4bdc9457SAndroid Build Coastguard Worker }
1053*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3_with_width_dilation)1054*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3_with_width_dilation) {
1055*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1056*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) {
1057*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1058*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1059*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1060*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1061*4bdc9457SAndroid Build Coastguard Worker .dilation_width(dilation_width)
1062*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1063*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1064*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1065*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1066*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1067*4bdc9457SAndroid Build Coastguard Worker }
1068*4bdc9457SAndroid Build Coastguard Worker }
1069*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3_with_height_dilation_and_stride)1070*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3_with_height_dilation_and_stride) {
1071*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1072*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1073*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1074*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1075*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1076*4bdc9457SAndroid Build Coastguard Worker .dilation_height(3)
1077*4bdc9457SAndroid Build Coastguard Worker .stride_height(2)
1078*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1079*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1080*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1081*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1082*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1083*4bdc9457SAndroid Build Coastguard Worker }
1084*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3_with_width_dilation_and_stride)1085*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3_with_width_dilation_and_stride) {
1086*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1087*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1088*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1089*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1090*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1091*4bdc9457SAndroid Build Coastguard Worker .dilation_width(3)
1092*4bdc9457SAndroid Build Coastguard Worker .stride_width(2)
1093*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1094*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1095*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1096*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1097*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1098*4bdc9457SAndroid Build Coastguard Worker }
1099*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3_with_input_stride)1100*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3_with_input_stride) {
1101*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1102*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1103*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1104*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1105*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1106*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1107*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1108*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1109*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(47)
1110*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1111*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1112*4bdc9457SAndroid Build Coastguard Worker }
1113*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3_with_output_stride)1114*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3_with_output_stride) {
1115*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1116*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1117*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1118*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1119*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1120*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1121*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1122*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr + 3)
1123*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qs8.gemm.nr * 2 + 13)
1124*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1125*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1126*4bdc9457SAndroid Build Coastguard Worker }
1127*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3_with_qmin)1128*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3_with_qmin) {
1129*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1130*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1131*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1132*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1133*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1134*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1135*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1136*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1137*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
1138*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1139*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1140*4bdc9457SAndroid Build Coastguard Worker }
1141*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3_with_qmax)1142*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3_with_qmax) {
1143*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1144*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1145*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1146*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1147*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1148*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1149*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1150*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1151*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
1152*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1153*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1154*4bdc9457SAndroid Build Coastguard Worker }
1155*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3_without_bias)1156*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3_without_bias) {
1157*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1158*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1159*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
1160*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1161*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1162*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1163*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1164*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1165*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1166*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1167*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1168*4bdc9457SAndroid Build Coastguard Worker }
1169*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,weights_cache_grouped_3x3)1170*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, weights_cache_grouped_3x3) {
1171*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1172*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1173*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1174*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1175*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1176*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1177*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
1178*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1179*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
1180*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1181*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1182*4bdc9457SAndroid Build Coastguard Worker }
1183*4bdc9457SAndroid Build Coastguard Worker
1184*4bdc9457SAndroid Build Coastguard Worker /**************************** CONV path, batched ****************************/
1185*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3)1186*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3) {
1187*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1188*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1189*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1190*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1191*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1192*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1193*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
1194*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1195*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1196*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1197*4bdc9457SAndroid Build Coastguard Worker }
1198*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_Kx3)1199*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_Kx3) {
1200*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1201*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 1; kernel_height <= 4; kernel_height *= 2) {
1202*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1203*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1204*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1205*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
1206*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
1207*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
1208*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1209*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1210*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1211*4bdc9457SAndroid Build Coastguard Worker }
1212*4bdc9457SAndroid Build Coastguard Worker }
1213*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3xK)1214*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3xK) {
1215*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1216*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 1; kernel_width <= 4; kernel_width *= 2) {
1217*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1218*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1219*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1220*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
1221*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
1222*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
1223*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1224*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1225*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1226*4bdc9457SAndroid Build Coastguard Worker }
1227*4bdc9457SAndroid Build Coastguard Worker }
1228*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3_varying_height_padding)1229*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3_varying_height_padding) {
1230*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1231*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
1232*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
1233*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1234*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1235*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1236*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
1237*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
1238*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
1239*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1240*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
1241*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1242*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
1243*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1244*4bdc9457SAndroid Build Coastguard Worker }
1245*4bdc9457SAndroid Build Coastguard Worker }
1246*4bdc9457SAndroid Build Coastguard Worker }
1247*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3_varying_width_padding)1248*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3_varying_width_padding) {
1249*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1250*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
1251*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
1252*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1253*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1254*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1255*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
1256*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
1257*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
1258*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1259*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
1260*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1261*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
1262*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1263*4bdc9457SAndroid Build Coastguard Worker }
1264*4bdc9457SAndroid Build Coastguard Worker }
1265*4bdc9457SAndroid Build Coastguard Worker }
1266*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3_varying_height_adjustment)1267*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3_varying_height_adjustment) {
1268*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1269*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 1; adjustment_height <= 2; adjustment_height++) {
1270*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1271*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1272*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1273*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1274*4bdc9457SAndroid Build Coastguard Worker .stride_height(adjustment_height + 1)
1275*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
1276*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1277*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
1278*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1279*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
1280*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1281*4bdc9457SAndroid Build Coastguard Worker }
1282*4bdc9457SAndroid Build Coastguard Worker }
1283*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3_varying_width_adjustment)1284*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3_varying_width_adjustment) {
1285*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1286*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 1; adjustment_width <= 2; adjustment_width++) {
1287*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1288*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1289*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1290*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1291*4bdc9457SAndroid Build Coastguard Worker .stride_width(adjustment_width + 1)
1292*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
1293*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1294*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
1295*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1296*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
1297*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1298*4bdc9457SAndroid Build Coastguard Worker }
1299*4bdc9457SAndroid Build Coastguard Worker }
1300*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3_varying_input_height)1301*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3_varying_input_height) {
1302*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1303*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
1304*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1305*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1306*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
1307*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1308*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1309*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
1310*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1311*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
1312*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1313*4bdc9457SAndroid Build Coastguard Worker }
1314*4bdc9457SAndroid Build Coastguard Worker }
1315*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3_varying_input_width)1316*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3_varying_input_width) {
1317*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1318*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
1319*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1320*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1321*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
1322*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1323*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1324*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
1325*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1326*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
1327*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1328*4bdc9457SAndroid Build Coastguard Worker }
1329*4bdc9457SAndroid Build Coastguard Worker }
1330*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3_varying_input_channels)1331*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3_varying_input_channels) {
1332*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1333*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
1334*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1335*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1336*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1337*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1338*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1339*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
1340*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1341*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
1342*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1343*4bdc9457SAndroid Build Coastguard Worker }
1344*4bdc9457SAndroid Build Coastguard Worker }
1345*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3_varying_output_channels)1346*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3_varying_output_channels) {
1347*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1348*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qs8.gemm.nr * 2; output_channels *= 2) {
1349*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1350*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1351*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1352*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1353*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1354*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1355*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
1356*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
1357*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1358*4bdc9457SAndroid Build Coastguard Worker }
1359*4bdc9457SAndroid Build Coastguard Worker }
1360*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3_with_height_dilation)1361*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3_with_height_dilation) {
1362*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1363*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_height = 2; dilation_height <= 3; dilation_height++) {
1364*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1365*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1366*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1367*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1368*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1369*4bdc9457SAndroid Build Coastguard Worker .dilation_height(dilation_height)
1370*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1371*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1372*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1373*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1374*4bdc9457SAndroid Build Coastguard Worker }
1375*4bdc9457SAndroid Build Coastguard Worker }
1376*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3_with_width_dilation)1377*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3_with_width_dilation) {
1378*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1379*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) {
1380*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1381*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1382*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1383*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1384*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1385*4bdc9457SAndroid Build Coastguard Worker .dilation_width(dilation_width)
1386*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1387*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1388*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1389*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1390*4bdc9457SAndroid Build Coastguard Worker }
1391*4bdc9457SAndroid Build Coastguard Worker }
1392*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3_with_height_dilation_and_stride)1393*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3_with_height_dilation_and_stride) {
1394*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1395*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1396*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1397*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1398*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1399*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1400*4bdc9457SAndroid Build Coastguard Worker .dilation_height(3)
1401*4bdc9457SAndroid Build Coastguard Worker .stride_height(2)
1402*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1403*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1404*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1405*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1406*4bdc9457SAndroid Build Coastguard Worker }
1407*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3_with_width_dilation_and_stride)1408*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3_with_width_dilation_and_stride) {
1409*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1410*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1411*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1412*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1413*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1414*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1415*4bdc9457SAndroid Build Coastguard Worker .dilation_width(3)
1416*4bdc9457SAndroid Build Coastguard Worker .stride_width(2)
1417*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1418*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1419*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1420*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1421*4bdc9457SAndroid Build Coastguard Worker }
1422*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3_with_input_stride)1423*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3_with_input_stride) {
1424*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1425*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1426*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1427*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1428*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1429*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1430*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1431*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1432*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
1433*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1434*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1435*4bdc9457SAndroid Build Coastguard Worker }
1436*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3_with_output_stride)1437*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3_with_output_stride) {
1438*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1439*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1440*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1441*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1442*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1443*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1444*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1445*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1446*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qs8.gemm.nr * 2 + 13)
1447*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1448*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1449*4bdc9457SAndroid Build Coastguard Worker }
1450*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3_with_qmin)1451*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3_with_qmin) {
1452*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1453*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1454*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1455*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1456*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1457*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1458*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1459*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1460*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
1461*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1462*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1463*4bdc9457SAndroid Build Coastguard Worker }
1464*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3_with_qmax)1465*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3_with_qmax) {
1466*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1467*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1468*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1469*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1470*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1471*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1472*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1473*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1474*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
1475*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1476*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1477*4bdc9457SAndroid Build Coastguard Worker }
1478*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3_without_bias)1479*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3_without_bias) {
1480*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1481*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1482*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
1483*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1484*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1485*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1486*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1487*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1488*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1489*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1490*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1491*4bdc9457SAndroid Build Coastguard Worker }
1492*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,weights_cache_batched_3x3)1493*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, weights_cache_batched_3x3) {
1494*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1495*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1496*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1497*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1498*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1499*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1500*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
1501*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1502*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
1503*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1504*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1505*4bdc9457SAndroid Build Coastguard Worker }
1506*4bdc9457SAndroid Build Coastguard Worker
1507*4bdc9457SAndroid Build Coastguard Worker /**************************** CONV path, grouped, batched ****************************/
1508*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3)1509*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3) {
1510*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1511*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1512*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1513*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1514*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1515*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1516*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1517*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
1518*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1519*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1520*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1521*4bdc9457SAndroid Build Coastguard Worker }
1522*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_Kx3)1523*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_Kx3) {
1524*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1525*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 1; kernel_height <= 4; kernel_height *= 2) {
1526*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1527*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1528*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1529*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
1530*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
1531*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1532*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
1533*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1534*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1535*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1536*4bdc9457SAndroid Build Coastguard Worker }
1537*4bdc9457SAndroid Build Coastguard Worker }
1538*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3xK)1539*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3xK) {
1540*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1541*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 1; kernel_width <= 4; kernel_width *= 2) {
1542*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1543*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1544*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1545*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
1546*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
1547*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1548*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
1549*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1550*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1551*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1552*4bdc9457SAndroid Build Coastguard Worker }
1553*4bdc9457SAndroid Build Coastguard Worker }
1554*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3_varying_height_padding)1555*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3_varying_height_padding) {
1556*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1557*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
1558*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
1559*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1560*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1561*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1562*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
1563*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
1564*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
1565*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1566*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1567*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
1568*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1569*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
1570*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1571*4bdc9457SAndroid Build Coastguard Worker }
1572*4bdc9457SAndroid Build Coastguard Worker }
1573*4bdc9457SAndroid Build Coastguard Worker }
1574*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3_varying_width_padding)1575*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3_varying_width_padding) {
1576*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1577*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
1578*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
1579*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1580*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1581*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1582*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
1583*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
1584*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
1585*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1586*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1587*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
1588*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1589*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
1590*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1591*4bdc9457SAndroid Build Coastguard Worker }
1592*4bdc9457SAndroid Build Coastguard Worker }
1593*4bdc9457SAndroid Build Coastguard Worker }
1594*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3_varying_height_adjustment)1595*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3_varying_height_adjustment) {
1596*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1597*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 1; adjustment_height <= 2; adjustment_height++) {
1598*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1599*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1600*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1601*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1602*4bdc9457SAndroid Build Coastguard Worker .stride_height(adjustment_height + 1)
1603*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
1604*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1605*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1606*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
1607*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1608*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
1609*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1610*4bdc9457SAndroid Build Coastguard Worker }
1611*4bdc9457SAndroid Build Coastguard Worker }
1612*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3_varying_width_adjustment)1613*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3_varying_width_adjustment) {
1614*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1615*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 1; adjustment_width <= 2; adjustment_width++) {
1616*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1617*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1618*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1619*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1620*4bdc9457SAndroid Build Coastguard Worker .stride_width(adjustment_width + 1)
1621*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
1622*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1623*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1624*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
1625*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1626*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
1627*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1628*4bdc9457SAndroid Build Coastguard Worker }
1629*4bdc9457SAndroid Build Coastguard Worker }
1630*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3_varying_input_height)1631*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3_varying_input_height) {
1632*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1633*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
1634*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1635*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1636*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
1637*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1638*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1639*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1640*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
1641*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1642*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
1643*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1644*4bdc9457SAndroid Build Coastguard Worker }
1645*4bdc9457SAndroid Build Coastguard Worker }
1646*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3_varying_input_width)1647*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3_varying_input_width) {
1648*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1649*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
1650*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1651*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1652*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
1653*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1654*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1655*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1656*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
1657*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1658*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
1659*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1660*4bdc9457SAndroid Build Coastguard Worker }
1661*4bdc9457SAndroid Build Coastguard Worker }
1662*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3_varying_input_channels)1663*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3_varying_input_channels) {
1664*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1665*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
1666*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1667*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1668*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1669*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1670*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1671*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1672*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
1673*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1674*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
1675*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1676*4bdc9457SAndroid Build Coastguard Worker }
1677*4bdc9457SAndroid Build Coastguard Worker }
1678*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3_varying_output_channels)1679*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3_varying_output_channels) {
1680*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1681*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qs8.gemm.nr * 2; output_channels *= 2) {
1682*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1683*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1684*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1685*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1686*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1687*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1688*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1689*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
1690*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
1691*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1692*4bdc9457SAndroid Build Coastguard Worker }
1693*4bdc9457SAndroid Build Coastguard Worker }
1694*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3_with_height_dilation)1695*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3_with_height_dilation) {
1696*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1697*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_height = 2; dilation_height <= 3; dilation_height++) {
1698*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1699*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1700*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1701*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1702*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1703*4bdc9457SAndroid Build Coastguard Worker .dilation_height(dilation_height)
1704*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1705*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1706*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1707*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1708*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1709*4bdc9457SAndroid Build Coastguard Worker }
1710*4bdc9457SAndroid Build Coastguard Worker }
1711*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3_with_width_dilation)1712*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3_with_width_dilation) {
1713*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1714*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) {
1715*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1716*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1717*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1718*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1719*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1720*4bdc9457SAndroid Build Coastguard Worker .dilation_width(dilation_width)
1721*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1722*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1723*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1724*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1725*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1726*4bdc9457SAndroid Build Coastguard Worker }
1727*4bdc9457SAndroid Build Coastguard Worker }
1728*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3_with_height_dilation_and_stride)1729*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3_with_height_dilation_and_stride) {
1730*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1731*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1732*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1733*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1734*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1735*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1736*4bdc9457SAndroid Build Coastguard Worker .dilation_height(3)
1737*4bdc9457SAndroid Build Coastguard Worker .stride_width(2)
1738*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1739*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1740*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1741*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1742*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1743*4bdc9457SAndroid Build Coastguard Worker }
1744*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3_with_width_dilation_and_stride)1745*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3_with_width_dilation_and_stride) {
1746*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1747*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1748*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1749*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1750*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1751*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1752*4bdc9457SAndroid Build Coastguard Worker .dilation_width(3)
1753*4bdc9457SAndroid Build Coastguard Worker .stride_width(2)
1754*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1755*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1756*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1757*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1758*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1759*4bdc9457SAndroid Build Coastguard Worker }
1760*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3_with_input_stride)1761*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3_with_input_stride) {
1762*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1763*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1764*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1765*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1766*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1767*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1768*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1769*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1770*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1771*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(47)
1772*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1773*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1774*4bdc9457SAndroid Build Coastguard Worker }
1775*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3_with_output_stride)1776*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3_with_output_stride) {
1777*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1778*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1779*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1780*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1781*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1782*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1783*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1784*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1785*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr + 3)
1786*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qs8.gemm.nr * 2 + 13)
1787*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1788*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1789*4bdc9457SAndroid Build Coastguard Worker }
1790*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3_with_qmin)1791*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3_with_qmin) {
1792*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1793*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1794*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1795*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1796*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1797*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1798*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1799*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1800*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1801*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
1802*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1803*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1804*4bdc9457SAndroid Build Coastguard Worker }
1805*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3_with_qmax)1806*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3_with_qmax) {
1807*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1808*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1809*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1810*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1811*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1812*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1813*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1814*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1815*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1816*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
1817*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1818*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1819*4bdc9457SAndroid Build Coastguard Worker }
1820*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3_without_bias)1821*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3_without_bias) {
1822*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1823*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1824*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
1825*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1826*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1827*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1828*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1829*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1830*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
1831*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1832*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1833*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1834*4bdc9457SAndroid Build Coastguard Worker }
1835*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,weights_cache_batched_grouped_3x3)1836*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, weights_cache_batched_grouped_3x3) {
1837*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1838*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1839*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1840*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1841*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1842*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1843*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1844*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
1845*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1846*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
1847*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1848*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1849*4bdc9457SAndroid Build Coastguard Worker }
1850*4bdc9457SAndroid Build Coastguard Worker
1851*4bdc9457SAndroid Build Coastguard Worker /**************************** CONV path, setup ****************************/
1852*4bdc9457SAndroid Build Coastguard Worker
1853*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3_setup_changing_batch) {
1854*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1855*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1856*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1857*4bdc9457SAndroid Build Coastguard Worker .next_batch_size(5)
1858*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1859*4bdc9457SAndroid Build Coastguard Worker .kernel_height(3)
1860*4bdc9457SAndroid Build Coastguard Worker .kernel_width(5)
1861*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1862*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
1863*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
1864*4bdc9457SAndroid Build Coastguard Worker .TestSetupQS8();
1865*4bdc9457SAndroid Build Coastguard Worker }
1866*4bdc9457SAndroid Build Coastguard Worker
1867*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3_setup_changing_height) {
1868*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1869*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1870*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1871*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1872*4bdc9457SAndroid Build Coastguard Worker .next_input_height(kUnstridedInputHeight + 3)
1873*4bdc9457SAndroid Build Coastguard Worker .kernel_height(3)
1874*4bdc9457SAndroid Build Coastguard Worker .kernel_width(5)
1875*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1876*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
1877*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
1878*4bdc9457SAndroid Build Coastguard Worker .TestSetupQS8();
1879*4bdc9457SAndroid Build Coastguard Worker }
1880*4bdc9457SAndroid Build Coastguard Worker
1881*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3_setup_changing_width) {
1882*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1883*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1884*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
1885*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
1886*4bdc9457SAndroid Build Coastguard Worker .next_input_width(kUnstridedInputWidth + 3)
1887*4bdc9457SAndroid Build Coastguard Worker .kernel_height(3)
1888*4bdc9457SAndroid Build Coastguard Worker .kernel_width(5)
1889*4bdc9457SAndroid Build Coastguard Worker .groups(2)
1890*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
1891*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
1892*4bdc9457SAndroid Build Coastguard Worker .TestSetupQS8();
1893*4bdc9457SAndroid Build Coastguard Worker }
1894*4bdc9457SAndroid Build Coastguard Worker
1895*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/IGEMM path ****************************/
1896*4bdc9457SAndroid Build Coastguard Worker
1897*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3s2) {
1898*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1899*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1900*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
1901*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1902*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1903*4bdc9457SAndroid Build Coastguard Worker .stride(2)
1904*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
1905*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1906*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1907*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1908*4bdc9457SAndroid Build Coastguard Worker }
1909*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,Kx3s2)1910*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, Kx3s2) {
1911*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1912*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 2; kernel_height <= 5; kernel_height++) {
1913*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1914*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
1915*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
1916*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
1917*4bdc9457SAndroid Build Coastguard Worker .stride(2)
1918*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
1919*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1920*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1921*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1922*4bdc9457SAndroid Build Coastguard Worker }
1923*4bdc9457SAndroid Build Coastguard Worker }
1924*4bdc9457SAndroid Build Coastguard Worker
1925*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3xKs2) {
1926*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1927*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 2; kernel_width <= 5; kernel_width++) {
1928*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1929*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
1930*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
1931*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
1932*4bdc9457SAndroid Build Coastguard Worker .stride(2)
1933*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
1934*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1935*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1936*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1937*4bdc9457SAndroid Build Coastguard Worker }
1938*4bdc9457SAndroid Build Coastguard Worker }
1939*4bdc9457SAndroid Build Coastguard Worker
1940*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3sSx1) {
1941*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1942*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_height = 2; stride_height <= 3; stride_height++) {
1943*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1944*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
1945*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1946*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
1947*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1948*4bdc9457SAndroid Build Coastguard Worker .stride_height(stride_height)
1949*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
1950*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1951*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1952*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1953*4bdc9457SAndroid Build Coastguard Worker }
1954*4bdc9457SAndroid Build Coastguard Worker }
1955*4bdc9457SAndroid Build Coastguard Worker
1956*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3s1xS) {
1957*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1958*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_width = 2; stride_width <= 3; stride_width++) {
1959*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1960*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
1961*4bdc9457SAndroid Build Coastguard Worker .padding(1)
1962*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
1963*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1964*4bdc9457SAndroid Build Coastguard Worker .stride_width(stride_width)
1965*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
1966*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1967*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
1968*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1969*4bdc9457SAndroid Build Coastguard Worker }
1970*4bdc9457SAndroid Build Coastguard Worker }
1971*4bdc9457SAndroid Build Coastguard Worker
1972*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3s2_varying_height_padding) {
1973*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1974*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
1975*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
1976*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1977*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
1978*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
1979*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
1980*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
1981*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
1982*4bdc9457SAndroid Build Coastguard Worker .stride(2)
1983*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
1984*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
1985*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
1986*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
1987*4bdc9457SAndroid Build Coastguard Worker }
1988*4bdc9457SAndroid Build Coastguard Worker }
1989*4bdc9457SAndroid Build Coastguard Worker }
1990*4bdc9457SAndroid Build Coastguard Worker
1991*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3s2_varying_width_padding) {
1992*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
1993*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
1994*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
1995*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
1996*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
1997*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
1998*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
1999*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
2000*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2001*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2002*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
2003*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2004*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2005*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2006*4bdc9457SAndroid Build Coastguard Worker }
2007*4bdc9457SAndroid Build Coastguard Worker }
2008*4bdc9457SAndroid Build Coastguard Worker }
2009*4bdc9457SAndroid Build Coastguard Worker
2010*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3s2_varying_height_adjustment) {
2011*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2012*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 0; adjustment_height <= 1; adjustment_height++) {
2013*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2014*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2015*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2016*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
2017*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2018*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2019*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
2020*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2021*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2022*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2023*4bdc9457SAndroid Build Coastguard Worker }
2024*4bdc9457SAndroid Build Coastguard Worker }
2025*4bdc9457SAndroid Build Coastguard Worker
2026*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3s2_varying_width_adjustment) {
2027*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2028*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 0; adjustment_width <= 1; adjustment_width++) {
2029*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2030*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2031*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2032*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
2033*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2034*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2035*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
2036*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2037*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2038*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2039*4bdc9457SAndroid Build Coastguard Worker }
2040*4bdc9457SAndroid Build Coastguard Worker }
2041*4bdc9457SAndroid Build Coastguard Worker
2042*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3s2_varying_input_height) {
2043*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2044*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
2045*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2046*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
2047*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2048*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2049*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2050*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
2051*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2052*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2053*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2054*4bdc9457SAndroid Build Coastguard Worker }
2055*4bdc9457SAndroid Build Coastguard Worker }
2056*4bdc9457SAndroid Build Coastguard Worker
2057*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3s2_varying_input_width) {
2058*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2059*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
2060*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2061*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2062*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2063*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2064*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2065*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
2066*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2067*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2068*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2069*4bdc9457SAndroid Build Coastguard Worker }
2070*4bdc9457SAndroid Build Coastguard Worker }
2071*4bdc9457SAndroid Build Coastguard Worker
2072*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3s2_varying_input_channels) {
2073*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2074*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
2075*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2076*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2077*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2078*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2079*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2080*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
2081*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2082*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2083*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2084*4bdc9457SAndroid Build Coastguard Worker }
2085*4bdc9457SAndroid Build Coastguard Worker }
2086*4bdc9457SAndroid Build Coastguard Worker
2087*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3s2_varying_output_channels) {
2088*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2089*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qs8.gemm.nr * 2; output_channels *= 2) {
2090*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2091*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2092*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2093*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2094*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2095*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
2096*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
2097*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2098*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2099*4bdc9457SAndroid Build Coastguard Worker }
2100*4bdc9457SAndroid Build Coastguard Worker }
2101*4bdc9457SAndroid Build Coastguard Worker
2102*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3s2_with_input_stride) {
2103*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2104*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2105*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2106*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2107*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2108*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2109*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
2110*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2111*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
2112*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2113*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2114*4bdc9457SAndroid Build Coastguard Worker }
2115*4bdc9457SAndroid Build Coastguard Worker
2116*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3s2_with_output_stride) {
2117*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2118*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2119*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2120*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2121*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2122*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2123*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
2124*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2125*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qs8.gemm.nr * 2 + 13)
2126*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2127*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2128*4bdc9457SAndroid Build Coastguard Worker }
2129*4bdc9457SAndroid Build Coastguard Worker
2130*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3s2_with_qmin) {
2131*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2132*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2133*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2134*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2135*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2136*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2137*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
2138*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2139*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
2140*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2141*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2142*4bdc9457SAndroid Build Coastguard Worker }
2143*4bdc9457SAndroid Build Coastguard Worker
2144*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3s2_with_qmax) {
2145*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2146*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2147*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2148*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2149*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2150*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2151*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
2152*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2153*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
2154*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2155*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2156*4bdc9457SAndroid Build Coastguard Worker }
2157*4bdc9457SAndroid Build Coastguard Worker
2158*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3s2_without_bias) {
2159*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2160*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2161*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
2162*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2163*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2164*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2165*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2166*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
2167*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2168*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2169*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2170*4bdc9457SAndroid Build Coastguard Worker }
2171*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,weights_cache_3x3s2)2172*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, weights_cache_3x3s2) {
2173*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2174*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2175*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2176*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2177*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2178*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2179*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
2180*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2181*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
2182*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2183*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2184*4bdc9457SAndroid Build Coastguard Worker }
2185*4bdc9457SAndroid Build Coastguard Worker
2186*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/IGEMM path, grouped ****************************/
2187*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3s2)2188*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3s2) {
2189*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2190*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2191*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2192*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2193*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2194*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2195*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2196*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2197*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2198*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2199*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2200*4bdc9457SAndroid Build Coastguard Worker }
2201*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_Kx3s2)2202*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_Kx3s2) {
2203*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2204*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 2; kernel_height <= 5; kernel_height++) {
2205*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2206*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2207*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
2208*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
2209*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2210*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2211*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2212*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2213*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2214*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2215*4bdc9457SAndroid Build Coastguard Worker }
2216*4bdc9457SAndroid Build Coastguard Worker }
2217*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3xKs2)2218*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3xKs2) {
2219*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2220*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 2; kernel_width <= 5; kernel_width++) {
2221*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2222*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2223*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
2224*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
2225*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2226*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2227*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2228*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2229*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2230*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2231*4bdc9457SAndroid Build Coastguard Worker }
2232*4bdc9457SAndroid Build Coastguard Worker }
2233*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3sSx1)2234*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3sSx1) {
2235*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2236*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_height = 2; stride_height <= 3; stride_height++) {
2237*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2238*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2239*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2240*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
2241*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2242*4bdc9457SAndroid Build Coastguard Worker .stride_height(stride_height)
2243*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2244*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2245*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2246*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2247*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2248*4bdc9457SAndroid Build Coastguard Worker }
2249*4bdc9457SAndroid Build Coastguard Worker }
2250*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3s1xS)2251*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3s1xS) {
2252*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2253*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_width = 2; stride_width <= 3; stride_width++) {
2254*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2255*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2256*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2257*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
2258*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2259*4bdc9457SAndroid Build Coastguard Worker .stride_width(stride_width)
2260*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2261*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2262*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2263*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2264*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2265*4bdc9457SAndroid Build Coastguard Worker }
2266*4bdc9457SAndroid Build Coastguard Worker }
2267*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3s2_varying_height_padding)2268*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3s2_varying_height_padding) {
2269*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2270*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
2271*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
2272*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2273*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2274*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
2275*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
2276*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
2277*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2278*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2279*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2280*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2281*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2282*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2283*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2284*4bdc9457SAndroid Build Coastguard Worker }
2285*4bdc9457SAndroid Build Coastguard Worker }
2286*4bdc9457SAndroid Build Coastguard Worker }
2287*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3s2_varying_width_padding)2288*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3s2_varying_width_padding) {
2289*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2290*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
2291*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
2292*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2293*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2294*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
2295*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
2296*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
2297*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2298*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2299*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2300*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2301*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2302*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2303*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2304*4bdc9457SAndroid Build Coastguard Worker }
2305*4bdc9457SAndroid Build Coastguard Worker }
2306*4bdc9457SAndroid Build Coastguard Worker }
2307*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3s2_varying_height_adjustment)2308*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3s2_varying_height_adjustment) {
2309*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2310*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 0; adjustment_height <= 1; adjustment_height++) {
2311*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2312*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2313*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2314*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
2315*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2316*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2317*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2318*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2319*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2320*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2321*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2322*4bdc9457SAndroid Build Coastguard Worker }
2323*4bdc9457SAndroid Build Coastguard Worker }
2324*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3s2_varying_width_adjustment)2325*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3s2_varying_width_adjustment) {
2326*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2327*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 0; adjustment_width <= 1; adjustment_width++) {
2328*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2329*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2330*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2331*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
2332*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2333*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2334*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2335*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2336*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2337*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2338*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2339*4bdc9457SAndroid Build Coastguard Worker }
2340*4bdc9457SAndroid Build Coastguard Worker }
2341*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3s2_varying_input_height)2342*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3s2_varying_input_height) {
2343*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2344*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
2345*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2346*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
2347*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2348*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2349*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2350*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2351*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2352*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2353*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2354*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2355*4bdc9457SAndroid Build Coastguard Worker }
2356*4bdc9457SAndroid Build Coastguard Worker }
2357*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3s2_varying_input_width)2358*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3s2_varying_input_width) {
2359*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2360*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
2361*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2362*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2363*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2364*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2365*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2366*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2367*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2368*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2369*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2370*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2371*4bdc9457SAndroid Build Coastguard Worker }
2372*4bdc9457SAndroid Build Coastguard Worker }
2373*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3s2_varying_input_channels)2374*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3s2_varying_input_channels) {
2375*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2376*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 14; input_channels <= 20; input_channels++) {
2377*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2378*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2379*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2380*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2381*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2382*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2383*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
2384*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2385*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2386*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2387*4bdc9457SAndroid Build Coastguard Worker }
2388*4bdc9457SAndroid Build Coastguard Worker }
2389*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3s2_varying_output_channels)2390*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3s2_varying_output_channels) {
2391*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2392*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qs8.gemm.nr * 2; output_channels *= 2) {
2393*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2394*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2395*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2396*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2397*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2398*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2399*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2400*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
2401*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2402*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2403*4bdc9457SAndroid Build Coastguard Worker }
2404*4bdc9457SAndroid Build Coastguard Worker }
2405*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3s2_with_input_stride)2406*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3s2_with_input_stride) {
2407*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2408*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2409*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2410*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2411*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2412*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2413*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2414*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2415*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2416*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(37)
2417*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2418*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2419*4bdc9457SAndroid Build Coastguard Worker }
2420*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3s2_with_output_stride)2421*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3s2_with_output_stride) {
2422*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2423*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2424*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2425*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2426*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2427*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2428*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2429*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2430*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr + 3)
2431*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qs8.gemm.nr * 2 + 13)
2432*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2433*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2434*4bdc9457SAndroid Build Coastguard Worker }
2435*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3s2_with_qmin)2436*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3s2_with_qmin) {
2437*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2438*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2439*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2440*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2441*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2442*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2443*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2444*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2445*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2446*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
2447*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2448*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2449*4bdc9457SAndroid Build Coastguard Worker }
2450*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3s2_with_qmax)2451*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3s2_with_qmax) {
2452*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2453*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2454*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2455*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2456*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2457*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2458*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2459*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2460*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2461*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
2462*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2463*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2464*4bdc9457SAndroid Build Coastguard Worker }
2465*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_3x3s2_without_bias)2466*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_3x3s2_without_bias) {
2467*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2468*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2469*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
2470*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2471*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2472*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2473*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2474*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2475*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2476*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2477*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2478*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2479*4bdc9457SAndroid Build Coastguard Worker }
2480*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,weights_cache_grouped_3x3s2)2481*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, weights_cache_grouped_3x3s2) {
2482*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2483*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2484*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2485*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2486*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2487*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2488*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2489*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2490*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2491*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
2492*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2493*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2494*4bdc9457SAndroid Build Coastguard Worker }
2495*4bdc9457SAndroid Build Coastguard Worker
2496*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/IGEMM path, batched ****************************/
2497*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3s2)2498*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3s2) {
2499*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2500*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2501*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2502*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2503*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2504*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2505*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2506*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
2507*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2508*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2509*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2510*4bdc9457SAndroid Build Coastguard Worker }
2511*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_Kx3s2)2512*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_Kx3s2) {
2513*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2514*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 2; kernel_height <= 5; kernel_height++) {
2515*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2516*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2517*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2518*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
2519*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
2520*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2521*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2522*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2523*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2524*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2525*4bdc9457SAndroid Build Coastguard Worker }
2526*4bdc9457SAndroid Build Coastguard Worker }
2527*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3xKs2)2528*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3xKs2) {
2529*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2530*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 2; kernel_width <= 5; kernel_width++) {
2531*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2532*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2533*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2534*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
2535*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
2536*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2537*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2538*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2539*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2540*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2541*4bdc9457SAndroid Build Coastguard Worker }
2542*4bdc9457SAndroid Build Coastguard Worker }
2543*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3sSx1)2544*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3sSx1) {
2545*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2546*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_height = 2; stride_height <= 3; stride_height++) {
2547*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2548*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2549*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2550*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2551*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
2552*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2553*4bdc9457SAndroid Build Coastguard Worker .stride_height(stride_height)
2554*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2555*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2556*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2557*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2558*4bdc9457SAndroid Build Coastguard Worker }
2559*4bdc9457SAndroid Build Coastguard Worker }
2560*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3s1xS)2561*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3s1xS) {
2562*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2563*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_width = 2; stride_width <= 3; stride_width++) {
2564*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2565*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2566*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2567*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2568*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
2569*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2570*4bdc9457SAndroid Build Coastguard Worker .stride_width(stride_width)
2571*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2572*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2573*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2574*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2575*4bdc9457SAndroid Build Coastguard Worker }
2576*4bdc9457SAndroid Build Coastguard Worker }
2577*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3s2_varying_height_padding)2578*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3s2_varying_height_padding) {
2579*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2580*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
2581*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
2582*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2583*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2584*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2585*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
2586*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
2587*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
2588*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2589*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2590*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
2591*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2592*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2593*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2594*4bdc9457SAndroid Build Coastguard Worker }
2595*4bdc9457SAndroid Build Coastguard Worker }
2596*4bdc9457SAndroid Build Coastguard Worker }
2597*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3s2_varying_width_padding)2598*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3s2_varying_width_padding) {
2599*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2600*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
2601*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
2602*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2603*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2604*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2605*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
2606*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
2607*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
2608*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2609*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2610*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
2611*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2612*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2613*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2614*4bdc9457SAndroid Build Coastguard Worker }
2615*4bdc9457SAndroid Build Coastguard Worker }
2616*4bdc9457SAndroid Build Coastguard Worker }
2617*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3s2_varying_height_adjustment)2618*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3s2_varying_height_adjustment) {
2619*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2620*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 0; adjustment_height <= 1; adjustment_height++) {
2621*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2622*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2623*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2624*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2625*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
2626*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2627*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2628*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
2629*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2630*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2631*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2632*4bdc9457SAndroid Build Coastguard Worker }
2633*4bdc9457SAndroid Build Coastguard Worker }
2634*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3s2_varying_width_adjustment)2635*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3s2_varying_width_adjustment) {
2636*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2637*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 0; adjustment_width <= 1; adjustment_width++) {
2638*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2639*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2640*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2641*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2642*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
2643*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2644*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2645*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
2646*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2647*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2648*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2649*4bdc9457SAndroid Build Coastguard Worker }
2650*4bdc9457SAndroid Build Coastguard Worker }
2651*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3s2_varying_input_height)2652*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3s2_varying_input_height) {
2653*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2654*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
2655*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2656*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2657*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
2658*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2659*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2660*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2661*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
2662*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2663*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2664*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2665*4bdc9457SAndroid Build Coastguard Worker }
2666*4bdc9457SAndroid Build Coastguard Worker }
2667*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3s2_varying_input_width)2668*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3s2_varying_input_width) {
2669*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2670*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
2671*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2672*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2673*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2674*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2675*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2676*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2677*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
2678*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2679*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2680*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2681*4bdc9457SAndroid Build Coastguard Worker }
2682*4bdc9457SAndroid Build Coastguard Worker }
2683*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3s2_varying_input_channels)2684*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3s2_varying_input_channels) {
2685*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2686*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
2687*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2688*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2689*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2690*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2691*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2692*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2693*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
2694*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2695*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2696*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2697*4bdc9457SAndroid Build Coastguard Worker }
2698*4bdc9457SAndroid Build Coastguard Worker }
2699*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3s2_varying_output_channels)2700*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3s2_varying_output_channels) {
2701*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2702*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qs8.gemm.nr * 2; output_channels *= 2) {
2703*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2704*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2705*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2706*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2707*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2708*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2709*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
2710*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
2711*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2712*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2713*4bdc9457SAndroid Build Coastguard Worker }
2714*4bdc9457SAndroid Build Coastguard Worker }
2715*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3s2_with_input_stride)2716*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3s2_with_input_stride) {
2717*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2718*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2719*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2720*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2721*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2722*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2723*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2724*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
2725*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2726*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
2727*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2728*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2729*4bdc9457SAndroid Build Coastguard Worker }
2730*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3s2_with_output_stride)2731*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3s2_with_output_stride) {
2732*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2733*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2734*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2735*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2736*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2737*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2738*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2739*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
2740*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2741*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qs8.gemm.nr * 2 + 13)
2742*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2743*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2744*4bdc9457SAndroid Build Coastguard Worker }
2745*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3s2_with_qmin)2746*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3s2_with_qmin) {
2747*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2748*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2749*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2750*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2751*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2752*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2753*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2754*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
2755*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2756*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
2757*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2758*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2759*4bdc9457SAndroid Build Coastguard Worker }
2760*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3s2_with_qmax)2761*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3s2_with_qmax) {
2762*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2763*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2764*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2765*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2766*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2767*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2768*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2769*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
2770*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2771*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
2772*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2773*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2774*4bdc9457SAndroid Build Coastguard Worker }
2775*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_3x3s2_without_bias)2776*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_3x3s2_without_bias) {
2777*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2778*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2779*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
2780*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2781*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2782*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2783*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2784*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2785*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
2786*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2787*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2788*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2789*4bdc9457SAndroid Build Coastguard Worker }
2790*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,weights_cache_batched_3x3s2)2791*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, weights_cache_batched_3x3s2) {
2792*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2793*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2794*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2795*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2796*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2797*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2798*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2799*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
2800*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2801*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
2802*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2803*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2804*4bdc9457SAndroid Build Coastguard Worker }
2805*4bdc9457SAndroid Build Coastguard Worker
2806*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/IGEMM path, grouped, batched ****************************/
2807*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3s2)2808*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3s2) {
2809*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2810*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2811*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2812*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2813*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2814*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2815*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2816*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2817*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2818*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2819*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2820*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2821*4bdc9457SAndroid Build Coastguard Worker }
2822*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_Kx3s2)2823*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_Kx3s2) {
2824*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2825*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 2; kernel_height <= 5; kernel_height++) {
2826*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2827*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2828*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2829*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
2830*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
2831*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2832*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2833*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2834*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2835*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2836*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2837*4bdc9457SAndroid Build Coastguard Worker }
2838*4bdc9457SAndroid Build Coastguard Worker }
2839*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3xKs2)2840*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3xKs2) {
2841*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2842*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 2; kernel_width <= 5; kernel_width++) {
2843*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2844*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2845*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2846*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
2847*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
2848*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2849*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2850*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2851*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2852*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2853*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2854*4bdc9457SAndroid Build Coastguard Worker }
2855*4bdc9457SAndroid Build Coastguard Worker }
2856*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3sSx1)2857*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3sSx1) {
2858*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2859*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_height = 2; stride_height <= 3; stride_height++) {
2860*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2861*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2862*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2863*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2864*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
2865*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2866*4bdc9457SAndroid Build Coastguard Worker .stride_height(stride_height)
2867*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2868*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2869*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2870*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2871*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2872*4bdc9457SAndroid Build Coastguard Worker }
2873*4bdc9457SAndroid Build Coastguard Worker }
2874*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3s1xS)2875*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3s1xS) {
2876*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2877*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_width = 2; stride_width <= 3; stride_width++) {
2878*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2879*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2880*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2881*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2882*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
2883*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2884*4bdc9457SAndroid Build Coastguard Worker .stride_width(stride_width)
2885*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2886*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2887*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2888*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
2889*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2890*4bdc9457SAndroid Build Coastguard Worker }
2891*4bdc9457SAndroid Build Coastguard Worker }
2892*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3s2_varying_height_padding)2893*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3s2_varying_height_padding) {
2894*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2895*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
2896*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
2897*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2898*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2899*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2900*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
2901*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
2902*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
2903*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2904*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2905*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2906*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2907*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2908*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2909*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2910*4bdc9457SAndroid Build Coastguard Worker }
2911*4bdc9457SAndroid Build Coastguard Worker }
2912*4bdc9457SAndroid Build Coastguard Worker }
2913*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3s2_varying_width_padding)2914*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3s2_varying_width_padding) {
2915*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2916*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
2917*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
2918*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2919*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2920*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2921*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
2922*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
2923*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
2924*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2925*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2926*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2927*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2928*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2929*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2930*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2931*4bdc9457SAndroid Build Coastguard Worker }
2932*4bdc9457SAndroid Build Coastguard Worker }
2933*4bdc9457SAndroid Build Coastguard Worker }
2934*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3s2_varying_height_adjustment)2935*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3s2_varying_height_adjustment) {
2936*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2937*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 0; adjustment_height <= 1; adjustment_height++) {
2938*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2939*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2940*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2941*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2942*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
2943*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2944*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2945*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2946*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2947*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2948*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2949*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2950*4bdc9457SAndroid Build Coastguard Worker }
2951*4bdc9457SAndroid Build Coastguard Worker }
2952*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3s2_varying_width_adjustment)2953*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3s2_varying_width_adjustment) {
2954*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2955*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 0; adjustment_width <= 1; adjustment_width++) {
2956*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2957*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2958*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2959*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2960*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
2961*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2962*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2963*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2964*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2965*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2966*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2967*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2968*4bdc9457SAndroid Build Coastguard Worker }
2969*4bdc9457SAndroid Build Coastguard Worker }
2970*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3s2_varying_input_height)2971*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3s2_varying_input_height) {
2972*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2973*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
2974*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2975*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2976*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
2977*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2978*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2979*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2980*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2981*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2982*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
2983*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
2984*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
2985*4bdc9457SAndroid Build Coastguard Worker }
2986*4bdc9457SAndroid Build Coastguard Worker }
2987*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3s2_varying_input_width)2988*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3s2_varying_input_width) {
2989*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
2990*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
2991*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
2992*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
2993*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
2994*4bdc9457SAndroid Build Coastguard Worker .padding(1)
2995*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
2996*4bdc9457SAndroid Build Coastguard Worker .stride(2)
2997*4bdc9457SAndroid Build Coastguard Worker .groups(2)
2998*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
2999*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3000*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3001*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3002*4bdc9457SAndroid Build Coastguard Worker }
3003*4bdc9457SAndroid Build Coastguard Worker }
3004*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3s2_varying_input_channels)3005*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3s2_varying_input_channels) {
3006*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3007*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 14; input_channels <= 20; input_channels++) {
3008*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3009*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3010*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3011*4bdc9457SAndroid Build Coastguard Worker .padding(1)
3012*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
3013*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3014*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3015*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
3016*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3017*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3018*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3019*4bdc9457SAndroid Build Coastguard Worker }
3020*4bdc9457SAndroid Build Coastguard Worker }
3021*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3s2_varying_output_channels)3022*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3s2_varying_output_channels) {
3023*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3024*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qs8.gemm.nr * 2; output_channels *= 2) {
3025*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3026*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3027*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3028*4bdc9457SAndroid Build Coastguard Worker .padding(1)
3029*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
3030*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3031*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3032*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3033*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
3034*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3035*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3036*4bdc9457SAndroid Build Coastguard Worker }
3037*4bdc9457SAndroid Build Coastguard Worker }
3038*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3s2_with_input_stride)3039*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3s2_with_input_stride) {
3040*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3041*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3042*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3043*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3044*4bdc9457SAndroid Build Coastguard Worker .padding(1)
3045*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
3046*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3047*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3048*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3049*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3050*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(37)
3051*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3052*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3053*4bdc9457SAndroid Build Coastguard Worker }
3054*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3s2_with_output_stride)3055*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3s2_with_output_stride) {
3056*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3057*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3058*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3059*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3060*4bdc9457SAndroid Build Coastguard Worker .padding(1)
3061*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
3062*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3063*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3064*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3065*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr + 3)
3066*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qs8.gemm.nr * 2 + 13)
3067*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3068*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3069*4bdc9457SAndroid Build Coastguard Worker }
3070*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3s2_with_qmin)3071*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3s2_with_qmin) {
3072*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3073*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3074*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3075*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3076*4bdc9457SAndroid Build Coastguard Worker .padding(1)
3077*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
3078*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3079*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3080*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3081*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3082*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
3083*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3084*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3085*4bdc9457SAndroid Build Coastguard Worker }
3086*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3s2_with_qmax)3087*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3s2_with_qmax) {
3088*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3089*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3090*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3091*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3092*4bdc9457SAndroid Build Coastguard Worker .padding(1)
3093*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
3094*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3095*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3096*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3097*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3098*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
3099*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3100*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3101*4bdc9457SAndroid Build Coastguard Worker }
3102*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_3x3s2_without_bias)3103*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_3x3s2_without_bias) {
3104*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3105*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3106*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
3107*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3108*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3109*4bdc9457SAndroid Build Coastguard Worker .padding(1)
3110*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
3111*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3112*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3113*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3114*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3115*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3116*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3117*4bdc9457SAndroid Build Coastguard Worker }
3118*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,weights_cache_batched_grouped_3x3s2)3119*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, weights_cache_batched_grouped_3x3s2) {
3120*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3121*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3122*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3123*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3124*4bdc9457SAndroid Build Coastguard Worker .padding(1)
3125*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
3126*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3127*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3128*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3129*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3130*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
3131*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3132*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3133*4bdc9457SAndroid Build Coastguard Worker }
3134*4bdc9457SAndroid Build Coastguard Worker
3135*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/IGEMM path, setup ****************************/
3136*4bdc9457SAndroid Build Coastguard Worker
3137*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3s2_setup_changing_batch) {
3138*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3139*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3140*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3141*4bdc9457SAndroid Build Coastguard Worker .next_batch_size(5)
3142*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3143*4bdc9457SAndroid Build Coastguard Worker .kernel_height(3)
3144*4bdc9457SAndroid Build Coastguard Worker .kernel_width(5)
3145*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3146*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3147*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
3148*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
3149*4bdc9457SAndroid Build Coastguard Worker .TestSetupQS8();
3150*4bdc9457SAndroid Build Coastguard Worker }
3151*4bdc9457SAndroid Build Coastguard Worker
3152*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3s2_setup_changing_height) {
3153*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3154*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3155*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3156*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3157*4bdc9457SAndroid Build Coastguard Worker .next_input_height(kStridedInputHeight + 3)
3158*4bdc9457SAndroid Build Coastguard Worker .kernel_height(3)
3159*4bdc9457SAndroid Build Coastguard Worker .kernel_width(5)
3160*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3161*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3162*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
3163*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
3164*4bdc9457SAndroid Build Coastguard Worker .TestSetupQS8();
3165*4bdc9457SAndroid Build Coastguard Worker }
3166*4bdc9457SAndroid Build Coastguard Worker
3167*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 3x3s2_setup_changing_width) {
3168*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3169*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3170*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3171*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3172*4bdc9457SAndroid Build Coastguard Worker .next_input_width(kStridedInputWidth + 3)
3173*4bdc9457SAndroid Build Coastguard Worker .kernel_height(3)
3174*4bdc9457SAndroid Build Coastguard Worker .kernel_width(5)
3175*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3176*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3177*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
3178*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
3179*4bdc9457SAndroid Build Coastguard Worker .TestSetupQS8();
3180*4bdc9457SAndroid Build Coastguard Worker }
3181*4bdc9457SAndroid Build Coastguard Worker
3182*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/GEMM path ****************************/
3183*4bdc9457SAndroid Build Coastguard Worker
3184*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 2x2s2) {
3185*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3186*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3187*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3188*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3189*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3190*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
3191*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3192*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3193*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3194*4bdc9457SAndroid Build Coastguard Worker }
3195*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,Kx2sKx2)3196*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, Kx2sKx2) {
3197*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3198*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 3; kernel_height <= 5; kernel_height++) {
3199*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3200*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3201*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 2)
3202*4bdc9457SAndroid Build Coastguard Worker .stride(kernel_height, 2)
3203*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3204*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3205*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3206*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3207*4bdc9457SAndroid Build Coastguard Worker }
3208*4bdc9457SAndroid Build Coastguard Worker }
3209*4bdc9457SAndroid Build Coastguard Worker
3210*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 2xKs2xK) {
3211*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3212*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 3; kernel_width <= 5; kernel_width++) {
3213*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3214*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3215*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, kernel_width)
3216*4bdc9457SAndroid Build Coastguard Worker .stride(2, kernel_width)
3217*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3218*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3219*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3220*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3221*4bdc9457SAndroid Build Coastguard Worker }
3222*4bdc9457SAndroid Build Coastguard Worker }
3223*4bdc9457SAndroid Build Coastguard Worker
3224*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 2x2s2_height_adjustment) {
3225*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3226*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3227*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3228*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(1)
3229*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3230*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3231*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
3232*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3233*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3234*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3235*4bdc9457SAndroid Build Coastguard Worker }
3236*4bdc9457SAndroid Build Coastguard Worker
3237*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 2x2s2_width_adjustment) {
3238*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3239*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3240*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3241*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(1)
3242*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3243*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3244*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
3245*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3246*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3247*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3248*4bdc9457SAndroid Build Coastguard Worker }
3249*4bdc9457SAndroid Build Coastguard Worker
3250*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 2x2s2_varying_input_height) {
3251*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3252*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
3253*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3254*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
3255*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3256*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3257*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
3258*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3259*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3260*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3261*4bdc9457SAndroid Build Coastguard Worker }
3262*4bdc9457SAndroid Build Coastguard Worker }
3263*4bdc9457SAndroid Build Coastguard Worker
3264*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 2x2s2_varying_input_width) {
3265*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3266*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
3267*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3268*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3269*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3270*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3271*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
3272*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3273*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3274*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3275*4bdc9457SAndroid Build Coastguard Worker }
3276*4bdc9457SAndroid Build Coastguard Worker }
3277*4bdc9457SAndroid Build Coastguard Worker
3278*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 2x2s2_varying_input_channels) {
3279*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3280*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
3281*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3282*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3283*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3284*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3285*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
3286*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3287*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3288*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3289*4bdc9457SAndroid Build Coastguard Worker }
3290*4bdc9457SAndroid Build Coastguard Worker }
3291*4bdc9457SAndroid Build Coastguard Worker
3292*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 2x2s2_varying_output_channels) {
3293*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3294*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qs8.gemm.nr * 2; output_channels *= 2) {
3295*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3296*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3297*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3298*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3299*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
3300*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
3301*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3302*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3303*4bdc9457SAndroid Build Coastguard Worker }
3304*4bdc9457SAndroid Build Coastguard Worker }
3305*4bdc9457SAndroid Build Coastguard Worker
3306*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 2x2s2_with_input_stride) {
3307*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3308*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3309*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3310*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3311*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3312*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
3313*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3314*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
3315*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3316*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3317*4bdc9457SAndroid Build Coastguard Worker }
3318*4bdc9457SAndroid Build Coastguard Worker
3319*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 2x2s2_with_output_stride) {
3320*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3321*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3322*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3323*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3324*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3325*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
3326*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3327*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qs8.gemm.nr * 2 + 13)
3328*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3329*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3330*4bdc9457SAndroid Build Coastguard Worker }
3331*4bdc9457SAndroid Build Coastguard Worker
3332*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 2x2s2_with_qmin) {
3333*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3334*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3335*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3336*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3337*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3338*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
3339*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3340*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
3341*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3342*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3343*4bdc9457SAndroid Build Coastguard Worker }
3344*4bdc9457SAndroid Build Coastguard Worker
3345*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 2x2s2_with_qmax) {
3346*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3347*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3348*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3349*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3350*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3351*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
3352*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3353*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
3354*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3355*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3356*4bdc9457SAndroid Build Coastguard Worker }
3357*4bdc9457SAndroid Build Coastguard Worker
3358*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 2x2s2_without_bias) {
3359*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3360*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3361*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
3362*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3363*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3364*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3365*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
3366*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3367*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3368*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3369*4bdc9457SAndroid Build Coastguard Worker }
3370*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,weights_cache_2x2s2)3371*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, weights_cache_2x2s2) {
3372*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3373*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3374*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3375*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3376*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3377*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
3378*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3379*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
3380*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3381*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3382*4bdc9457SAndroid Build Coastguard Worker }
3383*4bdc9457SAndroid Build Coastguard Worker
3384*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/GEMM path, grouped ****************************/
3385*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_2x2s2)3386*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_2x2s2) {
3387*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3388*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3389*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3390*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3391*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3392*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3393*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3394*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3395*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3396*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3397*4bdc9457SAndroid Build Coastguard Worker }
3398*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_Kx2sKx2)3399*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_Kx2sKx2) {
3400*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3401*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 3; kernel_height <= 5; kernel_height++) {
3402*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3403*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3404*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 2)
3405*4bdc9457SAndroid Build Coastguard Worker .stride(kernel_height, 2)
3406*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3407*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3408*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3409*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3410*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3411*4bdc9457SAndroid Build Coastguard Worker }
3412*4bdc9457SAndroid Build Coastguard Worker }
3413*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_2xKs2xK)3414*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_2xKs2xK) {
3415*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3416*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 3; kernel_width <= 5; kernel_width++) {
3417*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3418*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3419*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, kernel_width)
3420*4bdc9457SAndroid Build Coastguard Worker .stride(2, kernel_width)
3421*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3422*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3423*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3424*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3425*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3426*4bdc9457SAndroid Build Coastguard Worker }
3427*4bdc9457SAndroid Build Coastguard Worker }
3428*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_2x2s2_height_adjustment)3429*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_2x2s2_height_adjustment) {
3430*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3431*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3432*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3433*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(1)
3434*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3435*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3436*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3437*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3438*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3439*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3440*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3441*4bdc9457SAndroid Build Coastguard Worker }
3442*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_2x2s2_width_adjustment)3443*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_2x2s2_width_adjustment) {
3444*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3445*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3446*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3447*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(1)
3448*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3449*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3450*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3451*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3452*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3453*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3454*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3455*4bdc9457SAndroid Build Coastguard Worker }
3456*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_2x2s2_varying_input_height)3457*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_2x2s2_varying_input_height) {
3458*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3459*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
3460*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3461*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
3462*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3463*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3464*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3465*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3466*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3467*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3468*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3469*4bdc9457SAndroid Build Coastguard Worker }
3470*4bdc9457SAndroid Build Coastguard Worker }
3471*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_2x2s2_varying_input_width)3472*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_2x2s2_varying_input_width) {
3473*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3474*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
3475*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3476*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3477*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3478*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3479*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3480*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3481*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3482*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3483*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3484*4bdc9457SAndroid Build Coastguard Worker }
3485*4bdc9457SAndroid Build Coastguard Worker }
3486*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_2x2s2_varying_input_channels)3487*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_2x2s2_varying_input_channels) {
3488*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3489*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 14; input_channels <= 20; input_channels++) {
3490*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3491*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3492*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3493*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3494*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3495*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
3496*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3497*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3498*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3499*4bdc9457SAndroid Build Coastguard Worker }
3500*4bdc9457SAndroid Build Coastguard Worker }
3501*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_2x2s2_varying_output_channels)3502*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_2x2s2_varying_output_channels) {
3503*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3504*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qs8.gemm.nr * 2; output_channels *= 2) {
3505*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3506*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3507*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3508*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3509*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3510*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3511*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
3512*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3513*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3514*4bdc9457SAndroid Build Coastguard Worker }
3515*4bdc9457SAndroid Build Coastguard Worker }
3516*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_2x2s2_with_input_stride)3517*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_2x2s2_with_input_stride) {
3518*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3519*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3520*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3521*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3522*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3523*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3524*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3525*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3526*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(37)
3527*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3528*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3529*4bdc9457SAndroid Build Coastguard Worker }
3530*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_2x2s2_with_output_stride)3531*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_2x2s2_with_output_stride) {
3532*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3533*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3534*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3535*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3536*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3537*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3538*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3539*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr + 3)
3540*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qs8.gemm.nr * 2 + 13)
3541*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3542*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3543*4bdc9457SAndroid Build Coastguard Worker }
3544*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_2x2s2_with_qmin)3545*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_2x2s2_with_qmin) {
3546*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3547*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3548*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3549*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3550*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3551*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3552*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3553*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3554*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
3555*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3556*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3557*4bdc9457SAndroid Build Coastguard Worker }
3558*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_2x2s2_with_qmax)3559*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_2x2s2_with_qmax) {
3560*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3561*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3562*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3563*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3564*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3565*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3566*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3567*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3568*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
3569*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3570*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3571*4bdc9457SAndroid Build Coastguard Worker }
3572*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,grouped_2x2s2_without_bias)3573*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, grouped_2x2s2_without_bias) {
3574*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3575*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3576*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
3577*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3578*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3579*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3580*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3581*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3582*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3583*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3584*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3585*4bdc9457SAndroid Build Coastguard Worker }
3586*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,weights_cache_grouped_2x2s2)3587*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, weights_cache_grouped_2x2s2) {
3588*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3589*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3590*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3591*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3592*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3593*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3594*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3595*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3596*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
3597*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3598*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3599*4bdc9457SAndroid Build Coastguard Worker }
3600*4bdc9457SAndroid Build Coastguard Worker
3601*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/GEMM path, batched ****************************/
3602*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_2x2s2)3603*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_2x2s2) {
3604*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3605*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3606*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3607*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3608*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3609*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3610*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
3611*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3612*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3613*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3614*4bdc9457SAndroid Build Coastguard Worker }
3615*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_Kx2sKx2)3616*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_Kx2sKx2) {
3617*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3618*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 3; kernel_height <= 5; kernel_height++) {
3619*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3620*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3621*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3622*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 2)
3623*4bdc9457SAndroid Build Coastguard Worker .stride(kernel_height, 2)
3624*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3625*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3626*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3627*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3628*4bdc9457SAndroid Build Coastguard Worker }
3629*4bdc9457SAndroid Build Coastguard Worker }
3630*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_2xKs2xK)3631*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_2xKs2xK) {
3632*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3633*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 3; kernel_width <= 5; kernel_width++) {
3634*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3635*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3636*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3637*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, kernel_width)
3638*4bdc9457SAndroid Build Coastguard Worker .stride(2, kernel_width)
3639*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3640*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3641*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3642*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3643*4bdc9457SAndroid Build Coastguard Worker }
3644*4bdc9457SAndroid Build Coastguard Worker }
3645*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_2x2s2_height_adjustment)3646*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_2x2s2_height_adjustment) {
3647*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3648*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3649*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3650*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3651*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(1)
3652*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3653*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3654*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
3655*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3656*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3657*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3658*4bdc9457SAndroid Build Coastguard Worker }
3659*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_2x2s2_width_adjustment)3660*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_2x2s2_width_adjustment) {
3661*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3662*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3663*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3664*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3665*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(1)
3666*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3667*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3668*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
3669*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3670*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3671*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3672*4bdc9457SAndroid Build Coastguard Worker }
3673*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_2x2s2_varying_input_height)3674*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_2x2s2_varying_input_height) {
3675*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3676*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
3677*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3678*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3679*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
3680*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3681*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3682*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
3683*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3684*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3685*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3686*4bdc9457SAndroid Build Coastguard Worker }
3687*4bdc9457SAndroid Build Coastguard Worker }
3688*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_2x2s2_varying_input_width)3689*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_2x2s2_varying_input_width) {
3690*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3691*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
3692*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3693*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3694*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3695*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3696*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3697*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
3698*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3699*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3700*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3701*4bdc9457SAndroid Build Coastguard Worker }
3702*4bdc9457SAndroid Build Coastguard Worker }
3703*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_2x2s2_varying_input_channels)3704*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_2x2s2_varying_input_channels) {
3705*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3706*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
3707*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3708*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3709*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3710*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3711*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3712*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
3713*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3714*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3715*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3716*4bdc9457SAndroid Build Coastguard Worker }
3717*4bdc9457SAndroid Build Coastguard Worker }
3718*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_2x2s2_varying_output_channels)3719*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_2x2s2_varying_output_channels) {
3720*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3721*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qs8.gemm.nr * 2; output_channels *= 2) {
3722*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3723*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3724*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3725*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3726*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3727*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
3728*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
3729*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3730*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3731*4bdc9457SAndroid Build Coastguard Worker }
3732*4bdc9457SAndroid Build Coastguard Worker }
3733*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_2x2s2_with_input_stride)3734*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_2x2s2_with_input_stride) {
3735*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3736*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3737*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3738*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3739*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3740*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3741*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
3742*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3743*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
3744*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3745*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3746*4bdc9457SAndroid Build Coastguard Worker }
3747*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_2x2s2_with_output_stride)3748*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_2x2s2_with_output_stride) {
3749*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3750*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3751*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3752*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3753*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3754*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3755*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
3756*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3757*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qs8.gemm.nr * 2 + 13)
3758*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3759*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3760*4bdc9457SAndroid Build Coastguard Worker }
3761*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_2x2s2_with_qmin)3762*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_2x2s2_with_qmin) {
3763*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3764*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3765*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3766*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3767*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3768*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3769*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
3770*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3771*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
3772*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3773*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3774*4bdc9457SAndroid Build Coastguard Worker }
3775*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_2x2s2_with_qmax)3776*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_2x2s2_with_qmax) {
3777*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3778*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3779*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3780*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3781*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3782*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3783*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
3784*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3785*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
3786*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3787*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3788*4bdc9457SAndroid Build Coastguard Worker }
3789*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_2x2s2_without_bias)3790*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_2x2s2_without_bias) {
3791*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3792*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3793*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
3794*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3795*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3796*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3797*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3798*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
3799*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3800*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3801*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3802*4bdc9457SAndroid Build Coastguard Worker }
3803*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,weights_cache_batched_2x2s2)3804*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, weights_cache_batched_2x2s2) {
3805*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3806*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3807*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3808*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3809*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3810*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3811*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
3812*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3813*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
3814*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3815*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3816*4bdc9457SAndroid Build Coastguard Worker }
3817*4bdc9457SAndroid Build Coastguard Worker
3818*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/GEMM path, grouped, batched ****************************/
3819*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_2x2s2)3820*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_2x2s2) {
3821*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3822*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3823*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3824*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3825*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3826*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3827*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3828*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3829*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3830*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3831*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3832*4bdc9457SAndroid Build Coastguard Worker }
3833*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_Kx2sKx2)3834*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_Kx2sKx2) {
3835*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3836*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 3; kernel_height <= 5; kernel_height++) {
3837*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3838*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3839*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3840*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 2)
3841*4bdc9457SAndroid Build Coastguard Worker .stride(kernel_height, 2)
3842*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3843*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3844*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3845*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3846*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3847*4bdc9457SAndroid Build Coastguard Worker }
3848*4bdc9457SAndroid Build Coastguard Worker }
3849*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_2xKs2xK)3850*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_2xKs2xK) {
3851*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3852*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 3; kernel_width <= 5; kernel_width++) {
3853*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3854*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3855*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3856*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, kernel_width)
3857*4bdc9457SAndroid Build Coastguard Worker .stride(2, kernel_width)
3858*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3859*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3860*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3861*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3862*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3863*4bdc9457SAndroid Build Coastguard Worker }
3864*4bdc9457SAndroid Build Coastguard Worker }
3865*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_2x2s2_height_adjustment)3866*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_2x2s2_height_adjustment) {
3867*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3868*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3869*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3870*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3871*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(1)
3872*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3873*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3874*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3875*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3876*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3877*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3878*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3879*4bdc9457SAndroid Build Coastguard Worker }
3880*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_2x2s2_width_adjustment)3881*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_2x2s2_width_adjustment) {
3882*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3883*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3884*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3885*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3886*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(1)
3887*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3888*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3889*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3890*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3891*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3892*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3893*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3894*4bdc9457SAndroid Build Coastguard Worker }
3895*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_2x2s2_varying_input_height)3896*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_2x2s2_varying_input_height) {
3897*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3898*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
3899*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3900*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3901*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
3902*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3903*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3904*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3905*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3906*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3907*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3908*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3909*4bdc9457SAndroid Build Coastguard Worker }
3910*4bdc9457SAndroid Build Coastguard Worker }
3911*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_2x2s2_varying_input_width)3912*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_2x2s2_varying_input_width) {
3913*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3914*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
3915*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3916*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3917*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3918*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3919*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3920*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3921*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3922*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3923*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3924*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3925*4bdc9457SAndroid Build Coastguard Worker }
3926*4bdc9457SAndroid Build Coastguard Worker }
3927*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_2x2s2_varying_input_channels)3928*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_2x2s2_varying_input_channels) {
3929*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3930*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 14; input_channels <= 20; input_channels++) {
3931*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3932*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3933*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3934*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3935*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3936*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3937*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
3938*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3939*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3940*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3941*4bdc9457SAndroid Build Coastguard Worker }
3942*4bdc9457SAndroid Build Coastguard Worker }
3943*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_2x2s2_varying_output_channels)3944*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_2x2s2_varying_output_channels) {
3945*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3946*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qs8.gemm.nr * 2; output_channels *= 2) {
3947*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3948*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3949*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3950*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3951*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3952*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3953*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3954*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
3955*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
3956*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3957*4bdc9457SAndroid Build Coastguard Worker }
3958*4bdc9457SAndroid Build Coastguard Worker }
3959*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_2x2s2_with_input_stride)3960*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_2x2s2_with_input_stride) {
3961*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3962*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3963*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3964*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3965*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3966*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3967*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3968*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3969*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
3970*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(37)
3971*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3972*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3973*4bdc9457SAndroid Build Coastguard Worker }
3974*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_2x2s2_with_output_stride)3975*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_2x2s2_with_output_stride) {
3976*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3977*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3978*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3979*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3980*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3981*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3982*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3983*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3984*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr + 3)
3985*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qs8.gemm.nr * 2 + 13)
3986*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
3987*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
3988*4bdc9457SAndroid Build Coastguard Worker }
3989*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_2x2s2_with_qmin)3990*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_2x2s2_with_qmin) {
3991*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
3992*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
3993*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
3994*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
3995*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
3996*4bdc9457SAndroid Build Coastguard Worker .stride(2)
3997*4bdc9457SAndroid Build Coastguard Worker .groups(2)
3998*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
3999*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
4000*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
4001*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4002*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
4003*4bdc9457SAndroid Build Coastguard Worker }
4004*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_2x2s2_with_qmax)4005*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_2x2s2_with_qmax) {
4006*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4007*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4008*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
4009*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
4010*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
4011*4bdc9457SAndroid Build Coastguard Worker .stride(2)
4012*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4013*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
4014*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
4015*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
4016*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4017*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
4018*4bdc9457SAndroid Build Coastguard Worker }
4019*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,batched_grouped_2x2s2_without_bias)4020*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, batched_grouped_2x2s2_without_bias) {
4021*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4022*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4023*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
4024*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
4025*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
4026*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
4027*4bdc9457SAndroid Build Coastguard Worker .stride(2)
4028*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4029*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
4030*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
4031*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4032*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
4033*4bdc9457SAndroid Build Coastguard Worker }
4034*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QS8,weights_cache_batched_grouped_2x2s2)4035*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, weights_cache_batched_grouped_2x2s2) {
4036*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4037*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4038*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
4039*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
4040*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
4041*4bdc9457SAndroid Build Coastguard Worker .stride(2)
4042*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4043*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
4044*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qs8.gemm.nr * 2 + 3)
4045*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
4046*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4047*4bdc9457SAndroid Build Coastguard Worker .TestQS8();
4048*4bdc9457SAndroid Build Coastguard Worker }
4049*4bdc9457SAndroid Build Coastguard Worker
4050*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/GEMM path, setup ****************************/
4051*4bdc9457SAndroid Build Coastguard Worker
4052*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 2x2s2_setup_changing_batch) {
4053*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4054*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4055*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
4056*4bdc9457SAndroid Build Coastguard Worker .next_batch_size(5)
4057*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
4058*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
4059*4bdc9457SAndroid Build Coastguard Worker .stride(2)
4060*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4061*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
4062*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
4063*4bdc9457SAndroid Build Coastguard Worker .TestSetupQS8();
4064*4bdc9457SAndroid Build Coastguard Worker }
4065*4bdc9457SAndroid Build Coastguard Worker
4066*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 2x2s2_setup_changing_height) {
4067*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4068*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4069*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
4070*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
4071*4bdc9457SAndroid Build Coastguard Worker .next_input_height(kStridedInputHeight + 3)
4072*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
4073*4bdc9457SAndroid Build Coastguard Worker .stride(2)
4074*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4075*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
4076*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
4077*4bdc9457SAndroid Build Coastguard Worker .TestSetupQS8();
4078*4bdc9457SAndroid Build Coastguard Worker }
4079*4bdc9457SAndroid Build Coastguard Worker
4080*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QS8, 2x2s2_setup_changing_width) {
4081*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4082*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4083*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
4084*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
4085*4bdc9457SAndroid Build Coastguard Worker .next_input_width(kStridedInputWidth + 3)
4086*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
4087*4bdc9457SAndroid Build Coastguard Worker .stride(2)
4088*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4089*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
4090*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
4091*4bdc9457SAndroid Build Coastguard Worker .TestSetupQS8();
4092*4bdc9457SAndroid Build Coastguard Worker }
4093*4bdc9457SAndroid Build Coastguard Worker
4094*4bdc9457SAndroid Build Coastguard Worker /**************************** Future GEMM path ****************************/
4095*4bdc9457SAndroid Build Coastguard Worker
4096*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 1x1) {
4097*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4098*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4099*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4100*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4101*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4102*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4103*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4104*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4105*4bdc9457SAndroid Build Coastguard Worker }
4106*4bdc9457SAndroid Build Coastguard Worker
4107*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 1x1_varying_input_width) {
4108*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4109*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
4110*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4111*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
4112*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4113*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4114*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4115*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
4116*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4117*4bdc9457SAndroid Build Coastguard Worker }
4118*4bdc9457SAndroid Build Coastguard Worker }
4119*4bdc9457SAndroid Build Coastguard Worker
4120*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 1x1_varying_input_height) {
4121*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4122*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
4123*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4124*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
4125*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4126*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4127*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4128*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
4129*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4130*4bdc9457SAndroid Build Coastguard Worker }
4131*4bdc9457SAndroid Build Coastguard Worker }
4132*4bdc9457SAndroid Build Coastguard Worker
4133*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 1x1_varying_input_channels) {
4134*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4135*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
4136*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4137*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4138*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4139*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
4140*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4141*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
4142*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4143*4bdc9457SAndroid Build Coastguard Worker }
4144*4bdc9457SAndroid Build Coastguard Worker }
4145*4bdc9457SAndroid Build Coastguard Worker
4146*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 1x1_varying_output_channels) {
4147*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4148*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qu8.gemm.nr * 2; output_channels *= 2) {
4149*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4150*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4151*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4152*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4153*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
4154*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
4155*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4156*4bdc9457SAndroid Build Coastguard Worker }
4157*4bdc9457SAndroid Build Coastguard Worker }
4158*4bdc9457SAndroid Build Coastguard Worker
4159*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 1x1_with_input_stride) {
4160*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4161*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4162*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4163*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4164*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4165*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4166*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
4167*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4168*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4169*4bdc9457SAndroid Build Coastguard Worker }
4170*4bdc9457SAndroid Build Coastguard Worker
4171*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 1x1_with_output_stride) {
4172*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4173*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4174*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4175*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4176*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4177*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4178*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qu8.gemm.nr * 2 + 13)
4179*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4180*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4181*4bdc9457SAndroid Build Coastguard Worker }
4182*4bdc9457SAndroid Build Coastguard Worker
4183*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 1x1_with_qmin) {
4184*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4185*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4186*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4187*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4188*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4189*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4190*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
4191*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4192*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4193*4bdc9457SAndroid Build Coastguard Worker }
4194*4bdc9457SAndroid Build Coastguard Worker
4195*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 1x1_with_qmax) {
4196*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4197*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4198*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4199*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4200*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4201*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4202*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
4203*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4204*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4205*4bdc9457SAndroid Build Coastguard Worker }
4206*4bdc9457SAndroid Build Coastguard Worker
4207*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 1x1_without_bias) {
4208*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4209*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4210*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
4211*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4212*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4213*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4214*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4215*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4216*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4217*4bdc9457SAndroid Build Coastguard Worker }
4218*4bdc9457SAndroid Build Coastguard Worker
4219*4bdc9457SAndroid Build Coastguard Worker /**************************** Future GEMM path, grouped ****************************/
4220*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_1x1)4221*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_1x1) {
4222*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4223*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4224*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4225*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4226*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4227*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4228*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4229*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4230*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4231*4bdc9457SAndroid Build Coastguard Worker }
4232*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_1x1_varying_input_width)4233*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_1x1_varying_input_width) {
4234*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4235*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
4236*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4237*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
4238*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4239*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4240*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4241*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4242*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
4243*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4244*4bdc9457SAndroid Build Coastguard Worker }
4245*4bdc9457SAndroid Build Coastguard Worker }
4246*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_1x1_varying_input_height)4247*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_1x1_varying_input_height) {
4248*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4249*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
4250*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4251*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
4252*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4253*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4254*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4255*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4256*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
4257*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4258*4bdc9457SAndroid Build Coastguard Worker }
4259*4bdc9457SAndroid Build Coastguard Worker }
4260*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_1x1_varying_input_channels)4261*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_1x1_varying_input_channels) {
4262*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4263*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
4264*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4265*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4266*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4267*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4268*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
4269*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4270*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
4271*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4272*4bdc9457SAndroid Build Coastguard Worker }
4273*4bdc9457SAndroid Build Coastguard Worker }
4274*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_1x1_varying_output_channels)4275*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_1x1_varying_output_channels) {
4276*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4277*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qu8.gemm.nr * 2; output_channels *= 2) {
4278*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4279*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4280*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4281*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4282*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4283*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
4284*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
4285*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4286*4bdc9457SAndroid Build Coastguard Worker }
4287*4bdc9457SAndroid Build Coastguard Worker }
4288*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_1x1_with_input_stride)4289*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_1x1_with_input_stride) {
4290*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4291*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4292*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4293*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4294*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4295*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4296*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4297*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(47)
4298*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4299*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4300*4bdc9457SAndroid Build Coastguard Worker }
4301*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_1x1_with_output_stride)4302*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_1x1_with_output_stride) {
4303*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4304*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4305*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4306*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4307*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4308*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4309*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr + 3)
4310*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qu8.gemm.nr * 2 + 13)
4311*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4312*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4313*4bdc9457SAndroid Build Coastguard Worker }
4314*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_1x1_with_qmin)4315*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_1x1_with_qmin) {
4316*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4317*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4318*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4319*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4320*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4321*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4322*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4323*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
4324*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4325*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4326*4bdc9457SAndroid Build Coastguard Worker }
4327*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_1x1_with_qmax)4328*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_1x1_with_qmax) {
4329*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4330*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4331*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4332*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4333*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4334*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4335*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4336*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
4337*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4338*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4339*4bdc9457SAndroid Build Coastguard Worker }
4340*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_1x1_without_bias)4341*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_1x1_without_bias) {
4342*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4343*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4344*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
4345*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4346*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4347*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4348*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4349*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4350*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4351*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4352*4bdc9457SAndroid Build Coastguard Worker }
4353*4bdc9457SAndroid Build Coastguard Worker
4354*4bdc9457SAndroid Build Coastguard Worker /**************************** Future GEMM path, batched ****************************/
4355*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_1x1)4356*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_1x1) {
4357*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4358*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4359*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
4360*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4361*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4362*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4363*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4364*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4365*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4366*4bdc9457SAndroid Build Coastguard Worker }
4367*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_1x1_varying_input_width)4368*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_1x1_varying_input_width) {
4369*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4370*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
4371*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4372*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
4373*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
4374*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4375*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4376*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4377*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
4378*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4379*4bdc9457SAndroid Build Coastguard Worker }
4380*4bdc9457SAndroid Build Coastguard Worker }
4381*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_1x1_varying_input_height)4382*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_1x1_varying_input_height) {
4383*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4384*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
4385*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4386*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
4387*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
4388*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4389*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4390*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4391*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
4392*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4393*4bdc9457SAndroid Build Coastguard Worker }
4394*4bdc9457SAndroid Build Coastguard Worker }
4395*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_1x1_varying_input_channels)4396*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_1x1_varying_input_channels) {
4397*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4398*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
4399*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4400*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
4401*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4402*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4403*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
4404*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4405*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
4406*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4407*4bdc9457SAndroid Build Coastguard Worker }
4408*4bdc9457SAndroid Build Coastguard Worker }
4409*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_1x1_varying_output_channels)4410*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_1x1_varying_output_channels) {
4411*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4412*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qu8.gemm.nr * 2; output_channels *= 2) {
4413*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4414*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
4415*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4416*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4417*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4418*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
4419*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
4420*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4421*4bdc9457SAndroid Build Coastguard Worker }
4422*4bdc9457SAndroid Build Coastguard Worker }
4423*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_1x1_with_input_stride)4424*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_1x1_with_input_stride) {
4425*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4426*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4427*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
4428*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4429*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4430*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4431*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4432*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
4433*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4434*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4435*4bdc9457SAndroid Build Coastguard Worker }
4436*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_1x1_with_output_stride)4437*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_1x1_with_output_stride) {
4438*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4439*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4440*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
4441*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4442*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4443*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4444*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4445*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qu8.gemm.nr * 2 + 13)
4446*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4447*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4448*4bdc9457SAndroid Build Coastguard Worker }
4449*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_1x1_with_qmin)4450*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_1x1_with_qmin) {
4451*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4452*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4453*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
4454*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4455*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4456*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4457*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4458*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
4459*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4460*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4461*4bdc9457SAndroid Build Coastguard Worker }
4462*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_1x1_with_qmax)4463*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_1x1_with_qmax) {
4464*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4465*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4466*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
4467*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4468*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4469*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4470*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4471*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
4472*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4473*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4474*4bdc9457SAndroid Build Coastguard Worker }
4475*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_1x1_without_bias)4476*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_1x1_without_bias) {
4477*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4478*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4479*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
4480*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
4481*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4482*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4483*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4484*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4485*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4486*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4487*4bdc9457SAndroid Build Coastguard Worker }
4488*4bdc9457SAndroid Build Coastguard Worker
4489*4bdc9457SAndroid Build Coastguard Worker /**************************** Future GEMM path, batched, grouped ****************************/
4490*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_1x1)4491*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_1x1) {
4492*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4493*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4494*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
4495*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4496*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4497*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4498*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4499*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4500*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4501*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4502*4bdc9457SAndroid Build Coastguard Worker }
4503*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_1x1_varying_input_width)4504*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_1x1_varying_input_width) {
4505*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4506*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
4507*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4508*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
4509*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
4510*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4511*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4512*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4513*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4514*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
4515*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4516*4bdc9457SAndroid Build Coastguard Worker }
4517*4bdc9457SAndroid Build Coastguard Worker }
4518*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_1x1_varying_input_height)4519*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_1x1_varying_input_height) {
4520*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4521*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
4522*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4523*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
4524*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
4525*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4526*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4527*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4528*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4529*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
4530*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4531*4bdc9457SAndroid Build Coastguard Worker }
4532*4bdc9457SAndroid Build Coastguard Worker }
4533*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_1x1_varying_input_channels)4534*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_1x1_varying_input_channels) {
4535*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4536*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
4537*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4538*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
4539*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4540*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4541*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4542*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
4543*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4544*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
4545*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4546*4bdc9457SAndroid Build Coastguard Worker }
4547*4bdc9457SAndroid Build Coastguard Worker }
4548*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_1x1_varying_output_channels)4549*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_1x1_varying_output_channels) {
4550*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4551*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qu8.gemm.nr * 2; output_channels *= 2) {
4552*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4553*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
4554*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4555*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4556*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4557*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4558*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
4559*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
4560*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4561*4bdc9457SAndroid Build Coastguard Worker }
4562*4bdc9457SAndroid Build Coastguard Worker }
4563*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_1x1_with_input_stride)4564*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_1x1_with_input_stride) {
4565*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4566*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4567*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
4568*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4569*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4570*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4571*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4572*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4573*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(47)
4574*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4575*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4576*4bdc9457SAndroid Build Coastguard Worker }
4577*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_1x1_with_output_stride)4578*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_1x1_with_output_stride) {
4579*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4580*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4581*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
4582*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4583*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4584*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4585*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4586*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr + 3)
4587*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qu8.gemm.nr * 2 + 13)
4588*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4589*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4590*4bdc9457SAndroid Build Coastguard Worker }
4591*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_1x1_with_qmin)4592*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_1x1_with_qmin) {
4593*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4594*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4595*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
4596*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4597*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4598*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4599*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4600*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4601*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
4602*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4603*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4604*4bdc9457SAndroid Build Coastguard Worker }
4605*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_1x1_with_qmax)4606*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_1x1_with_qmax) {
4607*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4608*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4609*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
4610*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4611*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4612*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4613*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4614*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4615*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
4616*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4617*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4618*4bdc9457SAndroid Build Coastguard Worker }
4619*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_1x1_without_bias)4620*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_1x1_without_bias) {
4621*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4622*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4623*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
4624*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
4625*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4626*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
4627*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4628*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4629*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4630*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4631*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4632*4bdc9457SAndroid Build Coastguard Worker }
4633*4bdc9457SAndroid Build Coastguard Worker
4634*4bdc9457SAndroid Build Coastguard Worker /**************************** CONV path ****************************/
4635*4bdc9457SAndroid Build Coastguard Worker
4636*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3) {
4637*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4638*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4639*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4640*4bdc9457SAndroid Build Coastguard Worker .padding(1)
4641*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
4642*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
4643*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4644*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4645*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4646*4bdc9457SAndroid Build Coastguard Worker }
4647*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,Kx3)4648*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, Kx3) {
4649*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4650*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 1; kernel_height <= 4; kernel_height *= 2) {
4651*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4652*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4653*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
4654*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
4655*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
4656*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4657*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4658*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4659*4bdc9457SAndroid Build Coastguard Worker }
4660*4bdc9457SAndroid Build Coastguard Worker }
4661*4bdc9457SAndroid Build Coastguard Worker
4662*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3xK) {
4663*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4664*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 1; kernel_width <= 4; kernel_width *= 2) {
4665*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4666*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4667*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
4668*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
4669*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
4670*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4671*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4672*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4673*4bdc9457SAndroid Build Coastguard Worker }
4674*4bdc9457SAndroid Build Coastguard Worker }
4675*4bdc9457SAndroid Build Coastguard Worker
4676*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3_varying_height_padding) {
4677*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4678*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
4679*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
4680*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4681*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4682*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
4683*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
4684*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
4685*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
4686*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
4687*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4688*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
4689*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4690*4bdc9457SAndroid Build Coastguard Worker }
4691*4bdc9457SAndroid Build Coastguard Worker }
4692*4bdc9457SAndroid Build Coastguard Worker }
4693*4bdc9457SAndroid Build Coastguard Worker
4694*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3_varying_width_padding) {
4695*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4696*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
4697*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
4698*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4699*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4700*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
4701*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
4702*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
4703*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
4704*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
4705*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4706*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
4707*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4708*4bdc9457SAndroid Build Coastguard Worker }
4709*4bdc9457SAndroid Build Coastguard Worker }
4710*4bdc9457SAndroid Build Coastguard Worker }
4711*4bdc9457SAndroid Build Coastguard Worker
4712*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3_varying_height_adjustment) {
4713*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4714*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 1; adjustment_height <= 2; adjustment_height++) {
4715*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4716*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4717*4bdc9457SAndroid Build Coastguard Worker .padding(1)
4718*4bdc9457SAndroid Build Coastguard Worker .stride_height(adjustment_height + 1)
4719*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
4720*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
4721*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
4722*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4723*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
4724*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4725*4bdc9457SAndroid Build Coastguard Worker }
4726*4bdc9457SAndroid Build Coastguard Worker }
4727*4bdc9457SAndroid Build Coastguard Worker
4728*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3_varying_width_adjustment) {
4729*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4730*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 1; adjustment_width <= 2; adjustment_width++) {
4731*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4732*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4733*4bdc9457SAndroid Build Coastguard Worker .padding(1)
4734*4bdc9457SAndroid Build Coastguard Worker .stride_width(adjustment_width + 1)
4735*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
4736*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
4737*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
4738*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4739*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
4740*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4741*4bdc9457SAndroid Build Coastguard Worker }
4742*4bdc9457SAndroid Build Coastguard Worker }
4743*4bdc9457SAndroid Build Coastguard Worker
4744*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3_varying_input_height) {
4745*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4746*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
4747*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4748*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
4749*4bdc9457SAndroid Build Coastguard Worker .padding(1)
4750*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
4751*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
4752*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4753*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
4754*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4755*4bdc9457SAndroid Build Coastguard Worker }
4756*4bdc9457SAndroid Build Coastguard Worker }
4757*4bdc9457SAndroid Build Coastguard Worker
4758*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3_varying_input_width) {
4759*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4760*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
4761*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4762*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
4763*4bdc9457SAndroid Build Coastguard Worker .padding(1)
4764*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
4765*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
4766*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4767*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
4768*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4769*4bdc9457SAndroid Build Coastguard Worker }
4770*4bdc9457SAndroid Build Coastguard Worker }
4771*4bdc9457SAndroid Build Coastguard Worker
4772*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3_varying_input_channels) {
4773*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4774*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
4775*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4776*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4777*4bdc9457SAndroid Build Coastguard Worker .padding(1)
4778*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
4779*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
4780*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4781*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
4782*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4783*4bdc9457SAndroid Build Coastguard Worker }
4784*4bdc9457SAndroid Build Coastguard Worker }
4785*4bdc9457SAndroid Build Coastguard Worker
4786*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3_varying_output_channels) {
4787*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4788*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qu8.gemm.nr * 2; output_channels *= 2) {
4789*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4790*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4791*4bdc9457SAndroid Build Coastguard Worker .padding(1)
4792*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
4793*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4794*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
4795*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
4796*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4797*4bdc9457SAndroid Build Coastguard Worker }
4798*4bdc9457SAndroid Build Coastguard Worker }
4799*4bdc9457SAndroid Build Coastguard Worker
4800*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3_with_height_dilation) {
4801*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4802*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_height = 2; dilation_height <= 3; dilation_height++) {
4803*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4804*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4805*4bdc9457SAndroid Build Coastguard Worker .padding(1)
4806*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
4807*4bdc9457SAndroid Build Coastguard Worker .dilation_height(dilation_height)
4808*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4809*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4810*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4811*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4812*4bdc9457SAndroid Build Coastguard Worker }
4813*4bdc9457SAndroid Build Coastguard Worker }
4814*4bdc9457SAndroid Build Coastguard Worker
4815*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3_with_width_dilation) {
4816*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4817*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) {
4818*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4819*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4820*4bdc9457SAndroid Build Coastguard Worker .padding(1)
4821*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
4822*4bdc9457SAndroid Build Coastguard Worker .dilation_width(dilation_width)
4823*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4824*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4825*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4826*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4827*4bdc9457SAndroid Build Coastguard Worker }
4828*4bdc9457SAndroid Build Coastguard Worker }
4829*4bdc9457SAndroid Build Coastguard Worker
4830*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3_with_height_dilation_and_stride) {
4831*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4832*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4833*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4834*4bdc9457SAndroid Build Coastguard Worker .padding(1)
4835*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
4836*4bdc9457SAndroid Build Coastguard Worker .dilation_height(3)
4837*4bdc9457SAndroid Build Coastguard Worker .stride_height(2)
4838*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4839*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4840*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4841*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4842*4bdc9457SAndroid Build Coastguard Worker }
4843*4bdc9457SAndroid Build Coastguard Worker
4844*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3_with_width_dilation_and_stride) {
4845*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4846*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4847*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4848*4bdc9457SAndroid Build Coastguard Worker .padding(1)
4849*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
4850*4bdc9457SAndroid Build Coastguard Worker .dilation_width(3)
4851*4bdc9457SAndroid Build Coastguard Worker .stride_width(2)
4852*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4853*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4854*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4855*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4856*4bdc9457SAndroid Build Coastguard Worker }
4857*4bdc9457SAndroid Build Coastguard Worker
4858*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3_with_input_stride) {
4859*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4860*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4861*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4862*4bdc9457SAndroid Build Coastguard Worker .padding(1)
4863*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
4864*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4865*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4866*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
4867*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4868*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4869*4bdc9457SAndroid Build Coastguard Worker }
4870*4bdc9457SAndroid Build Coastguard Worker
4871*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3_with_output_stride) {
4872*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4873*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4874*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4875*4bdc9457SAndroid Build Coastguard Worker .padding(1)
4876*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
4877*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4878*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4879*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qu8.gemm.nr * 2 + 13)
4880*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4881*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4882*4bdc9457SAndroid Build Coastguard Worker }
4883*4bdc9457SAndroid Build Coastguard Worker
4884*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3_with_qmin) {
4885*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4886*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4887*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4888*4bdc9457SAndroid Build Coastguard Worker .padding(1)
4889*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
4890*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4891*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4892*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
4893*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4894*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4895*4bdc9457SAndroid Build Coastguard Worker }
4896*4bdc9457SAndroid Build Coastguard Worker
4897*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3_with_qmax) {
4898*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4899*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4900*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4901*4bdc9457SAndroid Build Coastguard Worker .padding(1)
4902*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
4903*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4904*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4905*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
4906*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4907*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4908*4bdc9457SAndroid Build Coastguard Worker }
4909*4bdc9457SAndroid Build Coastguard Worker
4910*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3_without_bias) {
4911*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4912*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4913*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
4914*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4915*4bdc9457SAndroid Build Coastguard Worker .padding(1)
4916*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
4917*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
4918*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4919*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4920*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4921*4bdc9457SAndroid Build Coastguard Worker }
4922*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,weights_cache_3x3)4923*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, weights_cache_3x3) {
4924*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4925*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4926*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4927*4bdc9457SAndroid Build Coastguard Worker .padding(1)
4928*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
4929*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
4930*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4931*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
4932*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4933*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4934*4bdc9457SAndroid Build Coastguard Worker }
4935*4bdc9457SAndroid Build Coastguard Worker
4936*4bdc9457SAndroid Build Coastguard Worker /**************************** CONV path, grouped ****************************/
4937*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3)4938*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3) {
4939*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4940*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4941*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4942*4bdc9457SAndroid Build Coastguard Worker .padding(1)
4943*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
4944*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4945*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
4946*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4947*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4948*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4949*4bdc9457SAndroid Build Coastguard Worker }
4950*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_Kx3)4951*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_Kx3) {
4952*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4953*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 1; kernel_height <= 4; kernel_height *= 2) {
4954*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4955*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4956*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
4957*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
4958*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4959*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
4960*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4961*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4962*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4963*4bdc9457SAndroid Build Coastguard Worker }
4964*4bdc9457SAndroid Build Coastguard Worker }
4965*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3xK)4966*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3xK) {
4967*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4968*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 1; kernel_width <= 4; kernel_width *= 2) {
4969*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4970*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4971*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
4972*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
4973*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4974*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
4975*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4976*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
4977*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4978*4bdc9457SAndroid Build Coastguard Worker }
4979*4bdc9457SAndroid Build Coastguard Worker }
4980*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3_varying_height_padding)4981*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3_varying_height_padding) {
4982*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
4983*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
4984*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
4985*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
4986*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
4987*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
4988*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
4989*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
4990*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
4991*4bdc9457SAndroid Build Coastguard Worker .groups(2)
4992*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
4993*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
4994*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
4995*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
4996*4bdc9457SAndroid Build Coastguard Worker }
4997*4bdc9457SAndroid Build Coastguard Worker }
4998*4bdc9457SAndroid Build Coastguard Worker }
4999*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3_varying_width_padding)5000*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3_varying_width_padding) {
5001*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5002*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
5003*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
5004*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5005*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5006*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
5007*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
5008*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
5009*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5010*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5011*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
5012*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5013*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
5014*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5015*4bdc9457SAndroid Build Coastguard Worker }
5016*4bdc9457SAndroid Build Coastguard Worker }
5017*4bdc9457SAndroid Build Coastguard Worker }
5018*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3_varying_height_adjustment)5019*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3_varying_height_adjustment) {
5020*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5021*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 1; adjustment_height <= 2; adjustment_height++) {
5022*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5023*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5024*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5025*4bdc9457SAndroid Build Coastguard Worker .stride_height(adjustment_height + 1)
5026*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
5027*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5028*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5029*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
5030*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5031*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
5032*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5033*4bdc9457SAndroid Build Coastguard Worker }
5034*4bdc9457SAndroid Build Coastguard Worker }
5035*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3_varying_width_adjustment)5036*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3_varying_width_adjustment) {
5037*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5038*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 1; adjustment_width <= 2; adjustment_width++) {
5039*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5040*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5041*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5042*4bdc9457SAndroid Build Coastguard Worker .stride_width(adjustment_width + 1)
5043*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
5044*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5045*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5046*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
5047*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5048*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
5049*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5050*4bdc9457SAndroid Build Coastguard Worker }
5051*4bdc9457SAndroid Build Coastguard Worker }
5052*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3_varying_input_height)5053*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3_varying_input_height) {
5054*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5055*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
5056*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5057*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
5058*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5059*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5060*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5061*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
5062*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5063*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
5064*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5065*4bdc9457SAndroid Build Coastguard Worker }
5066*4bdc9457SAndroid Build Coastguard Worker }
5067*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3_varying_input_width)5068*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3_varying_input_width) {
5069*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5070*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
5071*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5072*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
5073*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5074*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5075*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5076*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
5077*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5078*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
5079*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5080*4bdc9457SAndroid Build Coastguard Worker }
5081*4bdc9457SAndroid Build Coastguard Worker }
5082*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3_varying_input_channels)5083*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3_varying_input_channels) {
5084*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5085*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
5086*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5087*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5088*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5089*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5090*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5091*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
5092*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5093*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
5094*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5095*4bdc9457SAndroid Build Coastguard Worker }
5096*4bdc9457SAndroid Build Coastguard Worker }
5097*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3_varying_output_channels)5098*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3_varying_output_channels) {
5099*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5100*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qu8.gemm.nr * 2; output_channels *= 2) {
5101*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5102*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5103*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5104*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5105*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5106*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5107*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
5108*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
5109*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5110*4bdc9457SAndroid Build Coastguard Worker }
5111*4bdc9457SAndroid Build Coastguard Worker }
5112*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3_with_height_dilation)5113*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3_with_height_dilation) {
5114*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5115*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_height = 2; dilation_height <= 3; dilation_height++) {
5116*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5117*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5118*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5119*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5120*4bdc9457SAndroid Build Coastguard Worker .dilation_height(dilation_height)
5121*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5122*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5123*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5124*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5125*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5126*4bdc9457SAndroid Build Coastguard Worker }
5127*4bdc9457SAndroid Build Coastguard Worker }
5128*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3_with_width_dilation)5129*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3_with_width_dilation) {
5130*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5131*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) {
5132*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5133*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5134*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5135*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5136*4bdc9457SAndroid Build Coastguard Worker .dilation_width(dilation_width)
5137*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5138*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5139*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5140*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5141*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5142*4bdc9457SAndroid Build Coastguard Worker }
5143*4bdc9457SAndroid Build Coastguard Worker }
5144*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3_with_height_dilation_and_stride)5145*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3_with_height_dilation_and_stride) {
5146*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5147*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5148*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5149*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5150*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5151*4bdc9457SAndroid Build Coastguard Worker .dilation_height(3)
5152*4bdc9457SAndroid Build Coastguard Worker .stride_height(2)
5153*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5154*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5155*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5156*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5157*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5158*4bdc9457SAndroid Build Coastguard Worker }
5159*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3_with_width_dilation_and_stride)5160*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3_with_width_dilation_and_stride) {
5161*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5162*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5163*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5164*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5165*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5166*4bdc9457SAndroid Build Coastguard Worker .dilation_width(3)
5167*4bdc9457SAndroid Build Coastguard Worker .stride_width(2)
5168*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5169*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5170*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5171*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5172*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5173*4bdc9457SAndroid Build Coastguard Worker }
5174*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3_with_input_stride)5175*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3_with_input_stride) {
5176*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5177*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5178*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5179*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5180*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5181*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5182*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5183*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5184*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(47)
5185*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5186*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5187*4bdc9457SAndroid Build Coastguard Worker }
5188*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3_with_output_stride)5189*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3_with_output_stride) {
5190*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5191*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5192*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5193*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5194*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5195*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5196*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5197*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr + 3)
5198*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qu8.gemm.nr * 2 + 13)
5199*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5200*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5201*4bdc9457SAndroid Build Coastguard Worker }
5202*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3_with_qmin)5203*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3_with_qmin) {
5204*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5205*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5206*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5207*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5208*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5209*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5210*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5211*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5212*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
5213*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5214*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5215*4bdc9457SAndroid Build Coastguard Worker }
5216*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3_with_qmax)5217*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3_with_qmax) {
5218*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5219*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5220*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5221*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5222*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5223*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5224*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5225*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5226*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
5227*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5228*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5229*4bdc9457SAndroid Build Coastguard Worker }
5230*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3_without_bias)5231*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3_without_bias) {
5232*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5233*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5234*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
5235*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5236*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5237*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5238*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5239*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5240*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5241*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5242*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5243*4bdc9457SAndroid Build Coastguard Worker }
5244*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,weights_cache_grouped_3x3)5245*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, weights_cache_grouped_3x3) {
5246*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5247*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5248*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5249*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5250*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5251*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5252*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
5253*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5254*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
5255*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5256*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5257*4bdc9457SAndroid Build Coastguard Worker }
5258*4bdc9457SAndroid Build Coastguard Worker
5259*4bdc9457SAndroid Build Coastguard Worker /**************************** CONV path, batched ****************************/
5260*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3)5261*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3) {
5262*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5263*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5264*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5265*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5266*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5267*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5268*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
5269*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5270*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5271*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5272*4bdc9457SAndroid Build Coastguard Worker }
5273*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_Kx3)5274*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_Kx3) {
5275*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5276*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 1; kernel_height <= 4; kernel_height *= 2) {
5277*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5278*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5279*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5280*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
5281*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
5282*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
5283*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5284*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5285*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5286*4bdc9457SAndroid Build Coastguard Worker }
5287*4bdc9457SAndroid Build Coastguard Worker }
5288*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3xK)5289*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3xK) {
5290*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5291*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 1; kernel_width <= 4; kernel_width *= 2) {
5292*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5293*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5294*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5295*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
5296*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
5297*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
5298*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5299*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5300*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5301*4bdc9457SAndroid Build Coastguard Worker }
5302*4bdc9457SAndroid Build Coastguard Worker }
5303*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3_varying_height_padding)5304*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3_varying_height_padding) {
5305*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5306*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
5307*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
5308*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5309*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5310*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5311*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
5312*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
5313*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
5314*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5315*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
5316*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5317*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
5318*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5319*4bdc9457SAndroid Build Coastguard Worker }
5320*4bdc9457SAndroid Build Coastguard Worker }
5321*4bdc9457SAndroid Build Coastguard Worker }
5322*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3_varying_width_padding)5323*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3_varying_width_padding) {
5324*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5325*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
5326*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
5327*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5328*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5329*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5330*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
5331*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
5332*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
5333*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5334*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
5335*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5336*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
5337*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5338*4bdc9457SAndroid Build Coastguard Worker }
5339*4bdc9457SAndroid Build Coastguard Worker }
5340*4bdc9457SAndroid Build Coastguard Worker }
5341*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3_varying_height_adjustment)5342*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3_varying_height_adjustment) {
5343*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5344*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 1; adjustment_height <= 2; adjustment_height++) {
5345*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5346*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5347*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5348*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5349*4bdc9457SAndroid Build Coastguard Worker .stride_height(adjustment_height + 1)
5350*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
5351*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5352*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
5353*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5354*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
5355*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5356*4bdc9457SAndroid Build Coastguard Worker }
5357*4bdc9457SAndroid Build Coastguard Worker }
5358*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3_varying_width_adjustment)5359*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3_varying_width_adjustment) {
5360*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5361*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 1; adjustment_width <= 2; adjustment_width++) {
5362*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5363*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5364*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5365*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5366*4bdc9457SAndroid Build Coastguard Worker .stride_width(adjustment_width + 1)
5367*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
5368*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5369*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
5370*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5371*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
5372*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5373*4bdc9457SAndroid Build Coastguard Worker }
5374*4bdc9457SAndroid Build Coastguard Worker }
5375*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3_varying_input_height)5376*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3_varying_input_height) {
5377*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5378*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
5379*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5380*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5381*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
5382*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5383*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5384*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
5385*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5386*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
5387*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5388*4bdc9457SAndroid Build Coastguard Worker }
5389*4bdc9457SAndroid Build Coastguard Worker }
5390*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3_varying_input_width)5391*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3_varying_input_width) {
5392*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5393*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
5394*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5395*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5396*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
5397*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5398*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5399*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
5400*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5401*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
5402*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5403*4bdc9457SAndroid Build Coastguard Worker }
5404*4bdc9457SAndroid Build Coastguard Worker }
5405*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3_varying_input_channels)5406*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3_varying_input_channels) {
5407*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5408*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
5409*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5410*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5411*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5412*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5413*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5414*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
5415*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5416*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
5417*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5418*4bdc9457SAndroid Build Coastguard Worker }
5419*4bdc9457SAndroid Build Coastguard Worker }
5420*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3_varying_output_channels)5421*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3_varying_output_channels) {
5422*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5423*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qu8.gemm.nr * 2; output_channels *= 2) {
5424*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5425*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5426*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5427*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5428*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5429*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5430*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
5431*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
5432*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5433*4bdc9457SAndroid Build Coastguard Worker }
5434*4bdc9457SAndroid Build Coastguard Worker }
5435*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3_with_height_dilation)5436*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3_with_height_dilation) {
5437*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5438*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_height = 2; dilation_height <= 3; dilation_height++) {
5439*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5440*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5441*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5442*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5443*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5444*4bdc9457SAndroid Build Coastguard Worker .dilation_height(dilation_height)
5445*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5446*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5447*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5448*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5449*4bdc9457SAndroid Build Coastguard Worker }
5450*4bdc9457SAndroid Build Coastguard Worker }
5451*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3_with_width_dilation)5452*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3_with_width_dilation) {
5453*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5454*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) {
5455*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5456*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5457*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5458*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5459*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5460*4bdc9457SAndroid Build Coastguard Worker .dilation_width(dilation_width)
5461*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5462*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5463*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5464*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5465*4bdc9457SAndroid Build Coastguard Worker }
5466*4bdc9457SAndroid Build Coastguard Worker }
5467*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3_with_height_dilation_and_stride)5468*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3_with_height_dilation_and_stride) {
5469*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5470*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5471*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5472*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5473*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5474*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5475*4bdc9457SAndroid Build Coastguard Worker .dilation_height(3)
5476*4bdc9457SAndroid Build Coastguard Worker .stride_height(2)
5477*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5478*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5479*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5480*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5481*4bdc9457SAndroid Build Coastguard Worker }
5482*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3_with_width_dilation_and_stride)5483*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3_with_width_dilation_and_stride) {
5484*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5485*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5486*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5487*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5488*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5489*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5490*4bdc9457SAndroid Build Coastguard Worker .dilation_width(3)
5491*4bdc9457SAndroid Build Coastguard Worker .stride_width(2)
5492*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5493*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5494*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5495*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5496*4bdc9457SAndroid Build Coastguard Worker }
5497*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3_with_input_stride)5498*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3_with_input_stride) {
5499*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5500*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5501*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5502*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5503*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5504*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5505*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5506*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5507*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
5508*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5509*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5510*4bdc9457SAndroid Build Coastguard Worker }
5511*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3_with_output_stride)5512*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3_with_output_stride) {
5513*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5514*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5515*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5516*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5517*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5518*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5519*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5520*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5521*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qu8.gemm.nr * 2 + 13)
5522*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5523*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5524*4bdc9457SAndroid Build Coastguard Worker }
5525*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3_with_qmin)5526*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3_with_qmin) {
5527*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5528*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5529*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5530*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5531*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5532*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5533*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5534*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5535*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
5536*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5537*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5538*4bdc9457SAndroid Build Coastguard Worker }
5539*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3_with_qmax)5540*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3_with_qmax) {
5541*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5542*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5543*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5544*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5545*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5546*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5547*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5548*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5549*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
5550*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5551*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5552*4bdc9457SAndroid Build Coastguard Worker }
5553*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3_without_bias)5554*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3_without_bias) {
5555*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5556*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5557*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
5558*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5559*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5560*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5561*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5562*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5563*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5564*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5565*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5566*4bdc9457SAndroid Build Coastguard Worker }
5567*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,weights_cache_batched_3x3)5568*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, weights_cache_batched_3x3) {
5569*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5570*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5571*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5572*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5573*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5574*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5575*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
5576*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5577*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
5578*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5579*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5580*4bdc9457SAndroid Build Coastguard Worker }
5581*4bdc9457SAndroid Build Coastguard Worker
5582*4bdc9457SAndroid Build Coastguard Worker /**************************** CONV path, grouped, batched ****************************/
5583*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3)5584*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3) {
5585*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5586*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5587*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5588*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5589*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5590*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5591*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5592*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
5593*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5594*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5595*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5596*4bdc9457SAndroid Build Coastguard Worker }
5597*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_Kx3)5598*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_Kx3) {
5599*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5600*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 1; kernel_height <= 4; kernel_height *= 2) {
5601*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5602*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5603*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5604*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
5605*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
5606*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5607*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
5608*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5609*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5610*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5611*4bdc9457SAndroid Build Coastguard Worker }
5612*4bdc9457SAndroid Build Coastguard Worker }
5613*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3xK)5614*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3xK) {
5615*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5616*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 1; kernel_width <= 4; kernel_width *= 2) {
5617*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5618*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5619*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5620*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
5621*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
5622*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5623*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
5624*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5625*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5626*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5627*4bdc9457SAndroid Build Coastguard Worker }
5628*4bdc9457SAndroid Build Coastguard Worker }
5629*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3_varying_height_padding)5630*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3_varying_height_padding) {
5631*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5632*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
5633*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
5634*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5635*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5636*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5637*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
5638*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
5639*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
5640*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5641*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5642*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
5643*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5644*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
5645*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5646*4bdc9457SAndroid Build Coastguard Worker }
5647*4bdc9457SAndroid Build Coastguard Worker }
5648*4bdc9457SAndroid Build Coastguard Worker }
5649*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3_varying_width_padding)5650*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3_varying_width_padding) {
5651*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5652*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
5653*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
5654*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5655*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5656*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5657*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
5658*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
5659*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
5660*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5661*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5662*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
5663*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5664*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
5665*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5666*4bdc9457SAndroid Build Coastguard Worker }
5667*4bdc9457SAndroid Build Coastguard Worker }
5668*4bdc9457SAndroid Build Coastguard Worker }
5669*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3_varying_height_adjustment)5670*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3_varying_height_adjustment) {
5671*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5672*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 1; adjustment_height <= 2; adjustment_height++) {
5673*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5674*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5675*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5676*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5677*4bdc9457SAndroid Build Coastguard Worker .stride_height(adjustment_height + 1)
5678*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
5679*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5680*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5681*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
5682*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5683*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
5684*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5685*4bdc9457SAndroid Build Coastguard Worker }
5686*4bdc9457SAndroid Build Coastguard Worker }
5687*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3_varying_width_adjustment)5688*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3_varying_width_adjustment) {
5689*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5690*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 1; adjustment_width <= 2; adjustment_width++) {
5691*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5692*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5693*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5694*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5695*4bdc9457SAndroid Build Coastguard Worker .stride_width(adjustment_width + 1)
5696*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
5697*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5698*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5699*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
5700*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5701*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
5702*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5703*4bdc9457SAndroid Build Coastguard Worker }
5704*4bdc9457SAndroid Build Coastguard Worker }
5705*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3_varying_input_height)5706*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3_varying_input_height) {
5707*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5708*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
5709*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5710*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5711*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
5712*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5713*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5714*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5715*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
5716*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5717*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
5718*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5719*4bdc9457SAndroid Build Coastguard Worker }
5720*4bdc9457SAndroid Build Coastguard Worker }
5721*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3_varying_input_width)5722*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3_varying_input_width) {
5723*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5724*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
5725*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5726*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5727*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
5728*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5729*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5730*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5731*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
5732*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5733*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
5734*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5735*4bdc9457SAndroid Build Coastguard Worker }
5736*4bdc9457SAndroid Build Coastguard Worker }
5737*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3_varying_input_channels)5738*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3_varying_input_channels) {
5739*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5740*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
5741*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5742*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5743*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5744*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5745*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5746*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5747*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
5748*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5749*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
5750*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5751*4bdc9457SAndroid Build Coastguard Worker }
5752*4bdc9457SAndroid Build Coastguard Worker }
5753*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3_varying_output_channels)5754*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3_varying_output_channels) {
5755*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5756*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qu8.gemm.nr * 2; output_channels *= 2) {
5757*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5758*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5759*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5760*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5761*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5762*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5763*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5764*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
5765*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
5766*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5767*4bdc9457SAndroid Build Coastguard Worker }
5768*4bdc9457SAndroid Build Coastguard Worker }
5769*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3_with_height_dilation)5770*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3_with_height_dilation) {
5771*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5772*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_height = 2; dilation_height <= 3; dilation_height++) {
5773*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5774*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5775*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5776*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5777*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5778*4bdc9457SAndroid Build Coastguard Worker .dilation_height(dilation_height)
5779*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5780*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5781*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5782*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5783*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5784*4bdc9457SAndroid Build Coastguard Worker }
5785*4bdc9457SAndroid Build Coastguard Worker }
5786*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3_with_width_dilation)5787*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3_with_width_dilation) {
5788*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5789*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) {
5790*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5791*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5792*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5793*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5794*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5795*4bdc9457SAndroid Build Coastguard Worker .dilation_width(dilation_width)
5796*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5797*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5798*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5799*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5800*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5801*4bdc9457SAndroid Build Coastguard Worker }
5802*4bdc9457SAndroid Build Coastguard Worker }
5803*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3_with_height_dilation_and_stride)5804*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3_with_height_dilation_and_stride) {
5805*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5806*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5807*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5808*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5809*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5810*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5811*4bdc9457SAndroid Build Coastguard Worker .dilation_height(3)
5812*4bdc9457SAndroid Build Coastguard Worker .stride_width(2)
5813*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5814*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5815*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5816*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5817*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5818*4bdc9457SAndroid Build Coastguard Worker }
5819*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3_with_width_dilation_and_stride)5820*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3_with_width_dilation_and_stride) {
5821*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5822*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5823*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5824*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5825*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5826*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5827*4bdc9457SAndroid Build Coastguard Worker .dilation_width(3)
5828*4bdc9457SAndroid Build Coastguard Worker .stride_width(2)
5829*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5830*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5831*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5832*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5833*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5834*4bdc9457SAndroid Build Coastguard Worker }
5835*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3_with_input_stride)5836*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3_with_input_stride) {
5837*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5838*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5839*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5840*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5841*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5842*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5843*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5844*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5845*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5846*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(47)
5847*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5848*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5849*4bdc9457SAndroid Build Coastguard Worker }
5850*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3_with_output_stride)5851*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3_with_output_stride) {
5852*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5853*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5854*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5855*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5856*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5857*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5858*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5859*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5860*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr + 3)
5861*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qu8.gemm.nr * 2 + 13)
5862*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5863*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5864*4bdc9457SAndroid Build Coastguard Worker }
5865*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3_with_qmin)5866*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3_with_qmin) {
5867*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5868*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5869*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5870*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5871*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5872*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5873*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5874*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5875*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5876*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
5877*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5878*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5879*4bdc9457SAndroid Build Coastguard Worker }
5880*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3_with_qmax)5881*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3_with_qmax) {
5882*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5883*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5884*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5885*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5886*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5887*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5888*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5889*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5890*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5891*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
5892*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5893*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5894*4bdc9457SAndroid Build Coastguard Worker }
5895*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3_without_bias)5896*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3_without_bias) {
5897*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5898*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5899*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
5900*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5901*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5902*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5903*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5904*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5905*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
5906*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5907*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5908*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5909*4bdc9457SAndroid Build Coastguard Worker }
5910*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,weights_cache_batched_grouped_3x3)5911*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, weights_cache_batched_grouped_3x3) {
5912*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5913*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5914*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5915*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5916*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5917*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5918*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5919*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
5920*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5921*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
5922*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5923*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5924*4bdc9457SAndroid Build Coastguard Worker }
5925*4bdc9457SAndroid Build Coastguard Worker
5926*4bdc9457SAndroid Build Coastguard Worker /**************************** CONV path, setup ****************************/
5927*4bdc9457SAndroid Build Coastguard Worker
5928*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3_setup_changing_batch) {
5929*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5930*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5931*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5932*4bdc9457SAndroid Build Coastguard Worker .next_batch_size(5)
5933*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5934*4bdc9457SAndroid Build Coastguard Worker .kernel_height(3)
5935*4bdc9457SAndroid Build Coastguard Worker .kernel_width(5)
5936*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5937*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
5938*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
5939*4bdc9457SAndroid Build Coastguard Worker .TestSetupQU8();
5940*4bdc9457SAndroid Build Coastguard Worker }
5941*4bdc9457SAndroid Build Coastguard Worker
5942*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3_setup_changing_height) {
5943*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5944*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5945*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5946*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5947*4bdc9457SAndroid Build Coastguard Worker .next_input_height(kUnstridedInputHeight + 3)
5948*4bdc9457SAndroid Build Coastguard Worker .kernel_height(3)
5949*4bdc9457SAndroid Build Coastguard Worker .kernel_width(5)
5950*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5951*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
5952*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
5953*4bdc9457SAndroid Build Coastguard Worker .TestSetupQU8();
5954*4bdc9457SAndroid Build Coastguard Worker }
5955*4bdc9457SAndroid Build Coastguard Worker
5956*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3_setup_changing_width) {
5957*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5958*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5959*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
5960*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
5961*4bdc9457SAndroid Build Coastguard Worker .next_input_width(kUnstridedInputWidth + 3)
5962*4bdc9457SAndroid Build Coastguard Worker .kernel_height(3)
5963*4bdc9457SAndroid Build Coastguard Worker .kernel_width(5)
5964*4bdc9457SAndroid Build Coastguard Worker .groups(2)
5965*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
5966*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
5967*4bdc9457SAndroid Build Coastguard Worker .TestSetupQU8();
5968*4bdc9457SAndroid Build Coastguard Worker }
5969*4bdc9457SAndroid Build Coastguard Worker
5970*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/IGEMM path ****************************/
5971*4bdc9457SAndroid Build Coastguard Worker
5972*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3s2) {
5973*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5974*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5975*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
5976*4bdc9457SAndroid Build Coastguard Worker .padding(1)
5977*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
5978*4bdc9457SAndroid Build Coastguard Worker .stride(2)
5979*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
5980*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5981*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5982*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5983*4bdc9457SAndroid Build Coastguard Worker }
5984*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,Kx3s2)5985*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, Kx3s2) {
5986*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
5987*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 2; kernel_height <= 5; kernel_height++) {
5988*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
5989*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
5990*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
5991*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
5992*4bdc9457SAndroid Build Coastguard Worker .stride(2)
5993*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
5994*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
5995*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
5996*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
5997*4bdc9457SAndroid Build Coastguard Worker }
5998*4bdc9457SAndroid Build Coastguard Worker }
5999*4bdc9457SAndroid Build Coastguard Worker
6000*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3xKs2) {
6001*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6002*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 2; kernel_width <= 5; kernel_width++) {
6003*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6004*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6005*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
6006*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
6007*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6008*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6009*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6010*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6011*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6012*4bdc9457SAndroid Build Coastguard Worker }
6013*4bdc9457SAndroid Build Coastguard Worker }
6014*4bdc9457SAndroid Build Coastguard Worker
6015*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3sSx1) {
6016*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6017*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_height = 2; stride_height <= 3; stride_height++) {
6018*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6019*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6020*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6021*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
6022*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6023*4bdc9457SAndroid Build Coastguard Worker .stride_height(stride_height)
6024*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6025*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6026*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6027*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6028*4bdc9457SAndroid Build Coastguard Worker }
6029*4bdc9457SAndroid Build Coastguard Worker }
6030*4bdc9457SAndroid Build Coastguard Worker
6031*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3s1xS) {
6032*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6033*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_width = 2; stride_width <= 3; stride_width++) {
6034*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6035*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6036*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6037*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
6038*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6039*4bdc9457SAndroid Build Coastguard Worker .stride_width(stride_width)
6040*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6041*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6042*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6043*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6044*4bdc9457SAndroid Build Coastguard Worker }
6045*4bdc9457SAndroid Build Coastguard Worker }
6046*4bdc9457SAndroid Build Coastguard Worker
6047*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3s2_varying_height_padding) {
6048*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6049*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
6050*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
6051*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6052*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6053*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
6054*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
6055*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
6056*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6057*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6058*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
6059*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6060*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
6061*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6062*4bdc9457SAndroid Build Coastguard Worker }
6063*4bdc9457SAndroid Build Coastguard Worker }
6064*4bdc9457SAndroid Build Coastguard Worker }
6065*4bdc9457SAndroid Build Coastguard Worker
6066*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3s2_varying_width_padding) {
6067*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6068*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
6069*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
6070*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6071*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6072*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
6073*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
6074*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
6075*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6076*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6077*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
6078*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6079*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
6080*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6081*4bdc9457SAndroid Build Coastguard Worker }
6082*4bdc9457SAndroid Build Coastguard Worker }
6083*4bdc9457SAndroid Build Coastguard Worker }
6084*4bdc9457SAndroid Build Coastguard Worker
6085*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3s2_varying_height_adjustment) {
6086*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6087*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 0; adjustment_height <= 1; adjustment_height++) {
6088*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6089*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6090*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6091*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
6092*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6093*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6094*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
6095*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6096*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
6097*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6098*4bdc9457SAndroid Build Coastguard Worker }
6099*4bdc9457SAndroid Build Coastguard Worker }
6100*4bdc9457SAndroid Build Coastguard Worker
6101*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3s2_varying_width_adjustment) {
6102*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6103*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 0; adjustment_width <= 1; adjustment_width++) {
6104*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6105*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6106*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6107*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
6108*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6109*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6110*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
6111*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6112*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
6113*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6114*4bdc9457SAndroid Build Coastguard Worker }
6115*4bdc9457SAndroid Build Coastguard Worker }
6116*4bdc9457SAndroid Build Coastguard Worker
6117*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3s2_varying_input_height) {
6118*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6119*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
6120*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6121*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
6122*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6123*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6124*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6125*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
6126*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6127*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
6128*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6129*4bdc9457SAndroid Build Coastguard Worker }
6130*4bdc9457SAndroid Build Coastguard Worker }
6131*4bdc9457SAndroid Build Coastguard Worker
6132*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3s2_varying_input_width) {
6133*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6134*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
6135*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6136*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6137*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6138*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6139*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6140*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
6141*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6142*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
6143*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6144*4bdc9457SAndroid Build Coastguard Worker }
6145*4bdc9457SAndroid Build Coastguard Worker }
6146*4bdc9457SAndroid Build Coastguard Worker
6147*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3s2_varying_input_channels) {
6148*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6149*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
6150*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6151*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6152*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6153*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6154*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6155*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
6156*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6157*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
6158*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6159*4bdc9457SAndroid Build Coastguard Worker }
6160*4bdc9457SAndroid Build Coastguard Worker }
6161*4bdc9457SAndroid Build Coastguard Worker
6162*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3s2_varying_output_channels) {
6163*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6164*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qu8.gemm.nr * 2; output_channels *= 2) {
6165*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6166*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6167*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6168*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6169*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6170*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
6171*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
6172*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
6173*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6174*4bdc9457SAndroid Build Coastguard Worker }
6175*4bdc9457SAndroid Build Coastguard Worker }
6176*4bdc9457SAndroid Build Coastguard Worker
6177*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3s2_with_input_stride) {
6178*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6179*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6180*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6181*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6182*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6183*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6184*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
6185*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6186*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
6187*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6188*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6189*4bdc9457SAndroid Build Coastguard Worker }
6190*4bdc9457SAndroid Build Coastguard Worker
6191*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3s2_with_output_stride) {
6192*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6193*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6194*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6195*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6196*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6197*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6198*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
6199*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6200*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qu8.gemm.nr * 2 + 13)
6201*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6202*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6203*4bdc9457SAndroid Build Coastguard Worker }
6204*4bdc9457SAndroid Build Coastguard Worker
6205*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3s2_with_qmin) {
6206*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6207*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6208*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6209*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6210*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6211*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6212*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
6213*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6214*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
6215*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6216*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6217*4bdc9457SAndroid Build Coastguard Worker }
6218*4bdc9457SAndroid Build Coastguard Worker
6219*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3s2_with_qmax) {
6220*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6221*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6222*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6223*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6224*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6225*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6226*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
6227*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6228*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
6229*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6230*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6231*4bdc9457SAndroid Build Coastguard Worker }
6232*4bdc9457SAndroid Build Coastguard Worker
6233*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3s2_without_bias) {
6234*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6235*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6236*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
6237*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6238*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6239*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6240*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6241*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
6242*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6243*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6244*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6245*4bdc9457SAndroid Build Coastguard Worker }
6246*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,weights_cache_3x3s2)6247*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, weights_cache_3x3s2) {
6248*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6249*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6250*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6251*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6252*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6253*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6254*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
6255*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6256*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
6257*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6258*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6259*4bdc9457SAndroid Build Coastguard Worker }
6260*4bdc9457SAndroid Build Coastguard Worker
6261*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/IGEMM path, grouped ****************************/
6262*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3s2)6263*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3s2) {
6264*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6265*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6266*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6267*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6268*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6269*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6270*4bdc9457SAndroid Build Coastguard Worker .groups(2)
6271*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6272*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6273*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6274*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6275*4bdc9457SAndroid Build Coastguard Worker }
6276*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_Kx3s2)6277*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_Kx3s2) {
6278*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6279*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 2; kernel_height <= 5; kernel_height++) {
6280*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6281*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6282*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
6283*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
6284*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6285*4bdc9457SAndroid Build Coastguard Worker .groups(2)
6286*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6287*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6288*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6289*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6290*4bdc9457SAndroid Build Coastguard Worker }
6291*4bdc9457SAndroid Build Coastguard Worker }
6292*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3xKs2)6293*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3xKs2) {
6294*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6295*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 2; kernel_width <= 5; kernel_width++) {
6296*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6297*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6298*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
6299*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
6300*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6301*4bdc9457SAndroid Build Coastguard Worker .groups(2)
6302*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6303*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6304*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6305*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6306*4bdc9457SAndroid Build Coastguard Worker }
6307*4bdc9457SAndroid Build Coastguard Worker }
6308*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3sSx1)6309*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3sSx1) {
6310*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6311*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_height = 2; stride_height <= 3; stride_height++) {
6312*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6313*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6314*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6315*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
6316*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6317*4bdc9457SAndroid Build Coastguard Worker .stride_height(stride_height)
6318*4bdc9457SAndroid Build Coastguard Worker .groups(2)
6319*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6320*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6321*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6322*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6323*4bdc9457SAndroid Build Coastguard Worker }
6324*4bdc9457SAndroid Build Coastguard Worker }
6325*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3s1xS)6326*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3s1xS) {
6327*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6328*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_width = 2; stride_width <= 3; stride_width++) {
6329*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6330*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6331*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6332*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
6333*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6334*4bdc9457SAndroid Build Coastguard Worker .stride_width(stride_width)
6335*4bdc9457SAndroid Build Coastguard Worker .groups(2)
6336*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6337*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6338*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6339*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6340*4bdc9457SAndroid Build Coastguard Worker }
6341*4bdc9457SAndroid Build Coastguard Worker }
6342*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3s2_varying_height_padding)6343*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3s2_varying_height_padding) {
6344*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6345*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
6346*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
6347*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6348*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6349*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
6350*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
6351*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
6352*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6353*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6354*4bdc9457SAndroid Build Coastguard Worker .groups(2)
6355*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6356*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6357*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
6358*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6359*4bdc9457SAndroid Build Coastguard Worker }
6360*4bdc9457SAndroid Build Coastguard Worker }
6361*4bdc9457SAndroid Build Coastguard Worker }
6362*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3s2_varying_width_padding)6363*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3s2_varying_width_padding) {
6364*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6365*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
6366*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
6367*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6368*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6369*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
6370*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
6371*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
6372*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6373*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6374*4bdc9457SAndroid Build Coastguard Worker .groups(2)
6375*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6376*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6377*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
6378*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6379*4bdc9457SAndroid Build Coastguard Worker }
6380*4bdc9457SAndroid Build Coastguard Worker }
6381*4bdc9457SAndroid Build Coastguard Worker }
6382*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3s2_varying_height_adjustment)6383*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3s2_varying_height_adjustment) {
6384*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6385*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 0; adjustment_height <= 1; adjustment_height++) {
6386*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6387*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6388*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6389*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
6390*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6391*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6392*4bdc9457SAndroid Build Coastguard Worker .groups(2)
6393*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6394*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6395*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
6396*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6397*4bdc9457SAndroid Build Coastguard Worker }
6398*4bdc9457SAndroid Build Coastguard Worker }
6399*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3s2_varying_width_adjustment)6400*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3s2_varying_width_adjustment) {
6401*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6402*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 0; adjustment_width <= 1; adjustment_width++) {
6403*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6404*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6405*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6406*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
6407*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6408*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6409*4bdc9457SAndroid Build Coastguard Worker .groups(2)
6410*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6411*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6412*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
6413*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6414*4bdc9457SAndroid Build Coastguard Worker }
6415*4bdc9457SAndroid Build Coastguard Worker }
6416*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3s2_varying_input_height)6417*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3s2_varying_input_height) {
6418*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6419*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
6420*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6421*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
6422*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6423*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6424*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6425*4bdc9457SAndroid Build Coastguard Worker .groups(2)
6426*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6427*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6428*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
6429*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6430*4bdc9457SAndroid Build Coastguard Worker }
6431*4bdc9457SAndroid Build Coastguard Worker }
6432*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3s2_varying_input_width)6433*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3s2_varying_input_width) {
6434*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6435*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
6436*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6437*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6438*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6439*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6440*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6441*4bdc9457SAndroid Build Coastguard Worker .groups(2)
6442*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6443*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6444*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
6445*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6446*4bdc9457SAndroid Build Coastguard Worker }
6447*4bdc9457SAndroid Build Coastguard Worker }
6448*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3s2_varying_input_channels)6449*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3s2_varying_input_channels) {
6450*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6451*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 14; input_channels <= 20; input_channels++) {
6452*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6453*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6454*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6455*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6456*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6457*4bdc9457SAndroid Build Coastguard Worker .groups(2)
6458*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
6459*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6460*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
6461*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6462*4bdc9457SAndroid Build Coastguard Worker }
6463*4bdc9457SAndroid Build Coastguard Worker }
6464*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3s2_varying_output_channels)6465*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3s2_varying_output_channels) {
6466*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6467*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qu8.gemm.nr * 2; output_channels *= 2) {
6468*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6469*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6470*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6471*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6472*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6473*4bdc9457SAndroid Build Coastguard Worker .groups(2)
6474*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6475*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
6476*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
6477*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6478*4bdc9457SAndroid Build Coastguard Worker }
6479*4bdc9457SAndroid Build Coastguard Worker }
6480*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3s2_with_input_stride)6481*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3s2_with_input_stride) {
6482*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6483*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6484*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6485*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6486*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6487*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6488*4bdc9457SAndroid Build Coastguard Worker .groups(2)
6489*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6490*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6491*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(37)
6492*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6493*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6494*4bdc9457SAndroid Build Coastguard Worker }
6495*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3s2_with_output_stride)6496*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3s2_with_output_stride) {
6497*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6498*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6499*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6500*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6501*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6502*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6503*4bdc9457SAndroid Build Coastguard Worker .groups(2)
6504*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6505*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr + 3)
6506*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qu8.gemm.nr * 2 + 13)
6507*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6508*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6509*4bdc9457SAndroid Build Coastguard Worker }
6510*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3s2_with_qmin)6511*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3s2_with_qmin) {
6512*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6513*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6514*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6515*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6516*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6517*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6518*4bdc9457SAndroid Build Coastguard Worker .groups(2)
6519*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6520*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6521*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
6522*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6523*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6524*4bdc9457SAndroid Build Coastguard Worker }
6525*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3s2_with_qmax)6526*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3s2_with_qmax) {
6527*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6528*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6529*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6530*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6531*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6532*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6533*4bdc9457SAndroid Build Coastguard Worker .groups(2)
6534*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6535*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6536*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
6537*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6538*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6539*4bdc9457SAndroid Build Coastguard Worker }
6540*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_3x3s2_without_bias)6541*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_3x3s2_without_bias) {
6542*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6543*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6544*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
6545*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6546*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6547*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6548*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6549*4bdc9457SAndroid Build Coastguard Worker .groups(2)
6550*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6551*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6552*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6553*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6554*4bdc9457SAndroid Build Coastguard Worker }
6555*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,weights_cache_grouped_3x3s2)6556*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, weights_cache_grouped_3x3s2) {
6557*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6558*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6559*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6560*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6561*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6562*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6563*4bdc9457SAndroid Build Coastguard Worker .groups(2)
6564*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6565*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6566*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
6567*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6568*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6569*4bdc9457SAndroid Build Coastguard Worker }
6570*4bdc9457SAndroid Build Coastguard Worker
6571*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/IGEMM path, batched ****************************/
6572*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3s2)6573*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3s2) {
6574*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6575*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6576*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
6577*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6578*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6579*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6580*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6581*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
6582*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6583*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6584*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6585*4bdc9457SAndroid Build Coastguard Worker }
6586*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_Kx3s2)6587*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_Kx3s2) {
6588*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6589*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 2; kernel_height <= 5; kernel_height++) {
6590*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6591*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
6592*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6593*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
6594*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
6595*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6596*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6597*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6598*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6599*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6600*4bdc9457SAndroid Build Coastguard Worker }
6601*4bdc9457SAndroid Build Coastguard Worker }
6602*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3xKs2)6603*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3xKs2) {
6604*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6605*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 2; kernel_width <= 5; kernel_width++) {
6606*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6607*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
6608*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6609*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
6610*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
6611*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6612*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6613*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6614*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6615*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6616*4bdc9457SAndroid Build Coastguard Worker }
6617*4bdc9457SAndroid Build Coastguard Worker }
6618*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3sSx1)6619*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3sSx1) {
6620*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6621*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_height = 2; stride_height <= 3; stride_height++) {
6622*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6623*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
6624*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6625*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6626*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
6627*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6628*4bdc9457SAndroid Build Coastguard Worker .stride_height(stride_height)
6629*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6630*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6631*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6632*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6633*4bdc9457SAndroid Build Coastguard Worker }
6634*4bdc9457SAndroid Build Coastguard Worker }
6635*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3s1xS)6636*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3s1xS) {
6637*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6638*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_width = 2; stride_width <= 3; stride_width++) {
6639*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6640*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
6641*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6642*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6643*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
6644*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6645*4bdc9457SAndroid Build Coastguard Worker .stride_width(stride_width)
6646*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6647*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6648*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6649*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6650*4bdc9457SAndroid Build Coastguard Worker }
6651*4bdc9457SAndroid Build Coastguard Worker }
6652*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3s2_varying_height_padding)6653*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3s2_varying_height_padding) {
6654*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6655*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
6656*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
6657*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6658*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
6659*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6660*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
6661*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
6662*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
6663*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6664*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6665*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
6666*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6667*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
6668*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6669*4bdc9457SAndroid Build Coastguard Worker }
6670*4bdc9457SAndroid Build Coastguard Worker }
6671*4bdc9457SAndroid Build Coastguard Worker }
6672*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3s2_varying_width_padding)6673*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3s2_varying_width_padding) {
6674*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6675*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
6676*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
6677*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6678*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
6679*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6680*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
6681*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
6682*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
6683*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6684*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6685*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
6686*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6687*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
6688*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6689*4bdc9457SAndroid Build Coastguard Worker }
6690*4bdc9457SAndroid Build Coastguard Worker }
6691*4bdc9457SAndroid Build Coastguard Worker }
6692*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3s2_varying_height_adjustment)6693*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3s2_varying_height_adjustment) {
6694*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6695*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 0; adjustment_height <= 1; adjustment_height++) {
6696*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6697*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
6698*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6699*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6700*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
6701*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6702*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6703*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
6704*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6705*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
6706*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6707*4bdc9457SAndroid Build Coastguard Worker }
6708*4bdc9457SAndroid Build Coastguard Worker }
6709*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3s2_varying_width_adjustment)6710*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3s2_varying_width_adjustment) {
6711*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6712*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 0; adjustment_width <= 1; adjustment_width++) {
6713*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6714*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
6715*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6716*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6717*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
6718*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6719*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6720*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
6721*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6722*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
6723*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6724*4bdc9457SAndroid Build Coastguard Worker }
6725*4bdc9457SAndroid Build Coastguard Worker }
6726*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3s2_varying_input_height)6727*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3s2_varying_input_height) {
6728*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6729*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
6730*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6731*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
6732*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
6733*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6734*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6735*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6736*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
6737*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6738*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
6739*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6740*4bdc9457SAndroid Build Coastguard Worker }
6741*4bdc9457SAndroid Build Coastguard Worker }
6742*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3s2_varying_input_width)6743*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3s2_varying_input_width) {
6744*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6745*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
6746*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6747*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
6748*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6749*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6750*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6751*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6752*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
6753*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6754*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
6755*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6756*4bdc9457SAndroid Build Coastguard Worker }
6757*4bdc9457SAndroid Build Coastguard Worker }
6758*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3s2_varying_input_channels)6759*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3s2_varying_input_channels) {
6760*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6761*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
6762*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6763*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
6764*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6765*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6766*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6767*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6768*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
6769*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6770*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
6771*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6772*4bdc9457SAndroid Build Coastguard Worker }
6773*4bdc9457SAndroid Build Coastguard Worker }
6774*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3s2_varying_output_channels)6775*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3s2_varying_output_channels) {
6776*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6777*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qu8.gemm.nr * 2; output_channels *= 2) {
6778*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6779*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
6780*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6781*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6782*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6783*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6784*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
6785*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
6786*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
6787*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6788*4bdc9457SAndroid Build Coastguard Worker }
6789*4bdc9457SAndroid Build Coastguard Worker }
6790*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3s2_with_input_stride)6791*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3s2_with_input_stride) {
6792*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6793*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6794*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
6795*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6796*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6797*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6798*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6799*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
6800*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6801*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
6802*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6803*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6804*4bdc9457SAndroid Build Coastguard Worker }
6805*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3s2_with_output_stride)6806*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3s2_with_output_stride) {
6807*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6808*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6809*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
6810*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6811*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6812*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6813*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6814*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
6815*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6816*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qu8.gemm.nr * 2 + 13)
6817*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6818*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6819*4bdc9457SAndroid Build Coastguard Worker }
6820*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3s2_with_qmin)6821*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3s2_with_qmin) {
6822*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6823*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6824*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
6825*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6826*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6827*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6828*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6829*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
6830*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6831*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
6832*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6833*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6834*4bdc9457SAndroid Build Coastguard Worker }
6835*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3s2_with_qmax)6836*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3s2_with_qmax) {
6837*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6838*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6839*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
6840*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6841*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6842*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6843*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6844*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
6845*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6846*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
6847*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6848*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6849*4bdc9457SAndroid Build Coastguard Worker }
6850*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_3x3s2_without_bias)6851*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_3x3s2_without_bias) {
6852*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6853*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6854*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
6855*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
6856*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6857*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6858*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6859*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6860*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
6861*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6862*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6863*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6864*4bdc9457SAndroid Build Coastguard Worker }
6865*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,weights_cache_batched_3x3s2)6866*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, weights_cache_batched_3x3s2) {
6867*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6868*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6869*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
6870*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6871*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6872*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6873*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6874*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
6875*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6876*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
6877*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6878*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6879*4bdc9457SAndroid Build Coastguard Worker }
6880*4bdc9457SAndroid Build Coastguard Worker
6881*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/IGEMM path, grouped, batched ****************************/
6882*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3s2)6883*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3s2) {
6884*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6885*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6886*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
6887*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6888*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6889*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6890*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6891*4bdc9457SAndroid Build Coastguard Worker .groups(2)
6892*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6893*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6894*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6895*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6896*4bdc9457SAndroid Build Coastguard Worker }
6897*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_Kx3s2)6898*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_Kx3s2) {
6899*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6900*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 2; kernel_height <= 5; kernel_height++) {
6901*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6902*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
6903*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6904*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
6905*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
6906*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6907*4bdc9457SAndroid Build Coastguard Worker .groups(2)
6908*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6909*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6910*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6911*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6912*4bdc9457SAndroid Build Coastguard Worker }
6913*4bdc9457SAndroid Build Coastguard Worker }
6914*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3xKs2)6915*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3xKs2) {
6916*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6917*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 2; kernel_width <= 5; kernel_width++) {
6918*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6919*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
6920*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6921*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
6922*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
6923*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6924*4bdc9457SAndroid Build Coastguard Worker .groups(2)
6925*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6926*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6927*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6928*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6929*4bdc9457SAndroid Build Coastguard Worker }
6930*4bdc9457SAndroid Build Coastguard Worker }
6931*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3sSx1)6932*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3sSx1) {
6933*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6934*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_height = 2; stride_height <= 3; stride_height++) {
6935*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6936*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
6937*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6938*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6939*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
6940*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6941*4bdc9457SAndroid Build Coastguard Worker .stride_height(stride_height)
6942*4bdc9457SAndroid Build Coastguard Worker .groups(2)
6943*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6944*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6945*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6946*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6947*4bdc9457SAndroid Build Coastguard Worker }
6948*4bdc9457SAndroid Build Coastguard Worker }
6949*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3s1xS)6950*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3s1xS) {
6951*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6952*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_width = 2; stride_width <= 3; stride_width++) {
6953*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6954*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
6955*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6956*4bdc9457SAndroid Build Coastguard Worker .padding(1)
6957*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
6958*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6959*4bdc9457SAndroid Build Coastguard Worker .stride_width(stride_width)
6960*4bdc9457SAndroid Build Coastguard Worker .groups(2)
6961*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6962*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6963*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
6964*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6965*4bdc9457SAndroid Build Coastguard Worker }
6966*4bdc9457SAndroid Build Coastguard Worker }
6967*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3s2_varying_height_padding)6968*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3s2_varying_height_padding) {
6969*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6970*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
6971*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
6972*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6973*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
6974*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6975*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
6976*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
6977*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
6978*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
6979*4bdc9457SAndroid Build Coastguard Worker .stride(2)
6980*4bdc9457SAndroid Build Coastguard Worker .groups(2)
6981*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
6982*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
6983*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
6984*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
6985*4bdc9457SAndroid Build Coastguard Worker }
6986*4bdc9457SAndroid Build Coastguard Worker }
6987*4bdc9457SAndroid Build Coastguard Worker }
6988*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3s2_varying_width_padding)6989*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3s2_varying_width_padding) {
6990*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
6991*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
6992*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
6993*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
6994*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
6995*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
6996*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
6997*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
6998*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
6999*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
7000*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7001*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7002*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7003*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7004*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7005*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7006*4bdc9457SAndroid Build Coastguard Worker }
7007*4bdc9457SAndroid Build Coastguard Worker }
7008*4bdc9457SAndroid Build Coastguard Worker }
7009*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3s2_varying_height_adjustment)7010*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3s2_varying_height_adjustment) {
7011*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7012*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 0; adjustment_height <= 1; adjustment_height++) {
7013*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7014*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7015*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7016*4bdc9457SAndroid Build Coastguard Worker .padding(1)
7017*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
7018*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
7019*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7020*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7021*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7022*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7023*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7024*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7025*4bdc9457SAndroid Build Coastguard Worker }
7026*4bdc9457SAndroid Build Coastguard Worker }
7027*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3s2_varying_width_adjustment)7028*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3s2_varying_width_adjustment) {
7029*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7030*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 0; adjustment_width <= 1; adjustment_width++) {
7031*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7032*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7033*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7034*4bdc9457SAndroid Build Coastguard Worker .padding(1)
7035*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
7036*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
7037*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7038*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7039*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7040*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7041*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7042*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7043*4bdc9457SAndroid Build Coastguard Worker }
7044*4bdc9457SAndroid Build Coastguard Worker }
7045*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3s2_varying_input_height)7046*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3s2_varying_input_height) {
7047*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7048*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
7049*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7050*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7051*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
7052*4bdc9457SAndroid Build Coastguard Worker .padding(1)
7053*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
7054*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7055*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7056*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7057*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7058*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7059*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7060*4bdc9457SAndroid Build Coastguard Worker }
7061*4bdc9457SAndroid Build Coastguard Worker }
7062*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3s2_varying_input_width)7063*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3s2_varying_input_width) {
7064*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7065*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
7066*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7067*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7068*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7069*4bdc9457SAndroid Build Coastguard Worker .padding(1)
7070*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
7071*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7072*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7073*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7074*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7075*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7076*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7077*4bdc9457SAndroid Build Coastguard Worker }
7078*4bdc9457SAndroid Build Coastguard Worker }
7079*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3s2_varying_input_channels)7080*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3s2_varying_input_channels) {
7081*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7082*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 14; input_channels <= 20; input_channels++) {
7083*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7084*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7085*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7086*4bdc9457SAndroid Build Coastguard Worker .padding(1)
7087*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
7088*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7089*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7090*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
7091*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7092*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7093*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7094*4bdc9457SAndroid Build Coastguard Worker }
7095*4bdc9457SAndroid Build Coastguard Worker }
7096*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3s2_varying_output_channels)7097*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3s2_varying_output_channels) {
7098*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7099*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qu8.gemm.nr * 2; output_channels *= 2) {
7100*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7101*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7102*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7103*4bdc9457SAndroid Build Coastguard Worker .padding(1)
7104*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
7105*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7106*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7107*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7108*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
7109*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7110*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7111*4bdc9457SAndroid Build Coastguard Worker }
7112*4bdc9457SAndroid Build Coastguard Worker }
7113*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3s2_with_input_stride)7114*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3s2_with_input_stride) {
7115*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7116*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7117*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7118*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7119*4bdc9457SAndroid Build Coastguard Worker .padding(1)
7120*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
7121*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7122*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7123*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7124*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7125*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(37)
7126*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7127*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7128*4bdc9457SAndroid Build Coastguard Worker }
7129*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3s2_with_output_stride)7130*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3s2_with_output_stride) {
7131*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7132*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7133*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7134*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7135*4bdc9457SAndroid Build Coastguard Worker .padding(1)
7136*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
7137*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7138*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7139*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7140*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr + 3)
7141*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qu8.gemm.nr * 2 + 13)
7142*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7143*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7144*4bdc9457SAndroid Build Coastguard Worker }
7145*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3s2_with_qmin)7146*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3s2_with_qmin) {
7147*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7148*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7149*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7150*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7151*4bdc9457SAndroid Build Coastguard Worker .padding(1)
7152*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
7153*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7154*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7155*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7156*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7157*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
7158*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7159*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7160*4bdc9457SAndroid Build Coastguard Worker }
7161*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3s2_with_qmax)7162*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3s2_with_qmax) {
7163*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7164*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7165*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7166*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7167*4bdc9457SAndroid Build Coastguard Worker .padding(1)
7168*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
7169*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7170*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7171*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7172*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7173*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
7174*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7175*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7176*4bdc9457SAndroid Build Coastguard Worker }
7177*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_3x3s2_without_bias)7178*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_3x3s2_without_bias) {
7179*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7180*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7181*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
7182*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7183*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7184*4bdc9457SAndroid Build Coastguard Worker .padding(1)
7185*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
7186*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7187*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7188*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7189*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7190*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7191*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7192*4bdc9457SAndroid Build Coastguard Worker }
7193*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,weights_cache_batched_grouped_3x3s2)7194*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, weights_cache_batched_grouped_3x3s2) {
7195*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7196*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7197*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7198*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7199*4bdc9457SAndroid Build Coastguard Worker .padding(1)
7200*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
7201*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7202*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7203*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7204*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7205*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
7206*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7207*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7208*4bdc9457SAndroid Build Coastguard Worker }
7209*4bdc9457SAndroid Build Coastguard Worker
7210*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/IGEMM path, setup ****************************/
7211*4bdc9457SAndroid Build Coastguard Worker
7212*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3s2_setup_changing_batch) {
7213*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7214*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7215*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7216*4bdc9457SAndroid Build Coastguard Worker .next_batch_size(5)
7217*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7218*4bdc9457SAndroid Build Coastguard Worker .kernel_height(3)
7219*4bdc9457SAndroid Build Coastguard Worker .kernel_width(5)
7220*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7221*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7222*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
7223*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
7224*4bdc9457SAndroid Build Coastguard Worker .TestSetupQU8();
7225*4bdc9457SAndroid Build Coastguard Worker }
7226*4bdc9457SAndroid Build Coastguard Worker
7227*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3s2_setup_changing_height) {
7228*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7229*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7230*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7231*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7232*4bdc9457SAndroid Build Coastguard Worker .next_input_height(kStridedInputHeight + 3)
7233*4bdc9457SAndroid Build Coastguard Worker .kernel_height(3)
7234*4bdc9457SAndroid Build Coastguard Worker .kernel_width(5)
7235*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7236*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7237*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
7238*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
7239*4bdc9457SAndroid Build Coastguard Worker .TestSetupQU8();
7240*4bdc9457SAndroid Build Coastguard Worker }
7241*4bdc9457SAndroid Build Coastguard Worker
7242*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 3x3s2_setup_changing_width) {
7243*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7244*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7245*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7246*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7247*4bdc9457SAndroid Build Coastguard Worker .next_input_width(kStridedInputWidth + 3)
7248*4bdc9457SAndroid Build Coastguard Worker .kernel_height(3)
7249*4bdc9457SAndroid Build Coastguard Worker .kernel_width(5)
7250*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7251*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7252*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
7253*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
7254*4bdc9457SAndroid Build Coastguard Worker .TestSetupQU8();
7255*4bdc9457SAndroid Build Coastguard Worker }
7256*4bdc9457SAndroid Build Coastguard Worker
7257*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/GEMM path ****************************/
7258*4bdc9457SAndroid Build Coastguard Worker
7259*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 2x2s2) {
7260*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7261*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7262*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7263*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7264*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7265*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
7266*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7267*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7268*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7269*4bdc9457SAndroid Build Coastguard Worker }
7270*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,Kx2sKx2)7271*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, Kx2sKx2) {
7272*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7273*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 3; kernel_height <= 5; kernel_height++) {
7274*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7275*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7276*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 2)
7277*4bdc9457SAndroid Build Coastguard Worker .stride(kernel_height, 2)
7278*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7279*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7280*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7281*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7282*4bdc9457SAndroid Build Coastguard Worker }
7283*4bdc9457SAndroid Build Coastguard Worker }
7284*4bdc9457SAndroid Build Coastguard Worker
7285*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 2xKs2xK) {
7286*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7287*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 3; kernel_width <= 5; kernel_width++) {
7288*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7289*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7290*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, kernel_width)
7291*4bdc9457SAndroid Build Coastguard Worker .stride(2, kernel_width)
7292*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7293*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7294*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7295*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7296*4bdc9457SAndroid Build Coastguard Worker }
7297*4bdc9457SAndroid Build Coastguard Worker }
7298*4bdc9457SAndroid Build Coastguard Worker
7299*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 2x2s2_height_adjustment) {
7300*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7301*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7302*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7303*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(1)
7304*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7305*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7306*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
7307*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7308*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7309*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7310*4bdc9457SAndroid Build Coastguard Worker }
7311*4bdc9457SAndroid Build Coastguard Worker
7312*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 2x2s2_width_adjustment) {
7313*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7314*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7315*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7316*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(1)
7317*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7318*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7319*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
7320*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7321*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7322*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7323*4bdc9457SAndroid Build Coastguard Worker }
7324*4bdc9457SAndroid Build Coastguard Worker
7325*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 2x2s2_varying_input_height) {
7326*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7327*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
7328*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7329*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
7330*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7331*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7332*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
7333*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7334*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7335*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7336*4bdc9457SAndroid Build Coastguard Worker }
7337*4bdc9457SAndroid Build Coastguard Worker }
7338*4bdc9457SAndroid Build Coastguard Worker
7339*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 2x2s2_varying_input_width) {
7340*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7341*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
7342*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7343*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7344*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7345*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7346*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
7347*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7348*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7349*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7350*4bdc9457SAndroid Build Coastguard Worker }
7351*4bdc9457SAndroid Build Coastguard Worker }
7352*4bdc9457SAndroid Build Coastguard Worker
7353*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 2x2s2_varying_input_channels) {
7354*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7355*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
7356*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7357*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7358*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7359*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7360*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
7361*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7362*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7363*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7364*4bdc9457SAndroid Build Coastguard Worker }
7365*4bdc9457SAndroid Build Coastguard Worker }
7366*4bdc9457SAndroid Build Coastguard Worker
7367*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 2x2s2_varying_output_channels) {
7368*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7369*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qu8.gemm.nr * 2; output_channels *= 2) {
7370*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7371*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7372*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7373*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7374*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
7375*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
7376*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7377*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7378*4bdc9457SAndroid Build Coastguard Worker }
7379*4bdc9457SAndroid Build Coastguard Worker }
7380*4bdc9457SAndroid Build Coastguard Worker
7381*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 2x2s2_with_input_stride) {
7382*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7383*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7384*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7385*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7386*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7387*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
7388*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7389*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
7390*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7391*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7392*4bdc9457SAndroid Build Coastguard Worker }
7393*4bdc9457SAndroid Build Coastguard Worker
7394*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 2x2s2_with_output_stride) {
7395*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7396*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7397*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7398*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7399*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7400*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
7401*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7402*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qu8.gemm.nr * 2 + 13)
7403*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7404*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7405*4bdc9457SAndroid Build Coastguard Worker }
7406*4bdc9457SAndroid Build Coastguard Worker
7407*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 2x2s2_with_qmin) {
7408*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7409*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7410*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7411*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7412*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7413*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
7414*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7415*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
7416*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7417*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7418*4bdc9457SAndroid Build Coastguard Worker }
7419*4bdc9457SAndroid Build Coastguard Worker
7420*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 2x2s2_with_qmax) {
7421*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7422*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7423*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7424*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7425*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7426*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
7427*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7428*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
7429*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7430*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7431*4bdc9457SAndroid Build Coastguard Worker }
7432*4bdc9457SAndroid Build Coastguard Worker
7433*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 2x2s2_without_bias) {
7434*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7435*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7436*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
7437*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7438*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7439*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7440*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
7441*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7442*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7443*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7444*4bdc9457SAndroid Build Coastguard Worker }
7445*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,weights_cache_2x2s2)7446*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, weights_cache_2x2s2) {
7447*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7448*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7449*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7450*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7451*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7452*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
7453*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7454*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
7455*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7456*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7457*4bdc9457SAndroid Build Coastguard Worker }
7458*4bdc9457SAndroid Build Coastguard Worker
7459*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/GEMM path, grouped ****************************/
7460*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_2x2s2)7461*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_2x2s2) {
7462*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7463*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7464*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7465*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7466*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7467*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7468*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7469*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7470*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7471*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7472*4bdc9457SAndroid Build Coastguard Worker }
7473*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_Kx2sKx2)7474*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_Kx2sKx2) {
7475*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7476*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 3; kernel_height <= 5; kernel_height++) {
7477*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7478*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7479*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 2)
7480*4bdc9457SAndroid Build Coastguard Worker .stride(kernel_height, 2)
7481*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7482*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7483*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7484*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7485*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7486*4bdc9457SAndroid Build Coastguard Worker }
7487*4bdc9457SAndroid Build Coastguard Worker }
7488*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_2xKs2xK)7489*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_2xKs2xK) {
7490*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7491*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 3; kernel_width <= 5; kernel_width++) {
7492*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7493*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7494*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, kernel_width)
7495*4bdc9457SAndroid Build Coastguard Worker .stride(2, kernel_width)
7496*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7497*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7498*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7499*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7500*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7501*4bdc9457SAndroid Build Coastguard Worker }
7502*4bdc9457SAndroid Build Coastguard Worker }
7503*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_2x2s2_height_adjustment)7504*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_2x2s2_height_adjustment) {
7505*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7506*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7507*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7508*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(1)
7509*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7510*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7511*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7512*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7513*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7514*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7515*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7516*4bdc9457SAndroid Build Coastguard Worker }
7517*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_2x2s2_width_adjustment)7518*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_2x2s2_width_adjustment) {
7519*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7520*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7521*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7522*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(1)
7523*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7524*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7525*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7526*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7527*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7528*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7529*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7530*4bdc9457SAndroid Build Coastguard Worker }
7531*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_2x2s2_varying_input_height)7532*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_2x2s2_varying_input_height) {
7533*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7534*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
7535*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7536*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
7537*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7538*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7539*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7540*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7541*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7542*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7543*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7544*4bdc9457SAndroid Build Coastguard Worker }
7545*4bdc9457SAndroid Build Coastguard Worker }
7546*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_2x2s2_varying_input_width)7547*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_2x2s2_varying_input_width) {
7548*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7549*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
7550*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7551*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7552*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7553*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7554*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7555*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7556*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7557*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7558*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7559*4bdc9457SAndroid Build Coastguard Worker }
7560*4bdc9457SAndroid Build Coastguard Worker }
7561*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_2x2s2_varying_input_channels)7562*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_2x2s2_varying_input_channels) {
7563*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7564*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 14; input_channels <= 20; input_channels++) {
7565*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7566*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7567*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7568*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7569*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7570*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
7571*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7572*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7573*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7574*4bdc9457SAndroid Build Coastguard Worker }
7575*4bdc9457SAndroid Build Coastguard Worker }
7576*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_2x2s2_varying_output_channels)7577*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_2x2s2_varying_output_channels) {
7578*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7579*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qu8.gemm.nr * 2; output_channels *= 2) {
7580*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7581*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7582*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7583*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7584*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7585*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7586*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
7587*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7588*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7589*4bdc9457SAndroid Build Coastguard Worker }
7590*4bdc9457SAndroid Build Coastguard Worker }
7591*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_2x2s2_with_input_stride)7592*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_2x2s2_with_input_stride) {
7593*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7594*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7595*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7596*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7597*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7598*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7599*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7600*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7601*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(37)
7602*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7603*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7604*4bdc9457SAndroid Build Coastguard Worker }
7605*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_2x2s2_with_output_stride)7606*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_2x2s2_with_output_stride) {
7607*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7608*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7609*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7610*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7611*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7612*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7613*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7614*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr + 3)
7615*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qu8.gemm.nr * 2 + 13)
7616*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7617*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7618*4bdc9457SAndroid Build Coastguard Worker }
7619*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_2x2s2_with_qmin)7620*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_2x2s2_with_qmin) {
7621*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7622*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7623*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7624*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7625*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7626*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7627*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7628*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7629*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
7630*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7631*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7632*4bdc9457SAndroid Build Coastguard Worker }
7633*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_2x2s2_with_qmax)7634*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_2x2s2_with_qmax) {
7635*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7636*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7637*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7638*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7639*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7640*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7641*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7642*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7643*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
7644*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7645*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7646*4bdc9457SAndroid Build Coastguard Worker }
7647*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,grouped_2x2s2_without_bias)7648*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, grouped_2x2s2_without_bias) {
7649*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7650*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7651*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
7652*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7653*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7654*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7655*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7656*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7657*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7658*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7659*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7660*4bdc9457SAndroid Build Coastguard Worker }
7661*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,weights_cache_grouped_2x2s2)7662*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, weights_cache_grouped_2x2s2) {
7663*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7664*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7665*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7666*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7667*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7668*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7669*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7670*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7671*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
7672*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7673*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7674*4bdc9457SAndroid Build Coastguard Worker }
7675*4bdc9457SAndroid Build Coastguard Worker
7676*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/GEMM path, batched ****************************/
7677*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_2x2s2)7678*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_2x2s2) {
7679*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7680*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7681*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7682*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7683*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7684*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7685*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
7686*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7687*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7688*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7689*4bdc9457SAndroid Build Coastguard Worker }
7690*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_Kx2sKx2)7691*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_Kx2sKx2) {
7692*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7693*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 3; kernel_height <= 5; kernel_height++) {
7694*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7695*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7696*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7697*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 2)
7698*4bdc9457SAndroid Build Coastguard Worker .stride(kernel_height, 2)
7699*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7700*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7701*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7702*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7703*4bdc9457SAndroid Build Coastguard Worker }
7704*4bdc9457SAndroid Build Coastguard Worker }
7705*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_2xKs2xK)7706*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_2xKs2xK) {
7707*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7708*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 3; kernel_width <= 5; kernel_width++) {
7709*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7710*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7711*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7712*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, kernel_width)
7713*4bdc9457SAndroid Build Coastguard Worker .stride(2, kernel_width)
7714*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7715*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7716*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7717*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7718*4bdc9457SAndroid Build Coastguard Worker }
7719*4bdc9457SAndroid Build Coastguard Worker }
7720*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_2x2s2_height_adjustment)7721*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_2x2s2_height_adjustment) {
7722*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7723*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7724*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7725*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7726*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(1)
7727*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7728*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7729*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
7730*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7731*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7732*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7733*4bdc9457SAndroid Build Coastguard Worker }
7734*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_2x2s2_width_adjustment)7735*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_2x2s2_width_adjustment) {
7736*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7737*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7738*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7739*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7740*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(1)
7741*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7742*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7743*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
7744*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7745*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7746*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7747*4bdc9457SAndroid Build Coastguard Worker }
7748*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_2x2s2_varying_input_height)7749*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_2x2s2_varying_input_height) {
7750*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7751*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
7752*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7753*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7754*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
7755*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7756*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7757*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
7758*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7759*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7760*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7761*4bdc9457SAndroid Build Coastguard Worker }
7762*4bdc9457SAndroid Build Coastguard Worker }
7763*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_2x2s2_varying_input_width)7764*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_2x2s2_varying_input_width) {
7765*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7766*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
7767*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7768*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7769*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7770*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7771*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7772*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
7773*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7774*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7775*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7776*4bdc9457SAndroid Build Coastguard Worker }
7777*4bdc9457SAndroid Build Coastguard Worker }
7778*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_2x2s2_varying_input_channels)7779*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_2x2s2_varying_input_channels) {
7780*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7781*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
7782*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7783*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7784*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7785*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7786*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7787*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
7788*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7789*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7790*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7791*4bdc9457SAndroid Build Coastguard Worker }
7792*4bdc9457SAndroid Build Coastguard Worker }
7793*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_2x2s2_varying_output_channels)7794*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_2x2s2_varying_output_channels) {
7795*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7796*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qu8.gemm.nr * 2; output_channels *= 2) {
7797*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7798*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7799*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7800*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7801*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7802*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
7803*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
7804*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7805*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7806*4bdc9457SAndroid Build Coastguard Worker }
7807*4bdc9457SAndroid Build Coastguard Worker }
7808*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_2x2s2_with_input_stride)7809*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_2x2s2_with_input_stride) {
7810*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7811*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7812*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7813*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7814*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7815*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7816*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
7817*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7818*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
7819*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7820*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7821*4bdc9457SAndroid Build Coastguard Worker }
7822*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_2x2s2_with_output_stride)7823*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_2x2s2_with_output_stride) {
7824*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7825*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7826*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7827*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7828*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7829*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7830*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
7831*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7832*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qu8.gemm.nr * 2 + 13)
7833*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7834*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7835*4bdc9457SAndroid Build Coastguard Worker }
7836*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_2x2s2_with_qmin)7837*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_2x2s2_with_qmin) {
7838*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7839*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7840*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7841*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7842*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7843*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7844*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
7845*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7846*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
7847*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7848*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7849*4bdc9457SAndroid Build Coastguard Worker }
7850*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_2x2s2_with_qmax)7851*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_2x2s2_with_qmax) {
7852*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7853*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7854*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7855*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7856*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7857*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7858*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
7859*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7860*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
7861*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7862*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7863*4bdc9457SAndroid Build Coastguard Worker }
7864*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_2x2s2_without_bias)7865*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_2x2s2_without_bias) {
7866*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7867*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7868*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
7869*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7870*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7871*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7872*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7873*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
7874*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7875*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7876*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7877*4bdc9457SAndroid Build Coastguard Worker }
7878*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,weights_cache_batched_2x2s2)7879*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, weights_cache_batched_2x2s2) {
7880*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7881*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7882*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7883*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7884*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7885*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7886*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
7887*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7888*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
7889*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7890*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7891*4bdc9457SAndroid Build Coastguard Worker }
7892*4bdc9457SAndroid Build Coastguard Worker
7893*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/GEMM path, grouped, batched ****************************/
7894*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_2x2s2)7895*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_2x2s2) {
7896*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7897*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7898*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7899*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7900*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7901*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7902*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7903*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7904*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7905*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7906*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7907*4bdc9457SAndroid Build Coastguard Worker }
7908*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_Kx2sKx2)7909*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_Kx2sKx2) {
7910*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7911*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 3; kernel_height <= 5; kernel_height++) {
7912*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7913*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7914*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7915*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 2)
7916*4bdc9457SAndroid Build Coastguard Worker .stride(kernel_height, 2)
7917*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7918*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7919*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7920*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7921*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7922*4bdc9457SAndroid Build Coastguard Worker }
7923*4bdc9457SAndroid Build Coastguard Worker }
7924*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_2xKs2xK)7925*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_2xKs2xK) {
7926*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7927*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 3; kernel_width <= 5; kernel_width++) {
7928*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7929*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7930*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7931*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, kernel_width)
7932*4bdc9457SAndroid Build Coastguard Worker .stride(2, kernel_width)
7933*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7934*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7935*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7936*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
7937*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7938*4bdc9457SAndroid Build Coastguard Worker }
7939*4bdc9457SAndroid Build Coastguard Worker }
7940*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_2x2s2_height_adjustment)7941*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_2x2s2_height_adjustment) {
7942*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7943*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7944*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7945*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7946*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(1)
7947*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7948*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7949*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7950*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7951*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7952*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7953*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7954*4bdc9457SAndroid Build Coastguard Worker }
7955*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_2x2s2_width_adjustment)7956*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_2x2s2_width_adjustment) {
7957*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7958*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7959*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7960*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7961*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(1)
7962*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7963*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7964*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7965*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7966*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7967*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7968*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7969*4bdc9457SAndroid Build Coastguard Worker }
7970*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_2x2s2_varying_input_height)7971*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_2x2s2_varying_input_height) {
7972*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7973*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
7974*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7975*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7976*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
7977*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7978*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7979*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7980*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7981*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7982*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7983*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
7984*4bdc9457SAndroid Build Coastguard Worker }
7985*4bdc9457SAndroid Build Coastguard Worker }
7986*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_2x2s2_varying_input_width)7987*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_2x2s2_varying_input_width) {
7988*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
7989*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
7990*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
7991*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
7992*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
7993*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
7994*4bdc9457SAndroid Build Coastguard Worker .stride(2)
7995*4bdc9457SAndroid Build Coastguard Worker .groups(2)
7996*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
7997*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
7998*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
7999*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
8000*4bdc9457SAndroid Build Coastguard Worker }
8001*4bdc9457SAndroid Build Coastguard Worker }
8002*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_2x2s2_varying_input_channels)8003*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_2x2s2_varying_input_channels) {
8004*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8005*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 14; input_channels <= 20; input_channels++) {
8006*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8007*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8008*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
8009*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
8010*4bdc9457SAndroid Build Coastguard Worker .stride(2)
8011*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8012*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
8013*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
8014*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
8015*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
8016*4bdc9457SAndroid Build Coastguard Worker }
8017*4bdc9457SAndroid Build Coastguard Worker }
8018*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_2x2s2_varying_output_channels)8019*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_2x2s2_varying_output_channels) {
8020*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8021*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.qu8.gemm.nr * 2; output_channels *= 2) {
8022*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8023*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8024*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
8025*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
8026*4bdc9457SAndroid Build Coastguard Worker .stride(2)
8027*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8028*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
8029*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
8030*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
8031*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
8032*4bdc9457SAndroid Build Coastguard Worker }
8033*4bdc9457SAndroid Build Coastguard Worker }
8034*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_2x2s2_with_input_stride)8035*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_2x2s2_with_input_stride) {
8036*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8037*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8038*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8039*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
8040*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
8041*4bdc9457SAndroid Build Coastguard Worker .stride(2)
8042*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8043*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
8044*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
8045*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(37)
8046*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8047*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
8048*4bdc9457SAndroid Build Coastguard Worker }
8049*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_2x2s2_with_output_stride)8050*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_2x2s2_with_output_stride) {
8051*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8052*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8053*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8054*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
8055*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
8056*4bdc9457SAndroid Build Coastguard Worker .stride(2)
8057*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8058*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
8059*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr + 3)
8060*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.qu8.gemm.nr * 2 + 13)
8061*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8062*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
8063*4bdc9457SAndroid Build Coastguard Worker }
8064*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_2x2s2_with_qmin)8065*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_2x2s2_with_qmin) {
8066*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8067*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8068*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8069*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
8070*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
8071*4bdc9457SAndroid Build Coastguard Worker .stride(2)
8072*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8073*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
8074*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
8075*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
8076*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8077*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
8078*4bdc9457SAndroid Build Coastguard Worker }
8079*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_2x2s2_with_qmax)8080*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_2x2s2_with_qmax) {
8081*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8082*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8083*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8084*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
8085*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
8086*4bdc9457SAndroid Build Coastguard Worker .stride(2)
8087*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8088*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
8089*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
8090*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
8091*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8092*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
8093*4bdc9457SAndroid Build Coastguard Worker }
8094*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,batched_grouped_2x2s2_without_bias)8095*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, batched_grouped_2x2s2_without_bias) {
8096*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8097*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8098*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
8099*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8100*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
8101*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
8102*4bdc9457SAndroid Build Coastguard Worker .stride(2)
8103*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8104*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
8105*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
8106*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8107*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
8108*4bdc9457SAndroid Build Coastguard Worker }
8109*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_QU8,weights_cache_batched_grouped_2x2s2)8110*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, weights_cache_batched_grouped_2x2s2) {
8111*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8112*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8113*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8114*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
8115*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
8116*4bdc9457SAndroid Build Coastguard Worker .stride(2)
8117*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8118*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
8119*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.qu8.gemm.nr * 2 + 3)
8120*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
8121*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8122*4bdc9457SAndroid Build Coastguard Worker .TestQU8();
8123*4bdc9457SAndroid Build Coastguard Worker }
8124*4bdc9457SAndroid Build Coastguard Worker
8125*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/GEMM path, setup ****************************/
8126*4bdc9457SAndroid Build Coastguard Worker
8127*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 2x2s2_setup_changing_batch) {
8128*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8129*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8130*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8131*4bdc9457SAndroid Build Coastguard Worker .next_batch_size(5)
8132*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
8133*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
8134*4bdc9457SAndroid Build Coastguard Worker .stride(2)
8135*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8136*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
8137*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
8138*4bdc9457SAndroid Build Coastguard Worker .TestSetupQU8();
8139*4bdc9457SAndroid Build Coastguard Worker }
8140*4bdc9457SAndroid Build Coastguard Worker
8141*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 2x2s2_setup_changing_height) {
8142*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8143*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8144*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8145*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
8146*4bdc9457SAndroid Build Coastguard Worker .next_input_height(kStridedInputHeight + 3)
8147*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
8148*4bdc9457SAndroid Build Coastguard Worker .stride(2)
8149*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8150*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
8151*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
8152*4bdc9457SAndroid Build Coastguard Worker .TestSetupQU8();
8153*4bdc9457SAndroid Build Coastguard Worker }
8154*4bdc9457SAndroid Build Coastguard Worker
8155*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_QU8, 2x2s2_setup_changing_width) {
8156*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8157*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8158*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8159*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
8160*4bdc9457SAndroid Build Coastguard Worker .next_input_width(kStridedInputWidth + 3)
8161*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
8162*4bdc9457SAndroid Build Coastguard Worker .stride(2)
8163*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8164*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
8165*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
8166*4bdc9457SAndroid Build Coastguard Worker .TestSetupQU8();
8167*4bdc9457SAndroid Build Coastguard Worker }
8168*4bdc9457SAndroid Build Coastguard Worker
8169*4bdc9457SAndroid Build Coastguard Worker /**************************** Future GEMM path ****************************/
8170*4bdc9457SAndroid Build Coastguard Worker
8171*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 1x1) {
8172*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8173*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8174*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8175*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8176*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8177*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8178*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8179*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8180*4bdc9457SAndroid Build Coastguard Worker }
8181*4bdc9457SAndroid Build Coastguard Worker
8182*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 1x1_with_fp32_weights) {
8183*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8184*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8185*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8186*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8187*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8188*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8189*4bdc9457SAndroid Build Coastguard Worker .weights_type(DeconvolutionOperatorTester::WeightsType::FP32)
8190*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8191*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8192*4bdc9457SAndroid Build Coastguard Worker }
8193*4bdc9457SAndroid Build Coastguard Worker
8194*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 1x1_varying_input_width) {
8195*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8196*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
8197*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8198*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
8199*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8200*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8201*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8202*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
8203*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8204*4bdc9457SAndroid Build Coastguard Worker }
8205*4bdc9457SAndroid Build Coastguard Worker }
8206*4bdc9457SAndroid Build Coastguard Worker
8207*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 1x1_varying_input_height) {
8208*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8209*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
8210*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8211*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
8212*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8213*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8214*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8215*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
8216*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8217*4bdc9457SAndroid Build Coastguard Worker }
8218*4bdc9457SAndroid Build Coastguard Worker }
8219*4bdc9457SAndroid Build Coastguard Worker
8220*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 1x1_varying_input_channels) {
8221*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8222*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
8223*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8224*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8225*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8226*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
8227*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8228*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
8229*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8230*4bdc9457SAndroid Build Coastguard Worker }
8231*4bdc9457SAndroid Build Coastguard Worker }
8232*4bdc9457SAndroid Build Coastguard Worker
8233*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 1x1_varying_output_channels) {
8234*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8235*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f16.gemm.nr * 2; output_channels *= 2) {
8236*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8237*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8238*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8239*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8240*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
8241*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
8242*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8243*4bdc9457SAndroid Build Coastguard Worker }
8244*4bdc9457SAndroid Build Coastguard Worker }
8245*4bdc9457SAndroid Build Coastguard Worker
8246*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 1x1_with_input_stride) {
8247*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8248*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8249*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8250*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8251*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8252*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8253*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
8254*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8255*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8256*4bdc9457SAndroid Build Coastguard Worker }
8257*4bdc9457SAndroid Build Coastguard Worker
8258*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 1x1_with_output_stride) {
8259*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8260*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8261*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8262*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8263*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8264*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8265*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f16.gemm.nr * 2 + 13)
8266*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8267*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8268*4bdc9457SAndroid Build Coastguard Worker }
8269*4bdc9457SAndroid Build Coastguard Worker
8270*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 1x1_with_qmin) {
8271*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8272*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8273*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8274*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8275*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8276*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8277*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
8278*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8279*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8280*4bdc9457SAndroid Build Coastguard Worker }
8281*4bdc9457SAndroid Build Coastguard Worker
8282*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 1x1_with_qmax) {
8283*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8284*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8285*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8286*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8287*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8288*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8289*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
8290*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8291*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8292*4bdc9457SAndroid Build Coastguard Worker }
8293*4bdc9457SAndroid Build Coastguard Worker
8294*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 1x1_without_bias) {
8295*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8296*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8297*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
8298*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8299*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8300*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8301*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8302*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8303*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8304*4bdc9457SAndroid Build Coastguard Worker }
8305*4bdc9457SAndroid Build Coastguard Worker
8306*4bdc9457SAndroid Build Coastguard Worker /**************************** Future GEMM path, grouped ****************************/
8307*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_1x1)8308*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_1x1) {
8309*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8310*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8311*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8312*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8313*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8314*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8315*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8316*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8317*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8318*4bdc9457SAndroid Build Coastguard Worker }
8319*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_1x1_with_fp32_weights)8320*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_1x1_with_fp32_weights) {
8321*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8322*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8323*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8324*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8325*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8326*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8327*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8328*4bdc9457SAndroid Build Coastguard Worker .weights_type(DeconvolutionOperatorTester::WeightsType::FP32)
8329*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8330*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8331*4bdc9457SAndroid Build Coastguard Worker }
8332*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_1x1_varying_input_width)8333*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_1x1_varying_input_width) {
8334*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8335*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
8336*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8337*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
8338*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8339*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8340*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8341*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8342*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
8343*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8344*4bdc9457SAndroid Build Coastguard Worker }
8345*4bdc9457SAndroid Build Coastguard Worker }
8346*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_1x1_varying_input_height)8347*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_1x1_varying_input_height) {
8348*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8349*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
8350*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8351*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
8352*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8353*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8354*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8355*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8356*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
8357*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8358*4bdc9457SAndroid Build Coastguard Worker }
8359*4bdc9457SAndroid Build Coastguard Worker }
8360*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_1x1_varying_input_channels)8361*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_1x1_varying_input_channels) {
8362*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8363*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
8364*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8365*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8366*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8367*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8368*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
8369*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8370*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
8371*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8372*4bdc9457SAndroid Build Coastguard Worker }
8373*4bdc9457SAndroid Build Coastguard Worker }
8374*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_1x1_varying_output_channels)8375*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_1x1_varying_output_channels) {
8376*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8377*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f16.gemm.nr * 2; output_channels *= 2) {
8378*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8379*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8380*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8381*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8382*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8383*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
8384*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
8385*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8386*4bdc9457SAndroid Build Coastguard Worker }
8387*4bdc9457SAndroid Build Coastguard Worker }
8388*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_1x1_with_input_stride)8389*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_1x1_with_input_stride) {
8390*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8391*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8392*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8393*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8394*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8395*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8396*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8397*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(47)
8398*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8399*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8400*4bdc9457SAndroid Build Coastguard Worker }
8401*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_1x1_with_output_stride)8402*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_1x1_with_output_stride) {
8403*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8404*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8405*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8406*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8407*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8408*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8409*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr + 3)
8410*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f16.gemm.nr * 2 + 13)
8411*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8412*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8413*4bdc9457SAndroid Build Coastguard Worker }
8414*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_1x1_with_qmin)8415*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_1x1_with_qmin) {
8416*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8417*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8418*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8419*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8420*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8421*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8422*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8423*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
8424*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8425*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8426*4bdc9457SAndroid Build Coastguard Worker }
8427*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_1x1_with_qmax)8428*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_1x1_with_qmax) {
8429*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8430*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8431*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8432*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8433*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8434*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8435*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8436*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
8437*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8438*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8439*4bdc9457SAndroid Build Coastguard Worker }
8440*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_1x1_without_bias)8441*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_1x1_without_bias) {
8442*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8443*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8444*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
8445*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8446*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8447*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8448*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8449*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8450*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8451*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8452*4bdc9457SAndroid Build Coastguard Worker }
8453*4bdc9457SAndroid Build Coastguard Worker
8454*4bdc9457SAndroid Build Coastguard Worker /**************************** Future GEMM path, batched ****************************/
8455*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_1x1)8456*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_1x1) {
8457*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8458*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8459*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8460*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8461*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8462*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8463*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8464*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8465*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8466*4bdc9457SAndroid Build Coastguard Worker }
8467*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_1x1_with_fp32_weights)8468*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_1x1_with_fp32_weights) {
8469*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8470*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8471*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8472*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8473*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8474*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8475*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8476*4bdc9457SAndroid Build Coastguard Worker .weights_type(DeconvolutionOperatorTester::WeightsType::FP32)
8477*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8478*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8479*4bdc9457SAndroid Build Coastguard Worker }
8480*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_1x1_varying_input_width)8481*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_1x1_varying_input_width) {
8482*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8483*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
8484*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8485*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8486*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
8487*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8488*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8489*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8490*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
8491*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8492*4bdc9457SAndroid Build Coastguard Worker }
8493*4bdc9457SAndroid Build Coastguard Worker }
8494*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_1x1_varying_input_height)8495*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_1x1_varying_input_height) {
8496*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8497*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
8498*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8499*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8500*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
8501*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8502*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8503*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8504*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
8505*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8506*4bdc9457SAndroid Build Coastguard Worker }
8507*4bdc9457SAndroid Build Coastguard Worker }
8508*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_1x1_varying_input_channels)8509*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_1x1_varying_input_channels) {
8510*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8511*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
8512*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8513*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8514*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8515*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8516*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
8517*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8518*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
8519*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8520*4bdc9457SAndroid Build Coastguard Worker }
8521*4bdc9457SAndroid Build Coastguard Worker }
8522*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_1x1_varying_output_channels)8523*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_1x1_varying_output_channels) {
8524*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8525*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f16.gemm.nr * 2; output_channels *= 2) {
8526*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8527*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8528*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8529*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8530*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8531*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
8532*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
8533*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8534*4bdc9457SAndroid Build Coastguard Worker }
8535*4bdc9457SAndroid Build Coastguard Worker }
8536*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_1x1_with_input_stride)8537*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_1x1_with_input_stride) {
8538*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8539*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8540*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8541*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8542*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8543*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8544*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8545*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
8546*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8547*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8548*4bdc9457SAndroid Build Coastguard Worker }
8549*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_1x1_with_output_stride)8550*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_1x1_with_output_stride) {
8551*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8552*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8553*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8554*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8555*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8556*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8557*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8558*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f16.gemm.nr * 2 + 13)
8559*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8560*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8561*4bdc9457SAndroid Build Coastguard Worker }
8562*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_1x1_with_qmin)8563*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_1x1_with_qmin) {
8564*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8565*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8566*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8567*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8568*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8569*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8570*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8571*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
8572*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8573*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8574*4bdc9457SAndroid Build Coastguard Worker }
8575*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_1x1_with_qmax)8576*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_1x1_with_qmax) {
8577*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8578*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8579*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8580*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8581*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8582*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8583*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8584*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
8585*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8586*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8587*4bdc9457SAndroid Build Coastguard Worker }
8588*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_1x1_without_bias)8589*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_1x1_without_bias) {
8590*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8591*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8592*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
8593*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8594*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8595*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8596*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8597*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8598*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8599*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8600*4bdc9457SAndroid Build Coastguard Worker }
8601*4bdc9457SAndroid Build Coastguard Worker
8602*4bdc9457SAndroid Build Coastguard Worker /**************************** Future GEMM path, batched, grouped ****************************/
8603*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_1x1)8604*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_1x1) {
8605*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8606*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8607*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8608*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8609*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8610*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8611*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8612*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8613*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8614*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8615*4bdc9457SAndroid Build Coastguard Worker }
8616*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_1x1_with_fp32_weights)8617*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_1x1_with_fp32_weights) {
8618*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8619*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8620*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8621*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8622*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8623*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8624*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8625*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8626*4bdc9457SAndroid Build Coastguard Worker .weights_type(DeconvolutionOperatorTester::WeightsType::FP32)
8627*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8628*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8629*4bdc9457SAndroid Build Coastguard Worker }
8630*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_1x1_varying_input_width)8631*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_1x1_varying_input_width) {
8632*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8633*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
8634*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8635*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8636*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
8637*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8638*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8639*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8640*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8641*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
8642*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8643*4bdc9457SAndroid Build Coastguard Worker }
8644*4bdc9457SAndroid Build Coastguard Worker }
8645*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_1x1_varying_input_height)8646*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_1x1_varying_input_height) {
8647*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8648*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
8649*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8650*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8651*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
8652*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8653*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8654*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8655*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8656*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
8657*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8658*4bdc9457SAndroid Build Coastguard Worker }
8659*4bdc9457SAndroid Build Coastguard Worker }
8660*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_1x1_varying_input_channels)8661*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_1x1_varying_input_channels) {
8662*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8663*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
8664*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8665*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8666*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8667*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8668*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8669*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
8670*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8671*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
8672*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8673*4bdc9457SAndroid Build Coastguard Worker }
8674*4bdc9457SAndroid Build Coastguard Worker }
8675*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_1x1_varying_output_channels)8676*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_1x1_varying_output_channels) {
8677*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8678*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f16.gemm.nr * 2; output_channels *= 2) {
8679*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8680*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8681*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8682*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8683*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8684*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8685*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
8686*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
8687*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8688*4bdc9457SAndroid Build Coastguard Worker }
8689*4bdc9457SAndroid Build Coastguard Worker }
8690*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_1x1_with_input_stride)8691*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_1x1_with_input_stride) {
8692*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8693*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8694*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8695*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8696*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8697*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8698*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8699*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8700*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(47)
8701*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8702*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8703*4bdc9457SAndroid Build Coastguard Worker }
8704*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_1x1_with_output_stride)8705*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_1x1_with_output_stride) {
8706*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8707*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8708*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8709*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8710*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8711*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8712*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8713*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr + 3)
8714*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f16.gemm.nr * 2 + 13)
8715*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8716*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8717*4bdc9457SAndroid Build Coastguard Worker }
8718*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_1x1_with_qmin)8719*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_1x1_with_qmin) {
8720*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8721*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8722*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8723*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8724*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8725*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8726*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8727*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8728*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
8729*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8730*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8731*4bdc9457SAndroid Build Coastguard Worker }
8732*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_1x1_with_qmax)8733*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_1x1_with_qmax) {
8734*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8735*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8736*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8737*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8738*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8739*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8740*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8741*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8742*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
8743*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8744*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8745*4bdc9457SAndroid Build Coastguard Worker }
8746*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_1x1_without_bias)8747*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_1x1_without_bias) {
8748*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8749*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8750*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
8751*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
8752*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8753*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
8754*4bdc9457SAndroid Build Coastguard Worker .groups(2)
8755*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8756*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8757*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8758*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8759*4bdc9457SAndroid Build Coastguard Worker }
8760*4bdc9457SAndroid Build Coastguard Worker
8761*4bdc9457SAndroid Build Coastguard Worker /**************************** CONV path ****************************/
8762*4bdc9457SAndroid Build Coastguard Worker
8763*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3) {
8764*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8765*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8766*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8767*4bdc9457SAndroid Build Coastguard Worker .padding(1)
8768*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
8769*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
8770*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8771*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8772*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8773*4bdc9457SAndroid Build Coastguard Worker }
8774*4bdc9457SAndroid Build Coastguard Worker
8775*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3_with_fp32_weights) {
8776*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8777*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8778*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8779*4bdc9457SAndroid Build Coastguard Worker .padding(1)
8780*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
8781*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
8782*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8783*4bdc9457SAndroid Build Coastguard Worker .weights_type(DeconvolutionOperatorTester::WeightsType::FP32)
8784*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8785*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8786*4bdc9457SAndroid Build Coastguard Worker }
8787*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,Kx3)8788*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, Kx3) {
8789*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8790*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 1; kernel_height <= 4; kernel_height *= 2) {
8791*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8792*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8793*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
8794*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
8795*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
8796*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8797*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8798*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8799*4bdc9457SAndroid Build Coastguard Worker }
8800*4bdc9457SAndroid Build Coastguard Worker }
8801*4bdc9457SAndroid Build Coastguard Worker
8802*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3xK) {
8803*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8804*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 1; kernel_width <= 4; kernel_width *= 2) {
8805*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8806*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8807*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
8808*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
8809*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
8810*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8811*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8812*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8813*4bdc9457SAndroid Build Coastguard Worker }
8814*4bdc9457SAndroid Build Coastguard Worker }
8815*4bdc9457SAndroid Build Coastguard Worker
8816*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3_varying_height_padding) {
8817*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8818*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
8819*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
8820*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8821*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8822*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
8823*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
8824*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
8825*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
8826*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
8827*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8828*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
8829*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8830*4bdc9457SAndroid Build Coastguard Worker }
8831*4bdc9457SAndroid Build Coastguard Worker }
8832*4bdc9457SAndroid Build Coastguard Worker }
8833*4bdc9457SAndroid Build Coastguard Worker
8834*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3_varying_width_padding) {
8835*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8836*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
8837*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
8838*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8839*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8840*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
8841*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
8842*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
8843*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
8844*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
8845*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8846*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
8847*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8848*4bdc9457SAndroid Build Coastguard Worker }
8849*4bdc9457SAndroid Build Coastguard Worker }
8850*4bdc9457SAndroid Build Coastguard Worker }
8851*4bdc9457SAndroid Build Coastguard Worker
8852*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3_varying_height_adjustment) {
8853*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8854*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 1; adjustment_height <= 2; adjustment_height++) {
8855*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8856*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8857*4bdc9457SAndroid Build Coastguard Worker .padding(1)
8858*4bdc9457SAndroid Build Coastguard Worker .stride_height(adjustment_height + 1)
8859*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
8860*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
8861*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
8862*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8863*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
8864*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8865*4bdc9457SAndroid Build Coastguard Worker }
8866*4bdc9457SAndroid Build Coastguard Worker }
8867*4bdc9457SAndroid Build Coastguard Worker
8868*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3_varying_width_adjustment) {
8869*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8870*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 1; adjustment_width <= 2; adjustment_width++) {
8871*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8872*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8873*4bdc9457SAndroid Build Coastguard Worker .padding(1)
8874*4bdc9457SAndroid Build Coastguard Worker .stride_width(adjustment_width + 1)
8875*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
8876*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
8877*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
8878*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8879*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
8880*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8881*4bdc9457SAndroid Build Coastguard Worker }
8882*4bdc9457SAndroid Build Coastguard Worker }
8883*4bdc9457SAndroid Build Coastguard Worker
8884*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3_varying_input_height) {
8885*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8886*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
8887*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8888*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
8889*4bdc9457SAndroid Build Coastguard Worker .padding(1)
8890*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
8891*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
8892*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8893*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
8894*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8895*4bdc9457SAndroid Build Coastguard Worker }
8896*4bdc9457SAndroid Build Coastguard Worker }
8897*4bdc9457SAndroid Build Coastguard Worker
8898*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3_varying_input_width) {
8899*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8900*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
8901*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8902*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
8903*4bdc9457SAndroid Build Coastguard Worker .padding(1)
8904*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
8905*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
8906*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8907*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
8908*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8909*4bdc9457SAndroid Build Coastguard Worker }
8910*4bdc9457SAndroid Build Coastguard Worker }
8911*4bdc9457SAndroid Build Coastguard Worker
8912*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3_varying_input_channels) {
8913*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8914*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
8915*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8916*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8917*4bdc9457SAndroid Build Coastguard Worker .padding(1)
8918*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
8919*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
8920*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8921*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
8922*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8923*4bdc9457SAndroid Build Coastguard Worker }
8924*4bdc9457SAndroid Build Coastguard Worker }
8925*4bdc9457SAndroid Build Coastguard Worker
8926*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3_varying_output_channels) {
8927*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8928*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f16.gemm.nr * 2; output_channels *= 2) {
8929*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8930*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8931*4bdc9457SAndroid Build Coastguard Worker .padding(1)
8932*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
8933*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8934*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
8935*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
8936*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8937*4bdc9457SAndroid Build Coastguard Worker }
8938*4bdc9457SAndroid Build Coastguard Worker }
8939*4bdc9457SAndroid Build Coastguard Worker
8940*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3_with_height_dilation) {
8941*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8942*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_height = 2; dilation_height <= 3; dilation_height++) {
8943*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8944*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8945*4bdc9457SAndroid Build Coastguard Worker .padding(1)
8946*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
8947*4bdc9457SAndroid Build Coastguard Worker .dilation_height(dilation_height)
8948*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8949*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8950*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8951*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8952*4bdc9457SAndroid Build Coastguard Worker }
8953*4bdc9457SAndroid Build Coastguard Worker }
8954*4bdc9457SAndroid Build Coastguard Worker
8955*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3_with_width_dilation) {
8956*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8957*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) {
8958*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8959*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8960*4bdc9457SAndroid Build Coastguard Worker .padding(1)
8961*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
8962*4bdc9457SAndroid Build Coastguard Worker .dilation_width(dilation_width)
8963*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8964*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8965*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8966*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8967*4bdc9457SAndroid Build Coastguard Worker }
8968*4bdc9457SAndroid Build Coastguard Worker }
8969*4bdc9457SAndroid Build Coastguard Worker
8970*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3_with_height_dilation_and_stride) {
8971*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8972*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8973*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8974*4bdc9457SAndroid Build Coastguard Worker .padding(1)
8975*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
8976*4bdc9457SAndroid Build Coastguard Worker .dilation_height(3)
8977*4bdc9457SAndroid Build Coastguard Worker .stride_height(2)
8978*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8979*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8980*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8981*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8982*4bdc9457SAndroid Build Coastguard Worker }
8983*4bdc9457SAndroid Build Coastguard Worker
8984*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3_with_width_dilation_and_stride) {
8985*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
8986*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
8987*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
8988*4bdc9457SAndroid Build Coastguard Worker .padding(1)
8989*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
8990*4bdc9457SAndroid Build Coastguard Worker .dilation_width(3)
8991*4bdc9457SAndroid Build Coastguard Worker .stride_width(2)
8992*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
8993*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
8994*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
8995*4bdc9457SAndroid Build Coastguard Worker .TestF16();
8996*4bdc9457SAndroid Build Coastguard Worker }
8997*4bdc9457SAndroid Build Coastguard Worker
8998*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3_with_input_stride) {
8999*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9000*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9001*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9002*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9003*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9004*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9005*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9006*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
9007*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9008*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9009*4bdc9457SAndroid Build Coastguard Worker }
9010*4bdc9457SAndroid Build Coastguard Worker
9011*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3_with_output_stride) {
9012*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9013*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9014*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9015*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9016*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9017*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9018*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9019*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f16.gemm.nr * 2 + 13)
9020*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9021*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9022*4bdc9457SAndroid Build Coastguard Worker }
9023*4bdc9457SAndroid Build Coastguard Worker
9024*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3_with_qmin) {
9025*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9026*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9027*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9028*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9029*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9030*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9031*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9032*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
9033*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9034*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9035*4bdc9457SAndroid Build Coastguard Worker }
9036*4bdc9457SAndroid Build Coastguard Worker
9037*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3_with_qmax) {
9038*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9039*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9040*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9041*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9042*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9043*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9044*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9045*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
9046*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9047*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9048*4bdc9457SAndroid Build Coastguard Worker }
9049*4bdc9457SAndroid Build Coastguard Worker
9050*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3_without_bias) {
9051*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9052*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9053*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
9054*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9055*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9056*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9057*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9058*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9059*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9060*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9061*4bdc9457SAndroid Build Coastguard Worker }
9062*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,weights_cache_3x3)9063*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, weights_cache_3x3) {
9064*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9065*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9066*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9067*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9068*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9069*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9070*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9071*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
9072*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9073*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9074*4bdc9457SAndroid Build Coastguard Worker }
9075*4bdc9457SAndroid Build Coastguard Worker
9076*4bdc9457SAndroid Build Coastguard Worker /**************************** CONV path, grouped ****************************/
9077*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3)9078*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3) {
9079*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9080*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9081*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9082*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9083*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9084*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9085*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9086*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9087*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9088*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9089*4bdc9457SAndroid Build Coastguard Worker }
9090*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3_with_fp32_weights)9091*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3_with_fp32_weights) {
9092*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9093*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9094*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9095*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9096*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9097*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9098*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9099*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9100*4bdc9457SAndroid Build Coastguard Worker .weights_type(DeconvolutionOperatorTester::WeightsType::FP32)
9101*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9102*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9103*4bdc9457SAndroid Build Coastguard Worker }
9104*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_Kx3)9105*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_Kx3) {
9106*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9107*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 1; kernel_height <= 4; kernel_height *= 2) {
9108*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9109*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9110*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
9111*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
9112*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9113*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
9114*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9115*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9116*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9117*4bdc9457SAndroid Build Coastguard Worker }
9118*4bdc9457SAndroid Build Coastguard Worker }
9119*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3xK)9120*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3xK) {
9121*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9122*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 1; kernel_width <= 4; kernel_width *= 2) {
9123*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9124*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9125*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
9126*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
9127*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9128*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
9129*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9130*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9131*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9132*4bdc9457SAndroid Build Coastguard Worker }
9133*4bdc9457SAndroid Build Coastguard Worker }
9134*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3_varying_height_padding)9135*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3_varying_height_padding) {
9136*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9137*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
9138*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
9139*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9140*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9141*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
9142*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
9143*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
9144*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9145*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9146*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9147*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9148*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
9149*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9150*4bdc9457SAndroid Build Coastguard Worker }
9151*4bdc9457SAndroid Build Coastguard Worker }
9152*4bdc9457SAndroid Build Coastguard Worker }
9153*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3_varying_width_padding)9154*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3_varying_width_padding) {
9155*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9156*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
9157*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
9158*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9159*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9160*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
9161*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
9162*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
9163*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9164*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9165*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9166*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9167*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
9168*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9169*4bdc9457SAndroid Build Coastguard Worker }
9170*4bdc9457SAndroid Build Coastguard Worker }
9171*4bdc9457SAndroid Build Coastguard Worker }
9172*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3_varying_height_adjustment)9173*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3_varying_height_adjustment) {
9174*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9175*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 1; adjustment_height <= 2; adjustment_height++) {
9176*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9177*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9178*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9179*4bdc9457SAndroid Build Coastguard Worker .stride_height(adjustment_height + 1)
9180*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
9181*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9182*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9183*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9184*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9185*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
9186*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9187*4bdc9457SAndroid Build Coastguard Worker }
9188*4bdc9457SAndroid Build Coastguard Worker }
9189*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3_varying_width_adjustment)9190*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3_varying_width_adjustment) {
9191*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9192*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 1; adjustment_width <= 2; adjustment_width++) {
9193*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9194*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9195*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9196*4bdc9457SAndroid Build Coastguard Worker .stride_width(adjustment_width + 1)
9197*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
9198*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9199*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9200*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9201*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9202*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
9203*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9204*4bdc9457SAndroid Build Coastguard Worker }
9205*4bdc9457SAndroid Build Coastguard Worker }
9206*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3_varying_input_height)9207*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3_varying_input_height) {
9208*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9209*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
9210*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9211*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
9212*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9213*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9214*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9215*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9216*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9217*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
9218*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9219*4bdc9457SAndroid Build Coastguard Worker }
9220*4bdc9457SAndroid Build Coastguard Worker }
9221*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3_varying_input_width)9222*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3_varying_input_width) {
9223*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9224*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
9225*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9226*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
9227*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9228*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9229*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9230*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9231*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9232*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
9233*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9234*4bdc9457SAndroid Build Coastguard Worker }
9235*4bdc9457SAndroid Build Coastguard Worker }
9236*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3_varying_input_channels)9237*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3_varying_input_channels) {
9238*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9239*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
9240*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9241*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9242*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9243*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9244*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9245*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
9246*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9247*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
9248*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9249*4bdc9457SAndroid Build Coastguard Worker }
9250*4bdc9457SAndroid Build Coastguard Worker }
9251*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3_varying_output_channels)9252*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3_varying_output_channels) {
9253*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9254*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f16.gemm.nr * 2; output_channels *= 2) {
9255*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9256*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9257*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9258*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9259*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9260*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9261*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
9262*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
9263*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9264*4bdc9457SAndroid Build Coastguard Worker }
9265*4bdc9457SAndroid Build Coastguard Worker }
9266*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3_with_height_dilation)9267*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3_with_height_dilation) {
9268*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9269*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_height = 2; dilation_height <= 3; dilation_height++) {
9270*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9271*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9272*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9273*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9274*4bdc9457SAndroid Build Coastguard Worker .dilation_height(dilation_height)
9275*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9276*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9277*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9278*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9279*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9280*4bdc9457SAndroid Build Coastguard Worker }
9281*4bdc9457SAndroid Build Coastguard Worker }
9282*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3_with_width_dilation)9283*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3_with_width_dilation) {
9284*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9285*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) {
9286*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9287*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9288*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9289*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9290*4bdc9457SAndroid Build Coastguard Worker .dilation_width(dilation_width)
9291*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9292*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9293*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9294*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9295*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9296*4bdc9457SAndroid Build Coastguard Worker }
9297*4bdc9457SAndroid Build Coastguard Worker }
9298*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3_with_height_dilation_and_stride)9299*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3_with_height_dilation_and_stride) {
9300*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9301*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9302*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9303*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9304*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9305*4bdc9457SAndroid Build Coastguard Worker .dilation_height(3)
9306*4bdc9457SAndroid Build Coastguard Worker .stride_height(2)
9307*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9308*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9309*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9310*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9311*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9312*4bdc9457SAndroid Build Coastguard Worker }
9313*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3_with_width_dilation_and_stride)9314*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3_with_width_dilation_and_stride) {
9315*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9316*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9317*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9318*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9319*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9320*4bdc9457SAndroid Build Coastguard Worker .dilation_width(3)
9321*4bdc9457SAndroid Build Coastguard Worker .stride_width(2)
9322*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9323*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9324*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9325*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9326*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9327*4bdc9457SAndroid Build Coastguard Worker }
9328*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3_with_input_stride)9329*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3_with_input_stride) {
9330*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9331*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9332*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9333*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9334*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9335*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9336*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9337*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9338*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(47)
9339*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9340*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9341*4bdc9457SAndroid Build Coastguard Worker }
9342*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3_with_output_stride)9343*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3_with_output_stride) {
9344*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9345*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9346*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9347*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9348*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9349*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9350*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9351*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr + 3)
9352*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f16.gemm.nr * 2 + 13)
9353*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9354*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9355*4bdc9457SAndroid Build Coastguard Worker }
9356*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3_with_qmin)9357*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3_with_qmin) {
9358*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9359*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9360*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9361*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9362*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9363*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9364*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9365*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9366*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
9367*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9368*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9369*4bdc9457SAndroid Build Coastguard Worker }
9370*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3_with_qmax)9371*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3_with_qmax) {
9372*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9373*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9374*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9375*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9376*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9377*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9378*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9379*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9380*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
9381*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9382*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9383*4bdc9457SAndroid Build Coastguard Worker }
9384*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3_without_bias)9385*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3_without_bias) {
9386*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9387*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9388*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
9389*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9390*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9391*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9392*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9393*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9394*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9395*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9396*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9397*4bdc9457SAndroid Build Coastguard Worker }
9398*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,weights_cache_grouped_3x3)9399*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, weights_cache_grouped_3x3) {
9400*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9401*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9402*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9403*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9404*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9405*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9406*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9407*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9408*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
9409*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9410*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9411*4bdc9457SAndroid Build Coastguard Worker }
9412*4bdc9457SAndroid Build Coastguard Worker
9413*4bdc9457SAndroid Build Coastguard Worker /**************************** CONV path, batched ****************************/
9414*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3)9415*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3) {
9416*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9417*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9418*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9419*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9420*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9421*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9422*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9423*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9424*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9425*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9426*4bdc9457SAndroid Build Coastguard Worker }
9427*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3_with_fp32_weights)9428*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3_with_fp32_weights) {
9429*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9430*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9431*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9432*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9433*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9434*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9435*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9436*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9437*4bdc9457SAndroid Build Coastguard Worker .weights_type(DeconvolutionOperatorTester::WeightsType::FP32)
9438*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9439*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9440*4bdc9457SAndroid Build Coastguard Worker }
9441*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_Kx3)9442*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_Kx3) {
9443*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9444*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 1; kernel_height <= 4; kernel_height *= 2) {
9445*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9446*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9447*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9448*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
9449*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
9450*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
9451*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9452*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9453*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9454*4bdc9457SAndroid Build Coastguard Worker }
9455*4bdc9457SAndroid Build Coastguard Worker }
9456*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3xK)9457*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3xK) {
9458*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9459*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 1; kernel_width <= 4; kernel_width *= 2) {
9460*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9461*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9462*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9463*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
9464*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
9465*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
9466*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9467*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9468*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9469*4bdc9457SAndroid Build Coastguard Worker }
9470*4bdc9457SAndroid Build Coastguard Worker }
9471*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3_varying_height_padding)9472*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3_varying_height_padding) {
9473*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9474*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
9475*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
9476*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9477*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9478*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9479*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
9480*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
9481*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
9482*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9483*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9484*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9485*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
9486*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9487*4bdc9457SAndroid Build Coastguard Worker }
9488*4bdc9457SAndroid Build Coastguard Worker }
9489*4bdc9457SAndroid Build Coastguard Worker }
9490*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3_varying_width_padding)9491*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3_varying_width_padding) {
9492*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9493*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
9494*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
9495*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9496*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9497*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9498*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
9499*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
9500*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
9501*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9502*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9503*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9504*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
9505*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9506*4bdc9457SAndroid Build Coastguard Worker }
9507*4bdc9457SAndroid Build Coastguard Worker }
9508*4bdc9457SAndroid Build Coastguard Worker }
9509*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3_varying_height_adjustment)9510*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3_varying_height_adjustment) {
9511*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9512*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 1; adjustment_height <= 2; adjustment_height++) {
9513*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9514*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9515*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9516*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9517*4bdc9457SAndroid Build Coastguard Worker .stride_height(adjustment_height + 1)
9518*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
9519*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9520*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9521*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9522*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
9523*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9524*4bdc9457SAndroid Build Coastguard Worker }
9525*4bdc9457SAndroid Build Coastguard Worker }
9526*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3_varying_width_adjustment)9527*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3_varying_width_adjustment) {
9528*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9529*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 1; adjustment_width <= 2; adjustment_width++) {
9530*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9531*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9532*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9533*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9534*4bdc9457SAndroid Build Coastguard Worker .stride_width(adjustment_width + 1)
9535*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
9536*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9537*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9538*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9539*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
9540*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9541*4bdc9457SAndroid Build Coastguard Worker }
9542*4bdc9457SAndroid Build Coastguard Worker }
9543*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3_varying_input_height)9544*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3_varying_input_height) {
9545*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9546*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
9547*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9548*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9549*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
9550*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9551*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9552*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9553*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9554*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
9555*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9556*4bdc9457SAndroid Build Coastguard Worker }
9557*4bdc9457SAndroid Build Coastguard Worker }
9558*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3_varying_input_width)9559*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3_varying_input_width) {
9560*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9561*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
9562*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9563*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9564*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
9565*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9566*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9567*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9568*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9569*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
9570*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9571*4bdc9457SAndroid Build Coastguard Worker }
9572*4bdc9457SAndroid Build Coastguard Worker }
9573*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3_varying_input_channels)9574*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3_varying_input_channels) {
9575*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9576*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
9577*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9578*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9579*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9580*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9581*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9582*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
9583*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9584*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
9585*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9586*4bdc9457SAndroid Build Coastguard Worker }
9587*4bdc9457SAndroid Build Coastguard Worker }
9588*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3_varying_output_channels)9589*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3_varying_output_channels) {
9590*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9591*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f16.gemm.nr * 2; output_channels *= 2) {
9592*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9593*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9594*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9595*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9596*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9597*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9598*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
9599*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
9600*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9601*4bdc9457SAndroid Build Coastguard Worker }
9602*4bdc9457SAndroid Build Coastguard Worker }
9603*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3_with_height_dilation)9604*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3_with_height_dilation) {
9605*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9606*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_height = 2; dilation_height <= 3; dilation_height++) {
9607*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9608*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9609*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9610*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9611*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9612*4bdc9457SAndroid Build Coastguard Worker .dilation_height(dilation_height)
9613*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9614*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9615*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9616*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9617*4bdc9457SAndroid Build Coastguard Worker }
9618*4bdc9457SAndroid Build Coastguard Worker }
9619*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3_with_width_dilation)9620*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3_with_width_dilation) {
9621*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9622*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) {
9623*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9624*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9625*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9626*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9627*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9628*4bdc9457SAndroid Build Coastguard Worker .dilation_width(dilation_width)
9629*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9630*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9631*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9632*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9633*4bdc9457SAndroid Build Coastguard Worker }
9634*4bdc9457SAndroid Build Coastguard Worker }
9635*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3_with_height_dilation_and_stride)9636*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3_with_height_dilation_and_stride) {
9637*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9638*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9639*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9640*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9641*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9642*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9643*4bdc9457SAndroid Build Coastguard Worker .dilation_height(3)
9644*4bdc9457SAndroid Build Coastguard Worker .stride_height(2)
9645*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9646*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9647*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9648*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9649*4bdc9457SAndroid Build Coastguard Worker }
9650*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3_with_width_dilation_and_stride)9651*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3_with_width_dilation_and_stride) {
9652*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9653*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9654*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9655*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9656*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9657*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9658*4bdc9457SAndroid Build Coastguard Worker .dilation_width(3)
9659*4bdc9457SAndroid Build Coastguard Worker .stride_width(2)
9660*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9661*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9662*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9663*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9664*4bdc9457SAndroid Build Coastguard Worker }
9665*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3_with_input_stride)9666*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3_with_input_stride) {
9667*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9668*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9669*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9670*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9671*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9672*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9673*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9674*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9675*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
9676*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9677*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9678*4bdc9457SAndroid Build Coastguard Worker }
9679*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3_with_output_stride)9680*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3_with_output_stride) {
9681*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9682*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9683*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9684*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9685*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9686*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9687*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9688*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9689*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f16.gemm.nr * 2 + 13)
9690*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9691*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9692*4bdc9457SAndroid Build Coastguard Worker }
9693*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3_with_qmin)9694*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3_with_qmin) {
9695*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9696*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9697*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9698*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9699*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9700*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9701*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9702*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9703*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
9704*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9705*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9706*4bdc9457SAndroid Build Coastguard Worker }
9707*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3_with_qmax)9708*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3_with_qmax) {
9709*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9710*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9711*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9712*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9713*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9714*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9715*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9716*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9717*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
9718*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9719*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9720*4bdc9457SAndroid Build Coastguard Worker }
9721*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3_without_bias)9722*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3_without_bias) {
9723*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9724*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9725*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
9726*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9727*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9728*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9729*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9730*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9731*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9732*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9733*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9734*4bdc9457SAndroid Build Coastguard Worker }
9735*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,weights_cache_batched_3x3)9736*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, weights_cache_batched_3x3) {
9737*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9738*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9739*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9740*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9741*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9742*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9743*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9744*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9745*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
9746*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9747*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9748*4bdc9457SAndroid Build Coastguard Worker }
9749*4bdc9457SAndroid Build Coastguard Worker
9750*4bdc9457SAndroid Build Coastguard Worker /**************************** CONV path, grouped, batched ****************************/
9751*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3)9752*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3) {
9753*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9754*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9755*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9756*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9757*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9758*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9759*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9760*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9761*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9762*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9763*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9764*4bdc9457SAndroid Build Coastguard Worker }
9765*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3_with_fp32_weights)9766*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3_with_fp32_weights) {
9767*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9768*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9769*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9770*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9771*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9772*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9773*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9774*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9775*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9776*4bdc9457SAndroid Build Coastguard Worker .weights_type(DeconvolutionOperatorTester::WeightsType::FP32)
9777*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9778*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9779*4bdc9457SAndroid Build Coastguard Worker }
9780*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_Kx3)9781*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_Kx3) {
9782*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9783*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 1; kernel_height <= 4; kernel_height *= 2) {
9784*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9785*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9786*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9787*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
9788*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
9789*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9790*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
9791*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9792*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9793*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9794*4bdc9457SAndroid Build Coastguard Worker }
9795*4bdc9457SAndroid Build Coastguard Worker }
9796*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3xK)9797*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3xK) {
9798*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9799*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 1; kernel_width <= 4; kernel_width *= 2) {
9800*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9801*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9802*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9803*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
9804*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
9805*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9806*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
9807*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9808*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9809*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9810*4bdc9457SAndroid Build Coastguard Worker }
9811*4bdc9457SAndroid Build Coastguard Worker }
9812*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3_varying_height_padding)9813*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3_varying_height_padding) {
9814*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9815*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
9816*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
9817*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9818*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9819*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9820*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
9821*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
9822*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
9823*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9824*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9825*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9826*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9827*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
9828*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9829*4bdc9457SAndroid Build Coastguard Worker }
9830*4bdc9457SAndroid Build Coastguard Worker }
9831*4bdc9457SAndroid Build Coastguard Worker }
9832*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3_varying_width_padding)9833*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3_varying_width_padding) {
9834*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9835*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
9836*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
9837*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9838*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9839*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9840*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
9841*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
9842*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
9843*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9844*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9845*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9846*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9847*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
9848*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9849*4bdc9457SAndroid Build Coastguard Worker }
9850*4bdc9457SAndroid Build Coastguard Worker }
9851*4bdc9457SAndroid Build Coastguard Worker }
9852*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3_varying_height_adjustment)9853*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3_varying_height_adjustment) {
9854*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9855*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 1; adjustment_height <= 2; adjustment_height++) {
9856*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9857*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9858*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9859*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9860*4bdc9457SAndroid Build Coastguard Worker .stride_height(adjustment_height + 1)
9861*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
9862*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9863*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9864*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9865*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9866*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
9867*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9868*4bdc9457SAndroid Build Coastguard Worker }
9869*4bdc9457SAndroid Build Coastguard Worker }
9870*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3_varying_width_adjustment)9871*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3_varying_width_adjustment) {
9872*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9873*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 1; adjustment_width <= 2; adjustment_width++) {
9874*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9875*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9876*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9877*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9878*4bdc9457SAndroid Build Coastguard Worker .stride_width(adjustment_width + 1)
9879*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
9880*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9881*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9882*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9883*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9884*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
9885*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9886*4bdc9457SAndroid Build Coastguard Worker }
9887*4bdc9457SAndroid Build Coastguard Worker }
9888*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3_varying_input_height)9889*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3_varying_input_height) {
9890*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9891*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
9892*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9893*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9894*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
9895*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9896*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9897*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9898*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9899*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9900*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
9901*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9902*4bdc9457SAndroid Build Coastguard Worker }
9903*4bdc9457SAndroid Build Coastguard Worker }
9904*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3_varying_input_width)9905*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3_varying_input_width) {
9906*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9907*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
9908*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9909*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9910*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
9911*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9912*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9913*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9914*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
9915*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9916*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
9917*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9918*4bdc9457SAndroid Build Coastguard Worker }
9919*4bdc9457SAndroid Build Coastguard Worker }
9920*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3_varying_input_channels)9921*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3_varying_input_channels) {
9922*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9923*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
9924*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9925*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9926*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9927*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9928*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9929*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9930*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
9931*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9932*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
9933*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9934*4bdc9457SAndroid Build Coastguard Worker }
9935*4bdc9457SAndroid Build Coastguard Worker }
9936*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3_varying_output_channels)9937*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3_varying_output_channels) {
9938*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9939*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f16.gemm.nr * 2; output_channels *= 2) {
9940*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9941*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9942*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9943*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9944*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9945*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9946*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9947*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
9948*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
9949*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9950*4bdc9457SAndroid Build Coastguard Worker }
9951*4bdc9457SAndroid Build Coastguard Worker }
9952*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3_with_height_dilation)9953*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3_with_height_dilation) {
9954*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9955*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_height = 2; dilation_height <= 3; dilation_height++) {
9956*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9957*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9958*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9959*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9960*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9961*4bdc9457SAndroid Build Coastguard Worker .dilation_height(dilation_height)
9962*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9963*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9964*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9965*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9966*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9967*4bdc9457SAndroid Build Coastguard Worker }
9968*4bdc9457SAndroid Build Coastguard Worker }
9969*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3_with_width_dilation)9970*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3_with_width_dilation) {
9971*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9972*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) {
9973*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9974*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9975*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9976*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9977*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9978*4bdc9457SAndroid Build Coastguard Worker .dilation_width(dilation_width)
9979*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9980*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9981*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9982*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
9983*4bdc9457SAndroid Build Coastguard Worker .TestF16();
9984*4bdc9457SAndroid Build Coastguard Worker }
9985*4bdc9457SAndroid Build Coastguard Worker }
9986*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3_with_height_dilation_and_stride)9987*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3_with_height_dilation_and_stride) {
9988*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
9989*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
9990*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
9991*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
9992*4bdc9457SAndroid Build Coastguard Worker .padding(1)
9993*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
9994*4bdc9457SAndroid Build Coastguard Worker .dilation_height(3)
9995*4bdc9457SAndroid Build Coastguard Worker .stride_width(2)
9996*4bdc9457SAndroid Build Coastguard Worker .groups(2)
9997*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
9998*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
9999*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10000*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10001*4bdc9457SAndroid Build Coastguard Worker }
10002*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3_with_width_dilation_and_stride)10003*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3_with_width_dilation_and_stride) {
10004*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10005*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10006*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
10007*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
10008*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10009*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10010*4bdc9457SAndroid Build Coastguard Worker .dilation_width(3)
10011*4bdc9457SAndroid Build Coastguard Worker .stride_width(2)
10012*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10013*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
10014*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10015*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10016*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10017*4bdc9457SAndroid Build Coastguard Worker }
10018*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3_with_input_stride)10019*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3_with_input_stride) {
10020*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10021*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10022*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
10023*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
10024*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10025*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10026*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10027*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
10028*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10029*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(47)
10030*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10031*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10032*4bdc9457SAndroid Build Coastguard Worker }
10033*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3_with_output_stride)10034*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3_with_output_stride) {
10035*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10036*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10037*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
10038*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
10039*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10040*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10041*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10042*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
10043*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr + 3)
10044*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f16.gemm.nr * 2 + 13)
10045*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10046*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10047*4bdc9457SAndroid Build Coastguard Worker }
10048*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3_with_qmin)10049*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3_with_qmin) {
10050*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10051*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10052*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
10053*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
10054*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10055*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10056*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10057*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
10058*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10059*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
10060*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10061*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10062*4bdc9457SAndroid Build Coastguard Worker }
10063*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3_with_qmax)10064*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3_with_qmax) {
10065*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10066*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10067*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
10068*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
10069*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10070*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10071*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10072*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
10073*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10074*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
10075*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10076*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10077*4bdc9457SAndroid Build Coastguard Worker }
10078*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3_without_bias)10079*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3_without_bias) {
10080*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10081*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10082*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
10083*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
10084*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
10085*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10086*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10087*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10088*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
10089*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10090*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10091*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10092*4bdc9457SAndroid Build Coastguard Worker }
10093*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,weights_cache_batched_grouped_3x3)10094*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, weights_cache_batched_grouped_3x3) {
10095*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10096*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10097*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
10098*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
10099*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10100*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10101*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10102*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
10103*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10104*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
10105*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10106*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10107*4bdc9457SAndroid Build Coastguard Worker }
10108*4bdc9457SAndroid Build Coastguard Worker
10109*4bdc9457SAndroid Build Coastguard Worker /**************************** CONV path, setup ****************************/
10110*4bdc9457SAndroid Build Coastguard Worker
10111*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3_setup_changing_batch) {
10112*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10113*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10114*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
10115*4bdc9457SAndroid Build Coastguard Worker .next_batch_size(5)
10116*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
10117*4bdc9457SAndroid Build Coastguard Worker .kernel_height(3)
10118*4bdc9457SAndroid Build Coastguard Worker .kernel_width(5)
10119*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10120*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
10121*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
10122*4bdc9457SAndroid Build Coastguard Worker .TestSetupF16();
10123*4bdc9457SAndroid Build Coastguard Worker }
10124*4bdc9457SAndroid Build Coastguard Worker
10125*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3_setup_changing_height) {
10126*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10127*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10128*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
10129*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
10130*4bdc9457SAndroid Build Coastguard Worker .next_input_height(kUnstridedInputHeight + 3)
10131*4bdc9457SAndroid Build Coastguard Worker .kernel_height(3)
10132*4bdc9457SAndroid Build Coastguard Worker .kernel_width(5)
10133*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10134*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
10135*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
10136*4bdc9457SAndroid Build Coastguard Worker .TestSetupF16();
10137*4bdc9457SAndroid Build Coastguard Worker }
10138*4bdc9457SAndroid Build Coastguard Worker
10139*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3_setup_changing_width) {
10140*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10141*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10142*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
10143*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
10144*4bdc9457SAndroid Build Coastguard Worker .next_input_width(kUnstridedInputWidth + 3)
10145*4bdc9457SAndroid Build Coastguard Worker .kernel_height(3)
10146*4bdc9457SAndroid Build Coastguard Worker .kernel_width(5)
10147*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10148*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
10149*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
10150*4bdc9457SAndroid Build Coastguard Worker .TestSetupF16();
10151*4bdc9457SAndroid Build Coastguard Worker }
10152*4bdc9457SAndroid Build Coastguard Worker
10153*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/IGEMM path ****************************/
10154*4bdc9457SAndroid Build Coastguard Worker
10155*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3s2) {
10156*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10157*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10158*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10159*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10160*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10161*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10162*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
10163*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10164*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10165*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10166*4bdc9457SAndroid Build Coastguard Worker }
10167*4bdc9457SAndroid Build Coastguard Worker
10168*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3s2_with_fp32_weights) {
10169*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10170*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10171*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10172*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10173*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10174*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10175*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
10176*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10177*4bdc9457SAndroid Build Coastguard Worker .weights_type(DeconvolutionOperatorTester::WeightsType::FP32)
10178*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10179*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10180*4bdc9457SAndroid Build Coastguard Worker }
10181*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,Kx3s2)10182*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, Kx3s2) {
10183*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10184*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 2; kernel_height <= 5; kernel_height++) {
10185*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10186*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10187*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
10188*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
10189*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10190*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10191*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10192*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10193*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10194*4bdc9457SAndroid Build Coastguard Worker }
10195*4bdc9457SAndroid Build Coastguard Worker }
10196*4bdc9457SAndroid Build Coastguard Worker
10197*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3xKs2) {
10198*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10199*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 2; kernel_width <= 5; kernel_width++) {
10200*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10201*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10202*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
10203*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
10204*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10205*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10206*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10207*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10208*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10209*4bdc9457SAndroid Build Coastguard Worker }
10210*4bdc9457SAndroid Build Coastguard Worker }
10211*4bdc9457SAndroid Build Coastguard Worker
10212*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3sSx1) {
10213*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10214*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_height = 2; stride_height <= 3; stride_height++) {
10215*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10216*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10217*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10218*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
10219*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10220*4bdc9457SAndroid Build Coastguard Worker .stride_height(stride_height)
10221*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10222*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10223*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10224*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10225*4bdc9457SAndroid Build Coastguard Worker }
10226*4bdc9457SAndroid Build Coastguard Worker }
10227*4bdc9457SAndroid Build Coastguard Worker
10228*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3s1xS) {
10229*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10230*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_width = 2; stride_width <= 3; stride_width++) {
10231*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10232*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10233*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10234*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
10235*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10236*4bdc9457SAndroid Build Coastguard Worker .stride_width(stride_width)
10237*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10238*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10239*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10240*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10241*4bdc9457SAndroid Build Coastguard Worker }
10242*4bdc9457SAndroid Build Coastguard Worker }
10243*4bdc9457SAndroid Build Coastguard Worker
10244*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3s2_varying_height_padding) {
10245*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10246*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
10247*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
10248*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10249*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10250*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
10251*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
10252*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
10253*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10254*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10255*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
10256*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10257*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
10258*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10259*4bdc9457SAndroid Build Coastguard Worker }
10260*4bdc9457SAndroid Build Coastguard Worker }
10261*4bdc9457SAndroid Build Coastguard Worker }
10262*4bdc9457SAndroid Build Coastguard Worker
10263*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3s2_varying_width_padding) {
10264*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10265*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
10266*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
10267*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10268*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10269*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
10270*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
10271*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
10272*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10273*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10274*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
10275*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10276*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
10277*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10278*4bdc9457SAndroid Build Coastguard Worker }
10279*4bdc9457SAndroid Build Coastguard Worker }
10280*4bdc9457SAndroid Build Coastguard Worker }
10281*4bdc9457SAndroid Build Coastguard Worker
10282*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3s2_varying_height_adjustment) {
10283*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10284*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 0; adjustment_height <= 1; adjustment_height++) {
10285*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10286*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10287*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10288*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
10289*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10290*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10291*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
10292*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10293*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
10294*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10295*4bdc9457SAndroid Build Coastguard Worker }
10296*4bdc9457SAndroid Build Coastguard Worker }
10297*4bdc9457SAndroid Build Coastguard Worker
10298*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3s2_varying_width_adjustment) {
10299*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10300*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 0; adjustment_width <= 1; adjustment_width++) {
10301*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10302*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10303*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10304*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
10305*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10306*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10307*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
10308*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10309*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
10310*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10311*4bdc9457SAndroid Build Coastguard Worker }
10312*4bdc9457SAndroid Build Coastguard Worker }
10313*4bdc9457SAndroid Build Coastguard Worker
10314*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3s2_varying_input_height) {
10315*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10316*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
10317*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10318*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
10319*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10320*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10321*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10322*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
10323*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10324*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
10325*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10326*4bdc9457SAndroid Build Coastguard Worker }
10327*4bdc9457SAndroid Build Coastguard Worker }
10328*4bdc9457SAndroid Build Coastguard Worker
10329*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3s2_varying_input_width) {
10330*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10331*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
10332*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10333*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10334*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10335*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10336*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10337*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
10338*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10339*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
10340*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10341*4bdc9457SAndroid Build Coastguard Worker }
10342*4bdc9457SAndroid Build Coastguard Worker }
10343*4bdc9457SAndroid Build Coastguard Worker
10344*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3s2_varying_input_channels) {
10345*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10346*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
10347*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10348*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10349*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10350*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10351*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10352*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
10353*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10354*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
10355*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10356*4bdc9457SAndroid Build Coastguard Worker }
10357*4bdc9457SAndroid Build Coastguard Worker }
10358*4bdc9457SAndroid Build Coastguard Worker
10359*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3s2_varying_output_channels) {
10360*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10361*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f16.gemm.nr * 2; output_channels *= 2) {
10362*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10363*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10364*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10365*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10366*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10367*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
10368*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
10369*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
10370*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10371*4bdc9457SAndroid Build Coastguard Worker }
10372*4bdc9457SAndroid Build Coastguard Worker }
10373*4bdc9457SAndroid Build Coastguard Worker
10374*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3s2_with_input_stride) {
10375*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10376*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10377*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10378*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10379*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10380*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10381*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
10382*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10383*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
10384*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10385*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10386*4bdc9457SAndroid Build Coastguard Worker }
10387*4bdc9457SAndroid Build Coastguard Worker
10388*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3s2_with_output_stride) {
10389*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10390*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10391*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10392*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10393*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10394*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10395*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
10396*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10397*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f16.gemm.nr * 2 + 13)
10398*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10399*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10400*4bdc9457SAndroid Build Coastguard Worker }
10401*4bdc9457SAndroid Build Coastguard Worker
10402*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3s2_with_qmin) {
10403*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10404*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10405*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10406*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10407*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10408*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10409*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
10410*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10411*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
10412*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10413*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10414*4bdc9457SAndroid Build Coastguard Worker }
10415*4bdc9457SAndroid Build Coastguard Worker
10416*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3s2_with_qmax) {
10417*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10418*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10419*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10420*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10421*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10422*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10423*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
10424*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10425*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
10426*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10427*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10428*4bdc9457SAndroid Build Coastguard Worker }
10429*4bdc9457SAndroid Build Coastguard Worker
10430*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3s2_without_bias) {
10431*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10432*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10433*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
10434*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10435*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10436*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10437*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10438*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
10439*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10440*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10441*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10442*4bdc9457SAndroid Build Coastguard Worker }
10443*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,weights_cache_3x3s2)10444*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, weights_cache_3x3s2) {
10445*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10446*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10447*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10448*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10449*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10450*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10451*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
10452*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10453*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
10454*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10455*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10456*4bdc9457SAndroid Build Coastguard Worker }
10457*4bdc9457SAndroid Build Coastguard Worker
10458*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/IGEMM path, grouped ****************************/
10459*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3s2)10460*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3s2) {
10461*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10462*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10463*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10464*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10465*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10466*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10467*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10468*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10469*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10470*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10471*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10472*4bdc9457SAndroid Build Coastguard Worker }
10473*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3s2_with_fp32_weights)10474*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3s2_with_fp32_weights) {
10475*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10476*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10477*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10478*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10479*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10480*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10481*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10482*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10483*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10484*4bdc9457SAndroid Build Coastguard Worker .weights_type(DeconvolutionOperatorTester::WeightsType::FP32)
10485*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10486*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10487*4bdc9457SAndroid Build Coastguard Worker }
10488*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_Kx3s2)10489*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_Kx3s2) {
10490*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10491*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 2; kernel_height <= 5; kernel_height++) {
10492*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10493*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10494*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
10495*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
10496*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10497*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10498*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10499*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10500*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10501*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10502*4bdc9457SAndroid Build Coastguard Worker }
10503*4bdc9457SAndroid Build Coastguard Worker }
10504*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3xKs2)10505*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3xKs2) {
10506*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10507*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 2; kernel_width <= 5; kernel_width++) {
10508*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10509*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10510*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
10511*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
10512*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10513*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10514*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10515*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10516*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10517*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10518*4bdc9457SAndroid Build Coastguard Worker }
10519*4bdc9457SAndroid Build Coastguard Worker }
10520*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3sSx1)10521*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3sSx1) {
10522*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10523*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_height = 2; stride_height <= 3; stride_height++) {
10524*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10525*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10526*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10527*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
10528*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10529*4bdc9457SAndroid Build Coastguard Worker .stride_height(stride_height)
10530*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10531*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10532*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10533*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10534*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10535*4bdc9457SAndroid Build Coastguard Worker }
10536*4bdc9457SAndroid Build Coastguard Worker }
10537*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3s1xS)10538*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3s1xS) {
10539*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10540*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_width = 2; stride_width <= 3; stride_width++) {
10541*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10542*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10543*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10544*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
10545*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10546*4bdc9457SAndroid Build Coastguard Worker .stride_width(stride_width)
10547*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10548*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10549*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10550*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10551*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10552*4bdc9457SAndroid Build Coastguard Worker }
10553*4bdc9457SAndroid Build Coastguard Worker }
10554*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3s2_varying_height_padding)10555*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3s2_varying_height_padding) {
10556*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10557*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
10558*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
10559*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10560*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10561*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
10562*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
10563*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
10564*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10565*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10566*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10567*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10568*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10569*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
10570*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10571*4bdc9457SAndroid Build Coastguard Worker }
10572*4bdc9457SAndroid Build Coastguard Worker }
10573*4bdc9457SAndroid Build Coastguard Worker }
10574*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3s2_varying_width_padding)10575*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3s2_varying_width_padding) {
10576*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10577*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
10578*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
10579*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10580*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10581*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
10582*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
10583*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
10584*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10585*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10586*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10587*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10588*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10589*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
10590*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10591*4bdc9457SAndroid Build Coastguard Worker }
10592*4bdc9457SAndroid Build Coastguard Worker }
10593*4bdc9457SAndroid Build Coastguard Worker }
10594*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3s2_varying_height_adjustment)10595*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3s2_varying_height_adjustment) {
10596*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10597*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 0; adjustment_height <= 1; adjustment_height++) {
10598*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10599*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10600*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10601*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
10602*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10603*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10604*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10605*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10606*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10607*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
10608*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10609*4bdc9457SAndroid Build Coastguard Worker }
10610*4bdc9457SAndroid Build Coastguard Worker }
10611*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3s2_varying_width_adjustment)10612*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3s2_varying_width_adjustment) {
10613*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10614*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 0; adjustment_width <= 1; adjustment_width++) {
10615*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10616*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10617*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10618*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
10619*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10620*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10621*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10622*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10623*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10624*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
10625*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10626*4bdc9457SAndroid Build Coastguard Worker }
10627*4bdc9457SAndroid Build Coastguard Worker }
10628*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3s2_varying_input_height)10629*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3s2_varying_input_height) {
10630*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10631*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
10632*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10633*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
10634*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10635*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10636*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10637*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10638*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10639*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10640*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
10641*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10642*4bdc9457SAndroid Build Coastguard Worker }
10643*4bdc9457SAndroid Build Coastguard Worker }
10644*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3s2_varying_input_width)10645*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3s2_varying_input_width) {
10646*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10647*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
10648*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10649*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10650*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10651*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10652*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10653*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10654*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10655*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10656*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
10657*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10658*4bdc9457SAndroid Build Coastguard Worker }
10659*4bdc9457SAndroid Build Coastguard Worker }
10660*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3s2_varying_input_channels)10661*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3s2_varying_input_channels) {
10662*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10663*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 14; input_channels <= 20; input_channels++) {
10664*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10665*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10666*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10667*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10668*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10669*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10670*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
10671*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10672*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
10673*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10674*4bdc9457SAndroid Build Coastguard Worker }
10675*4bdc9457SAndroid Build Coastguard Worker }
10676*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3s2_varying_output_channels)10677*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3s2_varying_output_channels) {
10678*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10679*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f16.gemm.nr * 2; output_channels *= 2) {
10680*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10681*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10682*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10683*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10684*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10685*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10686*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10687*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
10688*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
10689*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10690*4bdc9457SAndroid Build Coastguard Worker }
10691*4bdc9457SAndroid Build Coastguard Worker }
10692*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3s2_with_input_stride)10693*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3s2_with_input_stride) {
10694*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10695*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10696*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10697*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10698*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10699*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10700*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10701*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10702*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10703*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(37)
10704*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10705*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10706*4bdc9457SAndroid Build Coastguard Worker }
10707*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3s2_with_output_stride)10708*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3s2_with_output_stride) {
10709*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10710*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10711*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10712*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10713*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10714*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10715*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10716*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10717*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr + 3)
10718*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f16.gemm.nr * 2 + 13)
10719*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10720*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10721*4bdc9457SAndroid Build Coastguard Worker }
10722*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3s2_with_qmin)10723*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3s2_with_qmin) {
10724*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10725*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10726*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10727*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10728*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10729*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10730*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10731*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10732*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10733*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
10734*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10735*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10736*4bdc9457SAndroid Build Coastguard Worker }
10737*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3s2_with_qmax)10738*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3s2_with_qmax) {
10739*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10740*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10741*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10742*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10743*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10744*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10745*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10746*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10747*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10748*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
10749*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10750*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10751*4bdc9457SAndroid Build Coastguard Worker }
10752*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_3x3s2_without_bias)10753*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_3x3s2_without_bias) {
10754*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10755*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10756*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
10757*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10758*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10759*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10760*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10761*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10762*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10763*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10764*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10765*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10766*4bdc9457SAndroid Build Coastguard Worker }
10767*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,weights_cache_grouped_3x3s2)10768*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, weights_cache_grouped_3x3s2) {
10769*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10770*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10771*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10772*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10773*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10774*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10775*4bdc9457SAndroid Build Coastguard Worker .groups(2)
10776*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10777*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10778*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
10779*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10780*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10781*4bdc9457SAndroid Build Coastguard Worker }
10782*4bdc9457SAndroid Build Coastguard Worker
10783*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/IGEMM path, batched ****************************/
10784*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3s2)10785*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3s2) {
10786*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10787*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10788*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
10789*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10790*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10791*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10792*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10793*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
10794*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10795*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10796*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10797*4bdc9457SAndroid Build Coastguard Worker }
10798*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3s2_with_fp32_weights)10799*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3s2_with_fp32_weights) {
10800*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10801*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10802*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
10803*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10804*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10805*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10806*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10807*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
10808*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10809*4bdc9457SAndroid Build Coastguard Worker .weights_type(DeconvolutionOperatorTester::WeightsType::FP32)
10810*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10811*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10812*4bdc9457SAndroid Build Coastguard Worker }
10813*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_Kx3s2)10814*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_Kx3s2) {
10815*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10816*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 2; kernel_height <= 5; kernel_height++) {
10817*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10818*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
10819*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10820*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
10821*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
10822*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10823*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10824*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10825*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10826*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10827*4bdc9457SAndroid Build Coastguard Worker }
10828*4bdc9457SAndroid Build Coastguard Worker }
10829*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3xKs2)10830*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3xKs2) {
10831*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10832*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 2; kernel_width <= 5; kernel_width++) {
10833*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10834*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
10835*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10836*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
10837*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
10838*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10839*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10840*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10841*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10842*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10843*4bdc9457SAndroid Build Coastguard Worker }
10844*4bdc9457SAndroid Build Coastguard Worker }
10845*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3sSx1)10846*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3sSx1) {
10847*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10848*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_height = 2; stride_height <= 3; stride_height++) {
10849*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10850*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
10851*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10852*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10853*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
10854*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10855*4bdc9457SAndroid Build Coastguard Worker .stride_height(stride_height)
10856*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10857*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10858*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10859*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10860*4bdc9457SAndroid Build Coastguard Worker }
10861*4bdc9457SAndroid Build Coastguard Worker }
10862*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3s1xS)10863*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3s1xS) {
10864*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10865*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_width = 2; stride_width <= 3; stride_width++) {
10866*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10867*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
10868*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10869*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10870*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
10871*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10872*4bdc9457SAndroid Build Coastguard Worker .stride_width(stride_width)
10873*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
10874*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10875*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
10876*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10877*4bdc9457SAndroid Build Coastguard Worker }
10878*4bdc9457SAndroid Build Coastguard Worker }
10879*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3s2_varying_height_padding)10880*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3s2_varying_height_padding) {
10881*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10882*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
10883*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
10884*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10885*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
10886*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10887*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
10888*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
10889*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
10890*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10891*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10892*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
10893*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10894*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
10895*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10896*4bdc9457SAndroid Build Coastguard Worker }
10897*4bdc9457SAndroid Build Coastguard Worker }
10898*4bdc9457SAndroid Build Coastguard Worker }
10899*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3s2_varying_width_padding)10900*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3s2_varying_width_padding) {
10901*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10902*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
10903*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
10904*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10905*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
10906*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10907*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
10908*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
10909*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
10910*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10911*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10912*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
10913*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10914*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
10915*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10916*4bdc9457SAndroid Build Coastguard Worker }
10917*4bdc9457SAndroid Build Coastguard Worker }
10918*4bdc9457SAndroid Build Coastguard Worker }
10919*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3s2_varying_height_adjustment)10920*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3s2_varying_height_adjustment) {
10921*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10922*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 0; adjustment_height <= 1; adjustment_height++) {
10923*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10924*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
10925*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10926*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10927*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
10928*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10929*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10930*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
10931*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10932*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
10933*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10934*4bdc9457SAndroid Build Coastguard Worker }
10935*4bdc9457SAndroid Build Coastguard Worker }
10936*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3s2_varying_width_adjustment)10937*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3s2_varying_width_adjustment) {
10938*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10939*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 0; adjustment_width <= 1; adjustment_width++) {
10940*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10941*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
10942*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10943*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10944*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
10945*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10946*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10947*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
10948*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10949*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
10950*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10951*4bdc9457SAndroid Build Coastguard Worker }
10952*4bdc9457SAndroid Build Coastguard Worker }
10953*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3s2_varying_input_height)10954*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3s2_varying_input_height) {
10955*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10956*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
10957*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10958*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
10959*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
10960*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10961*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10962*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10963*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
10964*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10965*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
10966*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10967*4bdc9457SAndroid Build Coastguard Worker }
10968*4bdc9457SAndroid Build Coastguard Worker }
10969*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3s2_varying_input_width)10970*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3s2_varying_input_width) {
10971*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10972*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
10973*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10974*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
10975*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10976*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10977*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10978*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10979*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
10980*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10981*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
10982*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10983*4bdc9457SAndroid Build Coastguard Worker }
10984*4bdc9457SAndroid Build Coastguard Worker }
10985*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3s2_varying_input_channels)10986*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3s2_varying_input_channels) {
10987*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
10988*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
10989*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
10990*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
10991*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
10992*4bdc9457SAndroid Build Coastguard Worker .padding(1)
10993*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
10994*4bdc9457SAndroid Build Coastguard Worker .stride(2)
10995*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
10996*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
10997*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
10998*4bdc9457SAndroid Build Coastguard Worker .TestF16();
10999*4bdc9457SAndroid Build Coastguard Worker }
11000*4bdc9457SAndroid Build Coastguard Worker }
11001*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3s2_varying_output_channels)11002*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3s2_varying_output_channels) {
11003*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11004*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f16.gemm.nr * 2; output_channels *= 2) {
11005*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11006*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11007*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11008*4bdc9457SAndroid Build Coastguard Worker .padding(1)
11009*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11010*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11011*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
11012*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
11013*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
11014*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11015*4bdc9457SAndroid Build Coastguard Worker }
11016*4bdc9457SAndroid Build Coastguard Worker }
11017*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3s2_with_input_stride)11018*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3s2_with_input_stride) {
11019*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11020*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11021*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11022*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11023*4bdc9457SAndroid Build Coastguard Worker .padding(1)
11024*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11025*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11026*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
11027*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11028*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
11029*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11030*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11031*4bdc9457SAndroid Build Coastguard Worker }
11032*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3s2_with_output_stride)11033*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3s2_with_output_stride) {
11034*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11035*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11036*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11037*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11038*4bdc9457SAndroid Build Coastguard Worker .padding(1)
11039*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11040*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11041*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
11042*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11043*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f16.gemm.nr * 2 + 13)
11044*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11045*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11046*4bdc9457SAndroid Build Coastguard Worker }
11047*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3s2_with_qmin)11048*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3s2_with_qmin) {
11049*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11050*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11051*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11052*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11053*4bdc9457SAndroid Build Coastguard Worker .padding(1)
11054*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11055*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11056*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
11057*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11058*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
11059*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11060*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11061*4bdc9457SAndroid Build Coastguard Worker }
11062*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3s2_with_qmax)11063*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3s2_with_qmax) {
11064*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11065*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11066*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11067*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11068*4bdc9457SAndroid Build Coastguard Worker .padding(1)
11069*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11070*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11071*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
11072*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11073*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
11074*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11075*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11076*4bdc9457SAndroid Build Coastguard Worker }
11077*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_3x3s2_without_bias)11078*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_3x3s2_without_bias) {
11079*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11080*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11081*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
11082*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11083*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11084*4bdc9457SAndroid Build Coastguard Worker .padding(1)
11085*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11086*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11087*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
11088*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11089*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11090*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11091*4bdc9457SAndroid Build Coastguard Worker }
11092*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,weights_cache_batched_3x3s2)11093*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, weights_cache_batched_3x3s2) {
11094*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11095*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11096*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11097*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11098*4bdc9457SAndroid Build Coastguard Worker .padding(1)
11099*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11100*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11101*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
11102*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11103*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
11104*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11105*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11106*4bdc9457SAndroid Build Coastguard Worker }
11107*4bdc9457SAndroid Build Coastguard Worker
11108*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/IGEMM path, grouped, batched ****************************/
11109*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3s2)11110*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3s2) {
11111*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11112*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11113*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11114*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11115*4bdc9457SAndroid Build Coastguard Worker .padding(1)
11116*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11117*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11118*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11119*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11120*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11121*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11122*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11123*4bdc9457SAndroid Build Coastguard Worker }
11124*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3s2_with_fp32_weights)11125*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3s2_with_fp32_weights) {
11126*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11127*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11128*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11129*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11130*4bdc9457SAndroid Build Coastguard Worker .padding(1)
11131*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11132*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11133*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11134*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11135*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11136*4bdc9457SAndroid Build Coastguard Worker .weights_type(DeconvolutionOperatorTester::WeightsType::FP32)
11137*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11138*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11139*4bdc9457SAndroid Build Coastguard Worker }
11140*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_Kx3s2)11141*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_Kx3s2) {
11142*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11143*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 2; kernel_height <= 5; kernel_height++) {
11144*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11145*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11146*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11147*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
11148*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
11149*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11150*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11151*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11152*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11153*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11154*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11155*4bdc9457SAndroid Build Coastguard Worker }
11156*4bdc9457SAndroid Build Coastguard Worker }
11157*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3xKs2)11158*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3xKs2) {
11159*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11160*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 2; kernel_width <= 5; kernel_width++) {
11161*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11162*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11163*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11164*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
11165*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
11166*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11167*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11168*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11169*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11170*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11171*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11172*4bdc9457SAndroid Build Coastguard Worker }
11173*4bdc9457SAndroid Build Coastguard Worker }
11174*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3sSx1)11175*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3sSx1) {
11176*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11177*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_height = 2; stride_height <= 3; stride_height++) {
11178*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11179*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11180*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11181*4bdc9457SAndroid Build Coastguard Worker .padding(1)
11182*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
11183*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11184*4bdc9457SAndroid Build Coastguard Worker .stride_height(stride_height)
11185*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11186*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11187*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11188*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11189*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11190*4bdc9457SAndroid Build Coastguard Worker }
11191*4bdc9457SAndroid Build Coastguard Worker }
11192*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3s1xS)11193*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3s1xS) {
11194*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11195*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_width = 2; stride_width <= 3; stride_width++) {
11196*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11197*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11198*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11199*4bdc9457SAndroid Build Coastguard Worker .padding(1)
11200*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
11201*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11202*4bdc9457SAndroid Build Coastguard Worker .stride_width(stride_width)
11203*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11204*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11205*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11206*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11207*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11208*4bdc9457SAndroid Build Coastguard Worker }
11209*4bdc9457SAndroid Build Coastguard Worker }
11210*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3s2_varying_height_padding)11211*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3s2_varying_height_padding) {
11212*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11213*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
11214*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
11215*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11216*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11217*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11218*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
11219*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
11220*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
11221*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11222*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11223*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11224*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11225*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11226*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
11227*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11228*4bdc9457SAndroid Build Coastguard Worker }
11229*4bdc9457SAndroid Build Coastguard Worker }
11230*4bdc9457SAndroid Build Coastguard Worker }
11231*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3s2_varying_width_padding)11232*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3s2_varying_width_padding) {
11233*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11234*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
11235*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
11236*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11237*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11238*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11239*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
11240*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
11241*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
11242*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11243*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11244*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11245*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11246*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11247*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
11248*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11249*4bdc9457SAndroid Build Coastguard Worker }
11250*4bdc9457SAndroid Build Coastguard Worker }
11251*4bdc9457SAndroid Build Coastguard Worker }
11252*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3s2_varying_height_adjustment)11253*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3s2_varying_height_adjustment) {
11254*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11255*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 0; adjustment_height <= 1; adjustment_height++) {
11256*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11257*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11258*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11259*4bdc9457SAndroid Build Coastguard Worker .padding(1)
11260*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
11261*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11262*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11263*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11264*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11265*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11266*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
11267*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11268*4bdc9457SAndroid Build Coastguard Worker }
11269*4bdc9457SAndroid Build Coastguard Worker }
11270*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3s2_varying_width_adjustment)11271*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3s2_varying_width_adjustment) {
11272*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11273*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 0; adjustment_width <= 1; adjustment_width++) {
11274*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11275*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11276*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11277*4bdc9457SAndroid Build Coastguard Worker .padding(1)
11278*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
11279*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11280*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11281*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11282*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11283*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11284*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
11285*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11286*4bdc9457SAndroid Build Coastguard Worker }
11287*4bdc9457SAndroid Build Coastguard Worker }
11288*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3s2_varying_input_height)11289*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3s2_varying_input_height) {
11290*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11291*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
11292*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11293*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11294*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
11295*4bdc9457SAndroid Build Coastguard Worker .padding(1)
11296*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11297*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11298*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11299*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11300*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11301*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
11302*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11303*4bdc9457SAndroid Build Coastguard Worker }
11304*4bdc9457SAndroid Build Coastguard Worker }
11305*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3s2_varying_input_width)11306*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3s2_varying_input_width) {
11307*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11308*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
11309*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11310*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11311*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11312*4bdc9457SAndroid Build Coastguard Worker .padding(1)
11313*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11314*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11315*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11316*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11317*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11318*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
11319*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11320*4bdc9457SAndroid Build Coastguard Worker }
11321*4bdc9457SAndroid Build Coastguard Worker }
11322*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3s2_varying_input_channels)11323*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3s2_varying_input_channels) {
11324*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11325*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 14; input_channels <= 20; input_channels++) {
11326*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11327*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11328*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11329*4bdc9457SAndroid Build Coastguard Worker .padding(1)
11330*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11331*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11332*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11333*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
11334*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11335*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
11336*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11337*4bdc9457SAndroid Build Coastguard Worker }
11338*4bdc9457SAndroid Build Coastguard Worker }
11339*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3s2_varying_output_channels)11340*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3s2_varying_output_channels) {
11341*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11342*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f16.gemm.nr * 2; output_channels *= 2) {
11343*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11344*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11345*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11346*4bdc9457SAndroid Build Coastguard Worker .padding(1)
11347*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11348*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11349*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11350*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11351*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
11352*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
11353*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11354*4bdc9457SAndroid Build Coastguard Worker }
11355*4bdc9457SAndroid Build Coastguard Worker }
11356*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3s2_with_input_stride)11357*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3s2_with_input_stride) {
11358*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11359*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11360*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11361*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11362*4bdc9457SAndroid Build Coastguard Worker .padding(1)
11363*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11364*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11365*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11366*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11367*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11368*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(37)
11369*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11370*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11371*4bdc9457SAndroid Build Coastguard Worker }
11372*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3s2_with_output_stride)11373*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3s2_with_output_stride) {
11374*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11375*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11376*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11377*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11378*4bdc9457SAndroid Build Coastguard Worker .padding(1)
11379*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11380*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11381*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11382*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11383*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr + 3)
11384*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f16.gemm.nr * 2 + 13)
11385*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11386*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11387*4bdc9457SAndroid Build Coastguard Worker }
11388*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3s2_with_qmin)11389*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3s2_with_qmin) {
11390*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11391*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11392*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11393*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11394*4bdc9457SAndroid Build Coastguard Worker .padding(1)
11395*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11396*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11397*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11398*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11399*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11400*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
11401*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11402*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11403*4bdc9457SAndroid Build Coastguard Worker }
11404*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3s2_with_qmax)11405*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3s2_with_qmax) {
11406*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11407*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11408*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11409*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11410*4bdc9457SAndroid Build Coastguard Worker .padding(1)
11411*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11412*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11413*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11414*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11415*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11416*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
11417*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11418*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11419*4bdc9457SAndroid Build Coastguard Worker }
11420*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_3x3s2_without_bias)11421*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_3x3s2_without_bias) {
11422*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11423*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11424*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
11425*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11426*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11427*4bdc9457SAndroid Build Coastguard Worker .padding(1)
11428*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11429*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11430*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11431*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11432*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11433*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11434*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11435*4bdc9457SAndroid Build Coastguard Worker }
11436*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,weights_cache_batched_grouped_3x3s2)11437*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, weights_cache_batched_grouped_3x3s2) {
11438*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11439*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11440*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11441*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11442*4bdc9457SAndroid Build Coastguard Worker .padding(1)
11443*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11444*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11445*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11446*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11447*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11448*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
11449*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11450*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11451*4bdc9457SAndroid Build Coastguard Worker }
11452*4bdc9457SAndroid Build Coastguard Worker
11453*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/IGEMM path, setup ****************************/
11454*4bdc9457SAndroid Build Coastguard Worker
11455*4bdc9457SAndroid Build Coastguard Worker
11456*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3s2_setup_changing_height) {
11457*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11458*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11459*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11460*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11461*4bdc9457SAndroid Build Coastguard Worker .next_input_height(kStridedInputHeight + 3)
11462*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11463*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11464*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11465*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
11466*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
11467*4bdc9457SAndroid Build Coastguard Worker .TestSetupF16();
11468*4bdc9457SAndroid Build Coastguard Worker }
11469*4bdc9457SAndroid Build Coastguard Worker
11470*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 3x3s2_setup_changing_width) {
11471*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11472*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11473*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11474*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11475*4bdc9457SAndroid Build Coastguard Worker .next_input_width(kStridedInputWidth + 3)
11476*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
11477*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11478*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11479*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
11480*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
11481*4bdc9457SAndroid Build Coastguard Worker .TestSetupF16();
11482*4bdc9457SAndroid Build Coastguard Worker }
11483*4bdc9457SAndroid Build Coastguard Worker
11484*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/GEMM path ****************************/
11485*4bdc9457SAndroid Build Coastguard Worker
11486*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 2x2s2) {
11487*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11488*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11489*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11490*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11491*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11492*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
11493*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11494*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11495*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11496*4bdc9457SAndroid Build Coastguard Worker }
11497*4bdc9457SAndroid Build Coastguard Worker
11498*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 2x2s2_with_fp32_weights) {
11499*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11500*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11501*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11502*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11503*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11504*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
11505*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11506*4bdc9457SAndroid Build Coastguard Worker .weights_type(DeconvolutionOperatorTester::WeightsType::FP32)
11507*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11508*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11509*4bdc9457SAndroid Build Coastguard Worker }
11510*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,Kx2sKx2)11511*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, Kx2sKx2) {
11512*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11513*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 3; kernel_height <= 5; kernel_height++) {
11514*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11515*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11516*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 2)
11517*4bdc9457SAndroid Build Coastguard Worker .stride(kernel_height, 2)
11518*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11519*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11520*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11521*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11522*4bdc9457SAndroid Build Coastguard Worker }
11523*4bdc9457SAndroid Build Coastguard Worker }
11524*4bdc9457SAndroid Build Coastguard Worker
11525*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 2xKs2xK) {
11526*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11527*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 3; kernel_width <= 5; kernel_width++) {
11528*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11529*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11530*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, kernel_width)
11531*4bdc9457SAndroid Build Coastguard Worker .stride(2, kernel_width)
11532*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11533*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11534*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11535*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11536*4bdc9457SAndroid Build Coastguard Worker }
11537*4bdc9457SAndroid Build Coastguard Worker }
11538*4bdc9457SAndroid Build Coastguard Worker
11539*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 2x2s2_height_adjustment) {
11540*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11541*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11542*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11543*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(1)
11544*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11545*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11546*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
11547*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11548*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
11549*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11550*4bdc9457SAndroid Build Coastguard Worker }
11551*4bdc9457SAndroid Build Coastguard Worker
11552*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 2x2s2_width_adjustment) {
11553*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11554*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11555*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11556*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(1)
11557*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11558*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11559*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
11560*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11561*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
11562*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11563*4bdc9457SAndroid Build Coastguard Worker }
11564*4bdc9457SAndroid Build Coastguard Worker
11565*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 2x2s2_varying_input_height) {
11566*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11567*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
11568*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11569*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
11570*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11571*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11572*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
11573*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11574*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
11575*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11576*4bdc9457SAndroid Build Coastguard Worker }
11577*4bdc9457SAndroid Build Coastguard Worker }
11578*4bdc9457SAndroid Build Coastguard Worker
11579*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 2x2s2_varying_input_width) {
11580*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11581*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
11582*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11583*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11584*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11585*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11586*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
11587*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11588*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
11589*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11590*4bdc9457SAndroid Build Coastguard Worker }
11591*4bdc9457SAndroid Build Coastguard Worker }
11592*4bdc9457SAndroid Build Coastguard Worker
11593*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 2x2s2_varying_input_channels) {
11594*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11595*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
11596*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11597*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11598*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11599*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11600*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
11601*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11602*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
11603*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11604*4bdc9457SAndroid Build Coastguard Worker }
11605*4bdc9457SAndroid Build Coastguard Worker }
11606*4bdc9457SAndroid Build Coastguard Worker
11607*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 2x2s2_varying_output_channels) {
11608*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11609*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f16.gemm.nr * 2; output_channels *= 2) {
11610*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11611*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11612*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11613*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11614*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
11615*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
11616*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
11617*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11618*4bdc9457SAndroid Build Coastguard Worker }
11619*4bdc9457SAndroid Build Coastguard Worker }
11620*4bdc9457SAndroid Build Coastguard Worker
11621*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 2x2s2_with_input_stride) {
11622*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11623*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11624*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11625*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11626*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11627*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
11628*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11629*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
11630*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11631*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11632*4bdc9457SAndroid Build Coastguard Worker }
11633*4bdc9457SAndroid Build Coastguard Worker
11634*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 2x2s2_with_output_stride) {
11635*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11636*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11637*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11638*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11639*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11640*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
11641*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11642*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f16.gemm.nr * 2 + 13)
11643*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11644*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11645*4bdc9457SAndroid Build Coastguard Worker }
11646*4bdc9457SAndroid Build Coastguard Worker
11647*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 2x2s2_with_qmin) {
11648*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11649*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11650*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11651*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11652*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11653*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
11654*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11655*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
11656*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11657*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11658*4bdc9457SAndroid Build Coastguard Worker }
11659*4bdc9457SAndroid Build Coastguard Worker
11660*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 2x2s2_with_qmax) {
11661*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11662*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11663*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11664*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11665*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11666*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
11667*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11668*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
11669*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11670*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11671*4bdc9457SAndroid Build Coastguard Worker }
11672*4bdc9457SAndroid Build Coastguard Worker
11673*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 2x2s2_without_bias) {
11674*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11675*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11676*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
11677*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11678*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11679*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11680*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
11681*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11682*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11683*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11684*4bdc9457SAndroid Build Coastguard Worker }
11685*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,weights_cache_2x2s2)11686*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, weights_cache_2x2s2) {
11687*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11688*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11689*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11690*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11691*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11692*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
11693*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11694*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
11695*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11696*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11697*4bdc9457SAndroid Build Coastguard Worker }
11698*4bdc9457SAndroid Build Coastguard Worker
11699*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/GEMM path, grouped ****************************/
11700*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_2x2s2)11701*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_2x2s2) {
11702*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11703*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11704*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11705*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11706*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11707*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11708*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11709*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11710*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11711*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11712*4bdc9457SAndroid Build Coastguard Worker }
11713*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_2x2s2_with_fp32_weights)11714*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_2x2s2_with_fp32_weights) {
11715*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11716*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11717*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11718*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11719*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11720*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11721*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11722*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11723*4bdc9457SAndroid Build Coastguard Worker .weights_type(DeconvolutionOperatorTester::WeightsType::FP32)
11724*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11725*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11726*4bdc9457SAndroid Build Coastguard Worker }
11727*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_Kx2sKx2)11728*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_Kx2sKx2) {
11729*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11730*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 3; kernel_height <= 5; kernel_height++) {
11731*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11732*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11733*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 2)
11734*4bdc9457SAndroid Build Coastguard Worker .stride(kernel_height, 2)
11735*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11736*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11737*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11738*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11739*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11740*4bdc9457SAndroid Build Coastguard Worker }
11741*4bdc9457SAndroid Build Coastguard Worker }
11742*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_2xKs2xK)11743*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_2xKs2xK) {
11744*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11745*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 3; kernel_width <= 5; kernel_width++) {
11746*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11747*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11748*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, kernel_width)
11749*4bdc9457SAndroid Build Coastguard Worker .stride(2, kernel_width)
11750*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11751*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11752*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11753*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11754*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11755*4bdc9457SAndroid Build Coastguard Worker }
11756*4bdc9457SAndroid Build Coastguard Worker }
11757*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_2x2s2_height_adjustment)11758*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_2x2s2_height_adjustment) {
11759*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11760*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11761*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11762*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(1)
11763*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11764*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11765*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11766*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11767*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11768*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
11769*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11770*4bdc9457SAndroid Build Coastguard Worker }
11771*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_2x2s2_width_adjustment)11772*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_2x2s2_width_adjustment) {
11773*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11774*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11775*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11776*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(1)
11777*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11778*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11779*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11780*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11781*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11782*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
11783*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11784*4bdc9457SAndroid Build Coastguard Worker }
11785*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_2x2s2_varying_input_height)11786*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_2x2s2_varying_input_height) {
11787*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11788*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
11789*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11790*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
11791*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11792*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11793*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11794*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11795*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11796*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
11797*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11798*4bdc9457SAndroid Build Coastguard Worker }
11799*4bdc9457SAndroid Build Coastguard Worker }
11800*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_2x2s2_varying_input_width)11801*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_2x2s2_varying_input_width) {
11802*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11803*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
11804*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11805*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11806*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11807*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11808*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11809*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11810*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11811*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
11812*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11813*4bdc9457SAndroid Build Coastguard Worker }
11814*4bdc9457SAndroid Build Coastguard Worker }
11815*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_2x2s2_varying_input_channels)11816*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_2x2s2_varying_input_channels) {
11817*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11818*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 14; input_channels <= 20; input_channels++) {
11819*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11820*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11821*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11822*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11823*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11824*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
11825*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11826*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
11827*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11828*4bdc9457SAndroid Build Coastguard Worker }
11829*4bdc9457SAndroid Build Coastguard Worker }
11830*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_2x2s2_varying_output_channels)11831*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_2x2s2_varying_output_channels) {
11832*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11833*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f16.gemm.nr * 2; output_channels *= 2) {
11834*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11835*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11836*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11837*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11838*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11839*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11840*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
11841*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
11842*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11843*4bdc9457SAndroid Build Coastguard Worker }
11844*4bdc9457SAndroid Build Coastguard Worker }
11845*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_2x2s2_with_input_stride)11846*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_2x2s2_with_input_stride) {
11847*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11848*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11849*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11850*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11851*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11852*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11853*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11854*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11855*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(37)
11856*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11857*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11858*4bdc9457SAndroid Build Coastguard Worker }
11859*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_2x2s2_with_output_stride)11860*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_2x2s2_with_output_stride) {
11861*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11862*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11863*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11864*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11865*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11866*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11867*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11868*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr + 3)
11869*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f16.gemm.nr * 2 + 13)
11870*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11871*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11872*4bdc9457SAndroid Build Coastguard Worker }
11873*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_2x2s2_with_qmin)11874*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_2x2s2_with_qmin) {
11875*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11876*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11877*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11878*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11879*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11880*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11881*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11882*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11883*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
11884*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11885*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11886*4bdc9457SAndroid Build Coastguard Worker }
11887*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_2x2s2_with_qmax)11888*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_2x2s2_with_qmax) {
11889*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11890*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11891*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11892*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11893*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11894*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11895*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11896*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11897*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
11898*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11899*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11900*4bdc9457SAndroid Build Coastguard Worker }
11901*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,grouped_2x2s2_without_bias)11902*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, grouped_2x2s2_without_bias) {
11903*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11904*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11905*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
11906*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11907*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11908*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11909*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11910*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11911*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11912*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11913*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11914*4bdc9457SAndroid Build Coastguard Worker }
11915*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,weights_cache_grouped_2x2s2)11916*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, weights_cache_grouped_2x2s2) {
11917*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11918*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11919*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11920*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11921*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11922*4bdc9457SAndroid Build Coastguard Worker .groups(2)
11923*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11924*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11925*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
11926*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11927*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11928*4bdc9457SAndroid Build Coastguard Worker }
11929*4bdc9457SAndroid Build Coastguard Worker
11930*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/GEMM path, batched ****************************/
11931*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_2x2s2)11932*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_2x2s2) {
11933*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11934*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11935*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11936*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11937*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11938*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11939*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
11940*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11941*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11942*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11943*4bdc9457SAndroid Build Coastguard Worker }
11944*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_2x2s2_with_fp32_weights)11945*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_2x2s2_with_fp32_weights) {
11946*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11947*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11948*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11949*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11950*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11951*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11952*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
11953*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11954*4bdc9457SAndroid Build Coastguard Worker .weights_type(DeconvolutionOperatorTester::WeightsType::FP32)
11955*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11956*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11957*4bdc9457SAndroid Build Coastguard Worker }
11958*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_Kx2sKx2)11959*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_Kx2sKx2) {
11960*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11961*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 3; kernel_height <= 5; kernel_height++) {
11962*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11963*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11964*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11965*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 2)
11966*4bdc9457SAndroid Build Coastguard Worker .stride(kernel_height, 2)
11967*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11968*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11969*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11970*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11971*4bdc9457SAndroid Build Coastguard Worker }
11972*4bdc9457SAndroid Build Coastguard Worker }
11973*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_2xKs2xK)11974*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_2xKs2xK) {
11975*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11976*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 3; kernel_width <= 5; kernel_width++) {
11977*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11978*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11979*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11980*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, kernel_width)
11981*4bdc9457SAndroid Build Coastguard Worker .stride(2, kernel_width)
11982*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
11983*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11984*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
11985*4bdc9457SAndroid Build Coastguard Worker .TestF16();
11986*4bdc9457SAndroid Build Coastguard Worker }
11987*4bdc9457SAndroid Build Coastguard Worker }
11988*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_2x2s2_height_adjustment)11989*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_2x2s2_height_adjustment) {
11990*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
11991*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
11992*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
11993*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
11994*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(1)
11995*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
11996*4bdc9457SAndroid Build Coastguard Worker .stride(2)
11997*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
11998*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
11999*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12000*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12001*4bdc9457SAndroid Build Coastguard Worker }
12002*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_2x2s2_width_adjustment)12003*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_2x2s2_width_adjustment) {
12004*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12005*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12006*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12007*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12008*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(1)
12009*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12010*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12011*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
12012*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
12013*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12014*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12015*4bdc9457SAndroid Build Coastguard Worker }
12016*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_2x2s2_varying_input_height)12017*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_2x2s2_varying_input_height) {
12018*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12019*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
12020*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12021*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12022*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
12023*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12024*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12025*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
12026*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
12027*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12028*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12029*4bdc9457SAndroid Build Coastguard Worker }
12030*4bdc9457SAndroid Build Coastguard Worker }
12031*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_2x2s2_varying_input_width)12032*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_2x2s2_varying_input_width) {
12033*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12034*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
12035*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12036*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12037*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12038*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12039*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12040*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
12041*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
12042*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12043*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12044*4bdc9457SAndroid Build Coastguard Worker }
12045*4bdc9457SAndroid Build Coastguard Worker }
12046*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_2x2s2_varying_input_channels)12047*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_2x2s2_varying_input_channels) {
12048*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12049*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
12050*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12051*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12052*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12053*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12054*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12055*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
12056*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
12057*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12058*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12059*4bdc9457SAndroid Build Coastguard Worker }
12060*4bdc9457SAndroid Build Coastguard Worker }
12061*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_2x2s2_varying_output_channels)12062*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_2x2s2_varying_output_channels) {
12063*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12064*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f16.gemm.nr * 2; output_channels *= 2) {
12065*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12066*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12067*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12068*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12069*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12070*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12071*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
12072*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12073*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12074*4bdc9457SAndroid Build Coastguard Worker }
12075*4bdc9457SAndroid Build Coastguard Worker }
12076*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_2x2s2_with_input_stride)12077*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_2x2s2_with_input_stride) {
12078*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12079*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12080*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12081*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12082*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12083*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12084*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12085*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
12086*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
12087*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12088*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12089*4bdc9457SAndroid Build Coastguard Worker }
12090*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_2x2s2_with_output_stride)12091*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_2x2s2_with_output_stride) {
12092*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12093*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12094*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12095*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12096*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12097*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12098*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12099*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
12100*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f16.gemm.nr * 2 + 13)
12101*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12102*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12103*4bdc9457SAndroid Build Coastguard Worker }
12104*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_2x2s2_with_qmin)12105*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_2x2s2_with_qmin) {
12106*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12107*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12108*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12109*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12110*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12111*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12112*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12113*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
12114*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
12115*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12116*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12117*4bdc9457SAndroid Build Coastguard Worker }
12118*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_2x2s2_with_qmax)12119*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_2x2s2_with_qmax) {
12120*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12121*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12122*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12123*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12124*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12125*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12126*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12127*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
12128*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
12129*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12130*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12131*4bdc9457SAndroid Build Coastguard Worker }
12132*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_2x2s2_without_bias)12133*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_2x2s2_without_bias) {
12134*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12135*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12136*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
12137*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12138*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12139*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12140*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12141*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12142*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
12143*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12144*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12145*4bdc9457SAndroid Build Coastguard Worker }
12146*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,weights_cache_batched_2x2s2)12147*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, weights_cache_batched_2x2s2) {
12148*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12149*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12150*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12151*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12152*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12153*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12154*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
12155*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
12156*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
12157*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12158*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12159*4bdc9457SAndroid Build Coastguard Worker }
12160*4bdc9457SAndroid Build Coastguard Worker
12161*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/GEMM path, grouped, batched ****************************/
12162*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_2x2s2)12163*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_2x2s2) {
12164*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12165*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12166*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12167*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12168*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12169*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12170*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12171*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
12172*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
12173*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12174*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12175*4bdc9457SAndroid Build Coastguard Worker }
12176*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_2x2s2_with_fp32_weights)12177*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_2x2s2_with_fp32_weights) {
12178*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12179*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12180*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12181*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12182*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12183*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12184*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12185*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
12186*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
12187*4bdc9457SAndroid Build Coastguard Worker .weights_type(DeconvolutionOperatorTester::WeightsType::FP32)
12188*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12189*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12190*4bdc9457SAndroid Build Coastguard Worker }
12191*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_Kx2sKx2)12192*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_Kx2sKx2) {
12193*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12194*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 3; kernel_height <= 5; kernel_height++) {
12195*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12196*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12197*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12198*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 2)
12199*4bdc9457SAndroid Build Coastguard Worker .stride(kernel_height, 2)
12200*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12201*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
12202*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
12203*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12204*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12205*4bdc9457SAndroid Build Coastguard Worker }
12206*4bdc9457SAndroid Build Coastguard Worker }
12207*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_2xKs2xK)12208*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_2xKs2xK) {
12209*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12210*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 3; kernel_width <= 5; kernel_width++) {
12211*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12212*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12213*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12214*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, kernel_width)
12215*4bdc9457SAndroid Build Coastguard Worker .stride(2, kernel_width)
12216*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12217*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
12218*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
12219*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12220*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12221*4bdc9457SAndroid Build Coastguard Worker }
12222*4bdc9457SAndroid Build Coastguard Worker }
12223*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_2x2s2_height_adjustment)12224*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_2x2s2_height_adjustment) {
12225*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12226*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12227*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12228*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12229*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(1)
12230*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12231*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12232*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12233*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
12234*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
12235*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12236*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12237*4bdc9457SAndroid Build Coastguard Worker }
12238*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_2x2s2_width_adjustment)12239*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_2x2s2_width_adjustment) {
12240*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12241*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12242*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12243*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12244*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(1)
12245*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12246*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12247*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12248*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
12249*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
12250*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12251*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12252*4bdc9457SAndroid Build Coastguard Worker }
12253*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_2x2s2_varying_input_height)12254*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_2x2s2_varying_input_height) {
12255*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12256*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
12257*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12258*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12259*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
12260*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12261*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12262*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12263*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
12264*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
12265*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12266*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12267*4bdc9457SAndroid Build Coastguard Worker }
12268*4bdc9457SAndroid Build Coastguard Worker }
12269*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_2x2s2_varying_input_width)12270*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_2x2s2_varying_input_width) {
12271*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12272*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
12273*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12274*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12275*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12276*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12277*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12278*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12279*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
12280*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
12281*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12282*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12283*4bdc9457SAndroid Build Coastguard Worker }
12284*4bdc9457SAndroid Build Coastguard Worker }
12285*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_2x2s2_varying_input_channels)12286*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_2x2s2_varying_input_channels) {
12287*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12288*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 14; input_channels <= 20; input_channels++) {
12289*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12290*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12291*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12292*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12293*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12294*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12295*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
12296*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
12297*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12298*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12299*4bdc9457SAndroid Build Coastguard Worker }
12300*4bdc9457SAndroid Build Coastguard Worker }
12301*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_2x2s2_varying_output_channels)12302*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_2x2s2_varying_output_channels) {
12303*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12304*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f16.gemm.nr * 2; output_channels *= 2) {
12305*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12306*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12307*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12308*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12309*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12310*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12311*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
12312*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
12313*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12314*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12315*4bdc9457SAndroid Build Coastguard Worker }
12316*4bdc9457SAndroid Build Coastguard Worker }
12317*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_2x2s2_with_input_stride)12318*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_2x2s2_with_input_stride) {
12319*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12320*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12321*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12322*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12323*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12324*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12325*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12326*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
12327*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
12328*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(37)
12329*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12330*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12331*4bdc9457SAndroid Build Coastguard Worker }
12332*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_2x2s2_with_output_stride)12333*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_2x2s2_with_output_stride) {
12334*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12335*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12336*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12337*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12338*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12339*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12340*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12341*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
12342*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr + 3)
12343*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f16.gemm.nr * 2 + 13)
12344*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12345*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12346*4bdc9457SAndroid Build Coastguard Worker }
12347*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_2x2s2_with_qmin)12348*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_2x2s2_with_qmin) {
12349*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12350*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12351*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12352*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12353*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12354*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12355*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12356*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
12357*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
12358*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
12359*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12360*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12361*4bdc9457SAndroid Build Coastguard Worker }
12362*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_2x2s2_with_qmax)12363*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_2x2s2_with_qmax) {
12364*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12365*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12366*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12367*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12368*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12369*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12370*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12371*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
12372*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
12373*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
12374*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12375*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12376*4bdc9457SAndroid Build Coastguard Worker }
12377*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,batched_grouped_2x2s2_without_bias)12378*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, batched_grouped_2x2s2_without_bias) {
12379*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12380*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12381*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
12382*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12383*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12384*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12385*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12386*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12387*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
12388*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
12389*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12390*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12391*4bdc9457SAndroid Build Coastguard Worker }
12392*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F16,weights_cache_batched_grouped_2x2s2)12393*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, weights_cache_batched_grouped_2x2s2) {
12394*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12395*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12396*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12397*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12398*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12399*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12400*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12401*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
12402*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f16.gemm.nr * 2 + 3)
12403*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
12404*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12405*4bdc9457SAndroid Build Coastguard Worker .TestF16();
12406*4bdc9457SAndroid Build Coastguard Worker }
12407*4bdc9457SAndroid Build Coastguard Worker
12408*4bdc9457SAndroid Build Coastguard Worker // /**************************** SUBCONV2D/GEMM path, setup ****************************/
12409*4bdc9457SAndroid Build Coastguard Worker
12410*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 2x2s2_setup_changing_batch) {
12411*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12412*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12413*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12414*4bdc9457SAndroid Build Coastguard Worker .next_batch_size(5)
12415*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12416*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12417*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12418*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12419*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
12420*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
12421*4bdc9457SAndroid Build Coastguard Worker .TestSetupF16();
12422*4bdc9457SAndroid Build Coastguard Worker }
12423*4bdc9457SAndroid Build Coastguard Worker
12424*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 2x2s2_setup_changing_height) {
12425*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12426*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12427*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12428*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12429*4bdc9457SAndroid Build Coastguard Worker .next_input_height(kStridedInputHeight + 3)
12430*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12431*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12432*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12433*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
12434*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
12435*4bdc9457SAndroid Build Coastguard Worker .TestSetupF16();
12436*4bdc9457SAndroid Build Coastguard Worker }
12437*4bdc9457SAndroid Build Coastguard Worker
12438*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F16, 2x2s2_setup_changing_width) {
12439*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12440*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12441*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12442*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
12443*4bdc9457SAndroid Build Coastguard Worker .next_input_width(kStridedInputWidth + 3)
12444*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
12445*4bdc9457SAndroid Build Coastguard Worker .stride(2)
12446*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12447*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
12448*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
12449*4bdc9457SAndroid Build Coastguard Worker .TestSetupF16();
12450*4bdc9457SAndroid Build Coastguard Worker }
12451*4bdc9457SAndroid Build Coastguard Worker
12452*4bdc9457SAndroid Build Coastguard Worker /**************************** Future GEMM path ****************************/
12453*4bdc9457SAndroid Build Coastguard Worker
12454*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 1x1) {
12455*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12456*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12457*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12458*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12459*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12460*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12461*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12462*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12463*4bdc9457SAndroid Build Coastguard Worker }
12464*4bdc9457SAndroid Build Coastguard Worker
12465*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 1x1_varying_input_width) {
12466*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12467*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
12468*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12469*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
12470*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12471*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12472*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12473*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12474*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12475*4bdc9457SAndroid Build Coastguard Worker }
12476*4bdc9457SAndroid Build Coastguard Worker }
12477*4bdc9457SAndroid Build Coastguard Worker
12478*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 1x1_varying_input_height) {
12479*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12480*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
12481*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12482*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
12483*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12484*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12485*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12486*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12487*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12488*4bdc9457SAndroid Build Coastguard Worker }
12489*4bdc9457SAndroid Build Coastguard Worker }
12490*4bdc9457SAndroid Build Coastguard Worker
12491*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 1x1_varying_input_channels) {
12492*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12493*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
12494*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12495*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12496*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12497*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
12498*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12499*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12500*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12501*4bdc9457SAndroid Build Coastguard Worker }
12502*4bdc9457SAndroid Build Coastguard Worker }
12503*4bdc9457SAndroid Build Coastguard Worker
12504*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 1x1_varying_output_channels) {
12505*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12506*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f32.gemm.nr * 2; output_channels *= 2) {
12507*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12508*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12509*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12510*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12511*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
12512*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12513*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12514*4bdc9457SAndroid Build Coastguard Worker }
12515*4bdc9457SAndroid Build Coastguard Worker }
12516*4bdc9457SAndroid Build Coastguard Worker
12517*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 1x1_with_input_stride) {
12518*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12519*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12520*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12521*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12522*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12523*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12524*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
12525*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12526*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12527*4bdc9457SAndroid Build Coastguard Worker }
12528*4bdc9457SAndroid Build Coastguard Worker
12529*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 1x1_with_output_stride) {
12530*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12531*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12532*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12533*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12534*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12535*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12536*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f32.gemm.nr * 2 + 13)
12537*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12538*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12539*4bdc9457SAndroid Build Coastguard Worker }
12540*4bdc9457SAndroid Build Coastguard Worker
12541*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 1x1_with_qmin) {
12542*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12543*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12544*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12545*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12546*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12547*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12548*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
12549*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12550*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12551*4bdc9457SAndroid Build Coastguard Worker }
12552*4bdc9457SAndroid Build Coastguard Worker
12553*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 1x1_with_qmax) {
12554*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12555*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12556*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12557*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12558*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12559*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12560*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
12561*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12562*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12563*4bdc9457SAndroid Build Coastguard Worker }
12564*4bdc9457SAndroid Build Coastguard Worker
12565*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 1x1_without_bias) {
12566*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12567*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12568*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
12569*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12570*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12571*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12572*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12573*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12574*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12575*4bdc9457SAndroid Build Coastguard Worker }
12576*4bdc9457SAndroid Build Coastguard Worker
12577*4bdc9457SAndroid Build Coastguard Worker /**************************** Future GEMM path, grouped ****************************/
12578*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_1x1)12579*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_1x1) {
12580*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12581*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12582*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12583*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12584*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12585*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12586*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12587*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12588*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12589*4bdc9457SAndroid Build Coastguard Worker }
12590*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_1x1_varying_input_width)12591*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_1x1_varying_input_width) {
12592*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12593*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
12594*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12595*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
12596*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12597*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12598*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12599*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12600*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12601*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12602*4bdc9457SAndroid Build Coastguard Worker }
12603*4bdc9457SAndroid Build Coastguard Worker }
12604*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_1x1_varying_input_height)12605*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_1x1_varying_input_height) {
12606*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12607*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
12608*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12609*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
12610*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12611*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12612*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12613*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12614*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12615*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12616*4bdc9457SAndroid Build Coastguard Worker }
12617*4bdc9457SAndroid Build Coastguard Worker }
12618*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_1x1_varying_input_channels)12619*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_1x1_varying_input_channels) {
12620*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12621*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
12622*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12623*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12624*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12625*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12626*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
12627*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12628*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12629*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12630*4bdc9457SAndroid Build Coastguard Worker }
12631*4bdc9457SAndroid Build Coastguard Worker }
12632*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_1x1_varying_output_channels)12633*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_1x1_varying_output_channels) {
12634*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12635*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f32.gemm.nr * 2; output_channels *= 2) {
12636*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12637*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12638*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12639*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12640*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12641*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
12642*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12643*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12644*4bdc9457SAndroid Build Coastguard Worker }
12645*4bdc9457SAndroid Build Coastguard Worker }
12646*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_1x1_with_input_stride)12647*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_1x1_with_input_stride) {
12648*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12649*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12650*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12651*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12652*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12653*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12654*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12655*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(47)
12656*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12657*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12658*4bdc9457SAndroid Build Coastguard Worker }
12659*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_1x1_with_output_stride)12660*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_1x1_with_output_stride) {
12661*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12662*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12663*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12664*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12665*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12666*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12667*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr + 3)
12668*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f32.gemm.nr * 2 + 13)
12669*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12670*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12671*4bdc9457SAndroid Build Coastguard Worker }
12672*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_1x1_with_qmin)12673*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_1x1_with_qmin) {
12674*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12675*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12676*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12677*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12678*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12679*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12680*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12681*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
12682*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12683*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12684*4bdc9457SAndroid Build Coastguard Worker }
12685*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_1x1_with_qmax)12686*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_1x1_with_qmax) {
12687*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12688*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12689*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12690*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12691*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12692*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12693*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12694*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
12695*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12696*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12697*4bdc9457SAndroid Build Coastguard Worker }
12698*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_1x1_without_bias)12699*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_1x1_without_bias) {
12700*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12701*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12702*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
12703*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12704*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12705*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12706*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12707*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12708*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12709*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12710*4bdc9457SAndroid Build Coastguard Worker }
12711*4bdc9457SAndroid Build Coastguard Worker
12712*4bdc9457SAndroid Build Coastguard Worker /**************************** Future GEMM path, batched ****************************/
12713*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_1x1)12714*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_1x1) {
12715*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12716*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12717*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12718*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12719*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12720*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12721*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12722*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12723*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12724*4bdc9457SAndroid Build Coastguard Worker }
12725*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_1x1_varying_input_width)12726*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_1x1_varying_input_width) {
12727*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12728*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
12729*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12730*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12731*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
12732*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12733*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12734*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12735*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12736*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12737*4bdc9457SAndroid Build Coastguard Worker }
12738*4bdc9457SAndroid Build Coastguard Worker }
12739*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_1x1_varying_input_height)12740*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_1x1_varying_input_height) {
12741*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12742*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
12743*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12744*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12745*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
12746*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12747*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12748*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12749*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12750*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12751*4bdc9457SAndroid Build Coastguard Worker }
12752*4bdc9457SAndroid Build Coastguard Worker }
12753*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_1x1_varying_input_channels)12754*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_1x1_varying_input_channels) {
12755*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12756*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
12757*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12758*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12759*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12760*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12761*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
12762*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12763*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12764*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12765*4bdc9457SAndroid Build Coastguard Worker }
12766*4bdc9457SAndroid Build Coastguard Worker }
12767*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_1x1_varying_output_channels)12768*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_1x1_varying_output_channels) {
12769*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12770*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f32.gemm.nr * 2; output_channels *= 2) {
12771*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12772*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12773*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12774*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12775*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12776*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
12777*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12778*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12779*4bdc9457SAndroid Build Coastguard Worker }
12780*4bdc9457SAndroid Build Coastguard Worker }
12781*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_1x1_with_input_stride)12782*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_1x1_with_input_stride) {
12783*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12784*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12785*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12786*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12787*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12788*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12789*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12790*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
12791*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12792*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12793*4bdc9457SAndroid Build Coastguard Worker }
12794*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_1x1_with_output_stride)12795*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_1x1_with_output_stride) {
12796*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12797*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12798*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12799*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12800*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12801*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12802*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12803*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f32.gemm.nr * 2 + 13)
12804*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12805*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12806*4bdc9457SAndroid Build Coastguard Worker }
12807*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_1x1_with_qmin)12808*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_1x1_with_qmin) {
12809*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12810*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12811*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12812*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12813*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12814*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12815*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12816*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
12817*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12818*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12819*4bdc9457SAndroid Build Coastguard Worker }
12820*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_1x1_with_qmax)12821*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_1x1_with_qmax) {
12822*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12823*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12824*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12825*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12826*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12827*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12828*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12829*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
12830*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12831*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12832*4bdc9457SAndroid Build Coastguard Worker }
12833*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_1x1_without_bias)12834*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_1x1_without_bias) {
12835*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12836*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12837*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
12838*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12839*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12840*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12841*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12842*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12843*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12844*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12845*4bdc9457SAndroid Build Coastguard Worker }
12846*4bdc9457SAndroid Build Coastguard Worker
12847*4bdc9457SAndroid Build Coastguard Worker /**************************** Future GEMM path, batched, grouped ****************************/
12848*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_1x1)12849*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_1x1) {
12850*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12851*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12852*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12853*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12854*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12855*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12856*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12857*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12858*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12859*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12860*4bdc9457SAndroid Build Coastguard Worker }
12861*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_1x1_varying_input_width)12862*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_1x1_varying_input_width) {
12863*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12864*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
12865*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12866*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12867*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
12868*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12869*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12870*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12871*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12872*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12873*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12874*4bdc9457SAndroid Build Coastguard Worker }
12875*4bdc9457SAndroid Build Coastguard Worker }
12876*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_1x1_varying_input_height)12877*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_1x1_varying_input_height) {
12878*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12879*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
12880*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12881*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12882*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
12883*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12884*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12885*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12886*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12887*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12888*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12889*4bdc9457SAndroid Build Coastguard Worker }
12890*4bdc9457SAndroid Build Coastguard Worker }
12891*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_1x1_varying_input_channels)12892*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_1x1_varying_input_channels) {
12893*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12894*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
12895*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12896*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12897*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12898*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12899*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12900*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
12901*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12902*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12903*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12904*4bdc9457SAndroid Build Coastguard Worker }
12905*4bdc9457SAndroid Build Coastguard Worker }
12906*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_1x1_varying_output_channels)12907*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_1x1_varying_output_channels) {
12908*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12909*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f32.gemm.nr * 2; output_channels *= 2) {
12910*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12911*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12912*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12913*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12914*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12915*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12916*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
12917*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
12918*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12919*4bdc9457SAndroid Build Coastguard Worker }
12920*4bdc9457SAndroid Build Coastguard Worker }
12921*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_1x1_with_input_stride)12922*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_1x1_with_input_stride) {
12923*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12924*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12925*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12926*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12927*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12928*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12929*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12930*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12931*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(47)
12932*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12933*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12934*4bdc9457SAndroid Build Coastguard Worker }
12935*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_1x1_with_output_stride)12936*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_1x1_with_output_stride) {
12937*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12938*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12939*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12940*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12941*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12942*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12943*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12944*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr + 3)
12945*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f32.gemm.nr * 2 + 13)
12946*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12947*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12948*4bdc9457SAndroid Build Coastguard Worker }
12949*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_1x1_with_qmin)12950*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_1x1_with_qmin) {
12951*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12952*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12953*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12954*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12955*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12956*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12957*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12958*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12959*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
12960*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12961*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12962*4bdc9457SAndroid Build Coastguard Worker }
12963*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_1x1_with_qmax)12964*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_1x1_with_qmax) {
12965*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12966*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12967*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12968*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12969*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12970*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12971*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12972*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12973*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
12974*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12975*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12976*4bdc9457SAndroid Build Coastguard Worker }
12977*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_1x1_without_bias)12978*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_1x1_without_bias) {
12979*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12980*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12981*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
12982*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
12983*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12984*4bdc9457SAndroid Build Coastguard Worker .kernel_size(1, 1)
12985*4bdc9457SAndroid Build Coastguard Worker .groups(2)
12986*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
12987*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
12988*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
12989*4bdc9457SAndroid Build Coastguard Worker .TestF32();
12990*4bdc9457SAndroid Build Coastguard Worker }
12991*4bdc9457SAndroid Build Coastguard Worker
12992*4bdc9457SAndroid Build Coastguard Worker /**************************** CONV path ****************************/
12993*4bdc9457SAndroid Build Coastguard Worker
12994*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3) {
12995*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
12996*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
12997*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
12998*4bdc9457SAndroid Build Coastguard Worker .padding(1)
12999*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13000*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
13001*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13002*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13003*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13004*4bdc9457SAndroid Build Coastguard Worker }
13005*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,Kx3)13006*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, Kx3) {
13007*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13008*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 1; kernel_height <= 4; kernel_height *= 2) {
13009*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13010*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13011*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
13012*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
13013*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
13014*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13015*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13016*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13017*4bdc9457SAndroid Build Coastguard Worker }
13018*4bdc9457SAndroid Build Coastguard Worker }
13019*4bdc9457SAndroid Build Coastguard Worker
13020*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3xK) {
13021*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13022*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 1; kernel_width <= 4; kernel_width *= 2) {
13023*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13024*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13025*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
13026*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
13027*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
13028*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13029*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13030*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13031*4bdc9457SAndroid Build Coastguard Worker }
13032*4bdc9457SAndroid Build Coastguard Worker }
13033*4bdc9457SAndroid Build Coastguard Worker
13034*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3_varying_height_padding) {
13035*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13036*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
13037*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
13038*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13039*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13040*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
13041*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
13042*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
13043*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13044*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
13045*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13046*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
13047*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13048*4bdc9457SAndroid Build Coastguard Worker }
13049*4bdc9457SAndroid Build Coastguard Worker }
13050*4bdc9457SAndroid Build Coastguard Worker }
13051*4bdc9457SAndroid Build Coastguard Worker
13052*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3_varying_width_padding) {
13053*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13054*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
13055*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
13056*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13057*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13058*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
13059*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
13060*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
13061*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13062*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
13063*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13064*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
13065*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13066*4bdc9457SAndroid Build Coastguard Worker }
13067*4bdc9457SAndroid Build Coastguard Worker }
13068*4bdc9457SAndroid Build Coastguard Worker }
13069*4bdc9457SAndroid Build Coastguard Worker
13070*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3_varying_height_adjustment) {
13071*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13072*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 1; adjustment_height <= 2; adjustment_height++) {
13073*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13074*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13075*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13076*4bdc9457SAndroid Build Coastguard Worker .stride_height(adjustment_height + 1)
13077*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
13078*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13079*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
13080*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13081*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
13082*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13083*4bdc9457SAndroid Build Coastguard Worker }
13084*4bdc9457SAndroid Build Coastguard Worker }
13085*4bdc9457SAndroid Build Coastguard Worker
13086*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3_varying_width_adjustment) {
13087*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13088*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 1; adjustment_width <= 2; adjustment_width++) {
13089*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13090*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13091*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13092*4bdc9457SAndroid Build Coastguard Worker .stride_width(adjustment_width + 1)
13093*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
13094*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13095*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
13096*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13097*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
13098*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13099*4bdc9457SAndroid Build Coastguard Worker }
13100*4bdc9457SAndroid Build Coastguard Worker }
13101*4bdc9457SAndroid Build Coastguard Worker
13102*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3_varying_input_height) {
13103*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13104*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
13105*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13106*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
13107*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13108*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13109*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
13110*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13111*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
13112*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13113*4bdc9457SAndroid Build Coastguard Worker }
13114*4bdc9457SAndroid Build Coastguard Worker }
13115*4bdc9457SAndroid Build Coastguard Worker
13116*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3_varying_input_width) {
13117*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13118*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
13119*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13120*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
13121*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13122*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13123*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
13124*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13125*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
13126*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13127*4bdc9457SAndroid Build Coastguard Worker }
13128*4bdc9457SAndroid Build Coastguard Worker }
13129*4bdc9457SAndroid Build Coastguard Worker
13130*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3_varying_input_channels) {
13131*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13132*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
13133*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13134*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13135*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13136*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13137*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
13138*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13139*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
13140*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13141*4bdc9457SAndroid Build Coastguard Worker }
13142*4bdc9457SAndroid Build Coastguard Worker }
13143*4bdc9457SAndroid Build Coastguard Worker
13144*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3_varying_output_channels) {
13145*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13146*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f32.gemm.nr * 2; output_channels *= 2) {
13147*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13148*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13149*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13150*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13151*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13152*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
13153*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
13154*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13155*4bdc9457SAndroid Build Coastguard Worker }
13156*4bdc9457SAndroid Build Coastguard Worker }
13157*4bdc9457SAndroid Build Coastguard Worker
13158*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3_with_height_dilation) {
13159*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13160*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_height = 2; dilation_height <= 3; dilation_height++) {
13161*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13162*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13163*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13164*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13165*4bdc9457SAndroid Build Coastguard Worker .dilation_height(dilation_height)
13166*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13167*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13168*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13169*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13170*4bdc9457SAndroid Build Coastguard Worker }
13171*4bdc9457SAndroid Build Coastguard Worker }
13172*4bdc9457SAndroid Build Coastguard Worker
13173*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3_with_width_dilation) {
13174*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13175*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) {
13176*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13177*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13178*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13179*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13180*4bdc9457SAndroid Build Coastguard Worker .dilation_width(dilation_width)
13181*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13182*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13183*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13184*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13185*4bdc9457SAndroid Build Coastguard Worker }
13186*4bdc9457SAndroid Build Coastguard Worker }
13187*4bdc9457SAndroid Build Coastguard Worker
13188*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3_with_height_dilation_and_stride) {
13189*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13190*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13191*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13192*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13193*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13194*4bdc9457SAndroid Build Coastguard Worker .dilation_height(3)
13195*4bdc9457SAndroid Build Coastguard Worker .stride_height(2)
13196*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13197*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13198*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13199*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13200*4bdc9457SAndroid Build Coastguard Worker }
13201*4bdc9457SAndroid Build Coastguard Worker
13202*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3_with_width_dilation_and_stride) {
13203*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13204*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13205*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13206*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13207*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13208*4bdc9457SAndroid Build Coastguard Worker .dilation_width(3)
13209*4bdc9457SAndroid Build Coastguard Worker .stride_width(2)
13210*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13211*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13212*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13213*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13214*4bdc9457SAndroid Build Coastguard Worker }
13215*4bdc9457SAndroid Build Coastguard Worker
13216*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3_with_input_stride) {
13217*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13218*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13219*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13220*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13221*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13222*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13223*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13224*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
13225*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13226*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13227*4bdc9457SAndroid Build Coastguard Worker }
13228*4bdc9457SAndroid Build Coastguard Worker
13229*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3_with_output_stride) {
13230*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13231*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13232*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13233*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13234*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13235*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13236*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13237*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f32.gemm.nr * 2 + 13)
13238*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13239*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13240*4bdc9457SAndroid Build Coastguard Worker }
13241*4bdc9457SAndroid Build Coastguard Worker
13242*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3_with_qmin) {
13243*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13244*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13245*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13246*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13247*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13248*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13249*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13250*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
13251*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13252*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13253*4bdc9457SAndroid Build Coastguard Worker }
13254*4bdc9457SAndroid Build Coastguard Worker
13255*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3_with_qmax) {
13256*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13257*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13258*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13259*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13260*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13261*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13262*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13263*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
13264*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13265*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13266*4bdc9457SAndroid Build Coastguard Worker }
13267*4bdc9457SAndroid Build Coastguard Worker
13268*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3_without_bias) {
13269*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13270*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13271*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
13272*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13273*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13274*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13275*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13276*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13277*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13278*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13279*4bdc9457SAndroid Build Coastguard Worker }
13280*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,weights_cache_3x3)13281*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, weights_cache_3x3) {
13282*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13283*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13284*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13285*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13286*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13287*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
13288*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13289*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
13290*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13291*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13292*4bdc9457SAndroid Build Coastguard Worker }
13293*4bdc9457SAndroid Build Coastguard Worker
13294*4bdc9457SAndroid Build Coastguard Worker /**************************** CONV path, grouped ****************************/
13295*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3)13296*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3) {
13297*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13298*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13299*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13300*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13301*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13302*4bdc9457SAndroid Build Coastguard Worker .groups(2)
13303*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
13304*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13305*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13306*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13307*4bdc9457SAndroid Build Coastguard Worker }
13308*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_Kx3)13309*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_Kx3) {
13310*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13311*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 1; kernel_height <= 4; kernel_height *= 2) {
13312*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13313*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13314*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
13315*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
13316*4bdc9457SAndroid Build Coastguard Worker .groups(2)
13317*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
13318*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13319*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13320*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13321*4bdc9457SAndroid Build Coastguard Worker }
13322*4bdc9457SAndroid Build Coastguard Worker }
13323*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3xK)13324*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3xK) {
13325*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13326*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 1; kernel_width <= 4; kernel_width *= 2) {
13327*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13328*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13329*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
13330*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
13331*4bdc9457SAndroid Build Coastguard Worker .groups(2)
13332*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
13333*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13334*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13335*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13336*4bdc9457SAndroid Build Coastguard Worker }
13337*4bdc9457SAndroid Build Coastguard Worker }
13338*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3_varying_height_padding)13339*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3_varying_height_padding) {
13340*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13341*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
13342*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
13343*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13344*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13345*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
13346*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
13347*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
13348*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13349*4bdc9457SAndroid Build Coastguard Worker .groups(2)
13350*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
13351*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13352*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
13353*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13354*4bdc9457SAndroid Build Coastguard Worker }
13355*4bdc9457SAndroid Build Coastguard Worker }
13356*4bdc9457SAndroid Build Coastguard Worker }
13357*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3_varying_width_padding)13358*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3_varying_width_padding) {
13359*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13360*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
13361*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
13362*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13363*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13364*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
13365*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
13366*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
13367*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13368*4bdc9457SAndroid Build Coastguard Worker .groups(2)
13369*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
13370*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13371*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
13372*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13373*4bdc9457SAndroid Build Coastguard Worker }
13374*4bdc9457SAndroid Build Coastguard Worker }
13375*4bdc9457SAndroid Build Coastguard Worker }
13376*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3_varying_height_adjustment)13377*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3_varying_height_adjustment) {
13378*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13379*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 1; adjustment_height <= 2; adjustment_height++) {
13380*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13381*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13382*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13383*4bdc9457SAndroid Build Coastguard Worker .stride_height(adjustment_height + 1)
13384*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
13385*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13386*4bdc9457SAndroid Build Coastguard Worker .groups(2)
13387*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
13388*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13389*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
13390*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13391*4bdc9457SAndroid Build Coastguard Worker }
13392*4bdc9457SAndroid Build Coastguard Worker }
13393*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3_varying_width_adjustment)13394*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3_varying_width_adjustment) {
13395*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13396*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 1; adjustment_width <= 2; adjustment_width++) {
13397*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13398*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13399*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13400*4bdc9457SAndroid Build Coastguard Worker .stride_width(adjustment_width + 1)
13401*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
13402*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13403*4bdc9457SAndroid Build Coastguard Worker .groups(2)
13404*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
13405*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13406*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
13407*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13408*4bdc9457SAndroid Build Coastguard Worker }
13409*4bdc9457SAndroid Build Coastguard Worker }
13410*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3_varying_input_height)13411*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3_varying_input_height) {
13412*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13413*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
13414*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13415*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
13416*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13417*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13418*4bdc9457SAndroid Build Coastguard Worker .groups(2)
13419*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
13420*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13421*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
13422*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13423*4bdc9457SAndroid Build Coastguard Worker }
13424*4bdc9457SAndroid Build Coastguard Worker }
13425*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3_varying_input_width)13426*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3_varying_input_width) {
13427*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13428*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
13429*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13430*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
13431*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13432*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13433*4bdc9457SAndroid Build Coastguard Worker .groups(2)
13434*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
13435*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13436*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
13437*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13438*4bdc9457SAndroid Build Coastguard Worker }
13439*4bdc9457SAndroid Build Coastguard Worker }
13440*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3_varying_input_channels)13441*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3_varying_input_channels) {
13442*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13443*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
13444*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13445*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13446*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13447*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13448*4bdc9457SAndroid Build Coastguard Worker .groups(2)
13449*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
13450*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13451*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
13452*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13453*4bdc9457SAndroid Build Coastguard Worker }
13454*4bdc9457SAndroid Build Coastguard Worker }
13455*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3_varying_output_channels)13456*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3_varying_output_channels) {
13457*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13458*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f32.gemm.nr * 2; output_channels *= 2) {
13459*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13460*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13461*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13462*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13463*4bdc9457SAndroid Build Coastguard Worker .groups(2)
13464*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13465*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
13466*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
13467*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13468*4bdc9457SAndroid Build Coastguard Worker }
13469*4bdc9457SAndroid Build Coastguard Worker }
13470*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3_with_height_dilation)13471*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3_with_height_dilation) {
13472*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13473*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_height = 2; dilation_height <= 3; dilation_height++) {
13474*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13475*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13476*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13477*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13478*4bdc9457SAndroid Build Coastguard Worker .dilation_height(dilation_height)
13479*4bdc9457SAndroid Build Coastguard Worker .groups(2)
13480*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13481*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13482*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13483*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13484*4bdc9457SAndroid Build Coastguard Worker }
13485*4bdc9457SAndroid Build Coastguard Worker }
13486*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3_with_width_dilation)13487*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3_with_width_dilation) {
13488*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13489*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) {
13490*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13491*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13492*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13493*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13494*4bdc9457SAndroid Build Coastguard Worker .dilation_width(dilation_width)
13495*4bdc9457SAndroid Build Coastguard Worker .groups(2)
13496*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13497*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13498*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13499*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13500*4bdc9457SAndroid Build Coastguard Worker }
13501*4bdc9457SAndroid Build Coastguard Worker }
13502*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3_with_height_dilation_and_stride)13503*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3_with_height_dilation_and_stride) {
13504*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13505*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13506*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13507*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13508*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13509*4bdc9457SAndroid Build Coastguard Worker .dilation_height(3)
13510*4bdc9457SAndroid Build Coastguard Worker .stride_height(2)
13511*4bdc9457SAndroid Build Coastguard Worker .groups(2)
13512*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13513*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13514*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13515*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13516*4bdc9457SAndroid Build Coastguard Worker }
13517*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3_with_width_dilation_and_stride)13518*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3_with_width_dilation_and_stride) {
13519*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13520*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13521*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13522*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13523*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13524*4bdc9457SAndroid Build Coastguard Worker .dilation_width(3)
13525*4bdc9457SAndroid Build Coastguard Worker .stride_width(2)
13526*4bdc9457SAndroid Build Coastguard Worker .groups(2)
13527*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13528*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13529*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13530*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13531*4bdc9457SAndroid Build Coastguard Worker }
13532*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3_with_input_stride)13533*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3_with_input_stride) {
13534*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13535*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13536*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13537*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13538*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13539*4bdc9457SAndroid Build Coastguard Worker .groups(2)
13540*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13541*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13542*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(47)
13543*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13544*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13545*4bdc9457SAndroid Build Coastguard Worker }
13546*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3_with_output_stride)13547*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3_with_output_stride) {
13548*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13549*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13550*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13551*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13552*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13553*4bdc9457SAndroid Build Coastguard Worker .groups(2)
13554*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13555*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr + 3)
13556*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f32.gemm.nr * 2 + 13)
13557*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13558*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13559*4bdc9457SAndroid Build Coastguard Worker }
13560*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3_with_qmin)13561*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3_with_qmin) {
13562*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13563*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13564*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13565*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13566*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13567*4bdc9457SAndroid Build Coastguard Worker .groups(2)
13568*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13569*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13570*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
13571*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13572*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13573*4bdc9457SAndroid Build Coastguard Worker }
13574*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3_with_qmax)13575*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3_with_qmax) {
13576*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13577*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13578*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13579*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13580*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13581*4bdc9457SAndroid Build Coastguard Worker .groups(2)
13582*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13583*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13584*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
13585*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13586*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13587*4bdc9457SAndroid Build Coastguard Worker }
13588*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3_without_bias)13589*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3_without_bias) {
13590*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13591*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13592*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
13593*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13594*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13595*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13596*4bdc9457SAndroid Build Coastguard Worker .groups(2)
13597*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13598*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13599*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13600*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13601*4bdc9457SAndroid Build Coastguard Worker }
13602*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,weights_cache_grouped_3x3)13603*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, weights_cache_grouped_3x3) {
13604*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13605*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13606*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13607*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13608*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13609*4bdc9457SAndroid Build Coastguard Worker .groups(2)
13610*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
13611*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13612*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
13613*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13614*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13615*4bdc9457SAndroid Build Coastguard Worker }
13616*4bdc9457SAndroid Build Coastguard Worker
13617*4bdc9457SAndroid Build Coastguard Worker /**************************** CONV path, batched ****************************/
13618*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3)13619*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3) {
13620*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13621*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13622*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
13623*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13624*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13625*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13626*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
13627*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13628*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13629*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13630*4bdc9457SAndroid Build Coastguard Worker }
13631*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_Kx3)13632*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_Kx3) {
13633*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13634*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 1; kernel_height <= 4; kernel_height *= 2) {
13635*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13636*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
13637*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13638*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
13639*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
13640*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
13641*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13642*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13643*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13644*4bdc9457SAndroid Build Coastguard Worker }
13645*4bdc9457SAndroid Build Coastguard Worker }
13646*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3xK)13647*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3xK) {
13648*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13649*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 1; kernel_width <= 4; kernel_width *= 2) {
13650*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13651*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
13652*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13653*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
13654*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
13655*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
13656*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13657*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13658*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13659*4bdc9457SAndroid Build Coastguard Worker }
13660*4bdc9457SAndroid Build Coastguard Worker }
13661*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3_varying_height_padding)13662*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3_varying_height_padding) {
13663*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13664*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
13665*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
13666*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13667*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
13668*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13669*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
13670*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
13671*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
13672*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13673*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
13674*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13675*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
13676*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13677*4bdc9457SAndroid Build Coastguard Worker }
13678*4bdc9457SAndroid Build Coastguard Worker }
13679*4bdc9457SAndroid Build Coastguard Worker }
13680*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3_varying_width_padding)13681*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3_varying_width_padding) {
13682*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13683*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
13684*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
13685*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13686*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
13687*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13688*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
13689*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
13690*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
13691*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13692*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
13693*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13694*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
13695*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13696*4bdc9457SAndroid Build Coastguard Worker }
13697*4bdc9457SAndroid Build Coastguard Worker }
13698*4bdc9457SAndroid Build Coastguard Worker }
13699*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3_varying_height_adjustment)13700*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3_varying_height_adjustment) {
13701*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13702*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 1; adjustment_height <= 2; adjustment_height++) {
13703*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13704*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
13705*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13706*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13707*4bdc9457SAndroid Build Coastguard Worker .stride_height(adjustment_height + 1)
13708*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
13709*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13710*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
13711*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13712*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
13713*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13714*4bdc9457SAndroid Build Coastguard Worker }
13715*4bdc9457SAndroid Build Coastguard Worker }
13716*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3_varying_width_adjustment)13717*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3_varying_width_adjustment) {
13718*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13719*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 1; adjustment_width <= 2; adjustment_width++) {
13720*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13721*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
13722*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13723*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13724*4bdc9457SAndroid Build Coastguard Worker .stride_width(adjustment_width + 1)
13725*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
13726*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13727*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
13728*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13729*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
13730*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13731*4bdc9457SAndroid Build Coastguard Worker }
13732*4bdc9457SAndroid Build Coastguard Worker }
13733*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3_varying_input_height)13734*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3_varying_input_height) {
13735*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13736*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
13737*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13738*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
13739*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
13740*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13741*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13742*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
13743*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13744*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
13745*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13746*4bdc9457SAndroid Build Coastguard Worker }
13747*4bdc9457SAndroid Build Coastguard Worker }
13748*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3_varying_input_width)13749*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3_varying_input_width) {
13750*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13751*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
13752*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13753*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
13754*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
13755*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13756*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13757*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
13758*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13759*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
13760*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13761*4bdc9457SAndroid Build Coastguard Worker }
13762*4bdc9457SAndroid Build Coastguard Worker }
13763*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3_varying_input_channels)13764*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3_varying_input_channels) {
13765*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13766*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
13767*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13768*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
13769*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13770*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13771*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13772*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
13773*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13774*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
13775*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13776*4bdc9457SAndroid Build Coastguard Worker }
13777*4bdc9457SAndroid Build Coastguard Worker }
13778*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3_varying_output_channels)13779*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3_varying_output_channels) {
13780*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13781*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f32.gemm.nr * 2; output_channels *= 2) {
13782*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13783*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
13784*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13785*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13786*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13787*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13788*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
13789*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
13790*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13791*4bdc9457SAndroid Build Coastguard Worker }
13792*4bdc9457SAndroid Build Coastguard Worker }
13793*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3_with_height_dilation)13794*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3_with_height_dilation) {
13795*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13796*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_height = 2; dilation_height <= 3; dilation_height++) {
13797*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13798*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
13799*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13800*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13801*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13802*4bdc9457SAndroid Build Coastguard Worker .dilation_height(dilation_height)
13803*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13804*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13805*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13806*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13807*4bdc9457SAndroid Build Coastguard Worker }
13808*4bdc9457SAndroid Build Coastguard Worker }
13809*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3_with_width_dilation)13810*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3_with_width_dilation) {
13811*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13812*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) {
13813*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13814*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
13815*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13816*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13817*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13818*4bdc9457SAndroid Build Coastguard Worker .dilation_width(dilation_width)
13819*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13820*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13821*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13822*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13823*4bdc9457SAndroid Build Coastguard Worker }
13824*4bdc9457SAndroid Build Coastguard Worker }
13825*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3_with_height_dilation_and_stride)13826*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3_with_height_dilation_and_stride) {
13827*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13828*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13829*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
13830*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13831*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13832*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13833*4bdc9457SAndroid Build Coastguard Worker .dilation_height(3)
13834*4bdc9457SAndroid Build Coastguard Worker .stride_height(2)
13835*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13836*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13837*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13838*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13839*4bdc9457SAndroid Build Coastguard Worker }
13840*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3_with_width_dilation_and_stride)13841*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3_with_width_dilation_and_stride) {
13842*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13843*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13844*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
13845*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13846*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13847*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13848*4bdc9457SAndroid Build Coastguard Worker .dilation_width(3)
13849*4bdc9457SAndroid Build Coastguard Worker .stride_width(2)
13850*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13851*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13852*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13853*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13854*4bdc9457SAndroid Build Coastguard Worker }
13855*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3_with_input_stride)13856*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3_with_input_stride) {
13857*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13858*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13859*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
13860*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13861*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13862*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13863*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13864*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13865*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
13866*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13867*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13868*4bdc9457SAndroid Build Coastguard Worker }
13869*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3_with_output_stride)13870*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3_with_output_stride) {
13871*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13872*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13873*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
13874*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13875*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13876*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13877*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13878*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13879*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f32.gemm.nr * 2 + 13)
13880*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13881*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13882*4bdc9457SAndroid Build Coastguard Worker }
13883*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3_with_qmin)13884*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3_with_qmin) {
13885*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13886*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13887*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
13888*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13889*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13890*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13891*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13892*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13893*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
13894*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13895*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13896*4bdc9457SAndroid Build Coastguard Worker }
13897*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3_with_qmax)13898*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3_with_qmax) {
13899*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13900*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13901*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
13902*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13903*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13904*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13905*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13906*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13907*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
13908*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13909*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13910*4bdc9457SAndroid Build Coastguard Worker }
13911*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3_without_bias)13912*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3_without_bias) {
13913*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13914*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13915*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
13916*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
13917*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13918*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13919*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13920*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
13921*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13922*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13923*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13924*4bdc9457SAndroid Build Coastguard Worker }
13925*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,weights_cache_batched_3x3)13926*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, weights_cache_batched_3x3) {
13927*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13928*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13929*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
13930*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13931*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13932*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13933*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
13934*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13935*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
13936*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13937*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13938*4bdc9457SAndroid Build Coastguard Worker }
13939*4bdc9457SAndroid Build Coastguard Worker
13940*4bdc9457SAndroid Build Coastguard Worker
13941*4bdc9457SAndroid Build Coastguard Worker /**************************** CONV path, grouped, batched ****************************/
13942*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3)13943*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3) {
13944*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13945*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13946*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
13947*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13948*4bdc9457SAndroid Build Coastguard Worker .padding(1)
13949*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
13950*4bdc9457SAndroid Build Coastguard Worker .groups(2)
13951*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
13952*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13953*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13954*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13955*4bdc9457SAndroid Build Coastguard Worker }
13956*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_Kx3)13957*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_Kx3) {
13958*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13959*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 1; kernel_height <= 4; kernel_height *= 2) {
13960*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13961*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
13962*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13963*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
13964*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
13965*4bdc9457SAndroid Build Coastguard Worker .groups(2)
13966*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
13967*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13968*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13969*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13970*4bdc9457SAndroid Build Coastguard Worker }
13971*4bdc9457SAndroid Build Coastguard Worker }
13972*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3xK)13973*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3xK) {
13974*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13975*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 1; kernel_width <= 4; kernel_width *= 2) {
13976*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13977*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
13978*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13979*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
13980*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
13981*4bdc9457SAndroid Build Coastguard Worker .groups(2)
13982*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
13983*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
13984*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
13985*4bdc9457SAndroid Build Coastguard Worker .TestF32();
13986*4bdc9457SAndroid Build Coastguard Worker }
13987*4bdc9457SAndroid Build Coastguard Worker }
13988*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3_varying_height_padding)13989*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3_varying_height_padding) {
13990*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
13991*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
13992*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
13993*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
13994*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
13995*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
13996*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
13997*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
13998*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
13999*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14000*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14001*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
14002*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14003*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
14004*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14005*4bdc9457SAndroid Build Coastguard Worker }
14006*4bdc9457SAndroid Build Coastguard Worker }
14007*4bdc9457SAndroid Build Coastguard Worker }
14008*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3_varying_width_padding)14009*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3_varying_width_padding) {
14010*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14011*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
14012*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
14013*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14014*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
14015*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
14016*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
14017*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
14018*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
14019*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14020*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14021*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
14022*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14023*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
14024*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14025*4bdc9457SAndroid Build Coastguard Worker }
14026*4bdc9457SAndroid Build Coastguard Worker }
14027*4bdc9457SAndroid Build Coastguard Worker }
14028*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3_varying_height_adjustment)14029*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3_varying_height_adjustment) {
14030*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14031*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 1; adjustment_height <= 2; adjustment_height++) {
14032*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14033*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
14034*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
14035*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14036*4bdc9457SAndroid Build Coastguard Worker .stride_height(adjustment_height + 1)
14037*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
14038*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14039*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14040*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
14041*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14042*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
14043*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14044*4bdc9457SAndroid Build Coastguard Worker }
14045*4bdc9457SAndroid Build Coastguard Worker }
14046*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3_varying_width_adjustment)14047*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3_varying_width_adjustment) {
14048*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14049*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 1; adjustment_width <= 2; adjustment_width++) {
14050*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14051*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
14052*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
14053*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14054*4bdc9457SAndroid Build Coastguard Worker .stride_width(adjustment_width + 1)
14055*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
14056*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14057*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14058*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
14059*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14060*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
14061*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14062*4bdc9457SAndroid Build Coastguard Worker }
14063*4bdc9457SAndroid Build Coastguard Worker }
14064*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3_varying_input_height)14065*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3_varying_input_height) {
14066*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14067*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kUnstridedInputHeight - 2; input_height <= kUnstridedInputHeight + 2; input_height++) {
14068*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14069*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
14070*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kUnstridedInputWidth)
14071*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14072*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14073*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14074*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
14075*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14076*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
14077*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14078*4bdc9457SAndroid Build Coastguard Worker }
14079*4bdc9457SAndroid Build Coastguard Worker }
14080*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3_varying_input_width)14081*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3_varying_input_width) {
14082*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14083*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kUnstridedInputWidth - 2; input_width <= kUnstridedInputWidth + 2; input_width++) {
14084*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14085*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
14086*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, input_width)
14087*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14088*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14089*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14090*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
14091*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14092*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
14093*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14094*4bdc9457SAndroid Build Coastguard Worker }
14095*4bdc9457SAndroid Build Coastguard Worker }
14096*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3_varying_input_channels)14097*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3_varying_input_channels) {
14098*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14099*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
14100*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14101*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
14102*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
14103*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14104*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14105*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14106*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
14107*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14108*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
14109*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14110*4bdc9457SAndroid Build Coastguard Worker }
14111*4bdc9457SAndroid Build Coastguard Worker }
14112*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3_varying_output_channels)14113*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3_varying_output_channels) {
14114*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14115*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f32.gemm.nr * 2; output_channels *= 2) {
14116*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14117*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
14118*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
14119*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14120*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14121*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14122*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
14123*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
14124*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
14125*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14126*4bdc9457SAndroid Build Coastguard Worker }
14127*4bdc9457SAndroid Build Coastguard Worker }
14128*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3_with_height_dilation)14129*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3_with_height_dilation) {
14130*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14131*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_height = 2; dilation_height <= 3; dilation_height++) {
14132*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14133*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
14134*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
14135*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14136*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14137*4bdc9457SAndroid Build Coastguard Worker .dilation_height(dilation_height)
14138*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14139*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
14140*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14141*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14142*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14143*4bdc9457SAndroid Build Coastguard Worker }
14144*4bdc9457SAndroid Build Coastguard Worker }
14145*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3_with_width_dilation)14146*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3_with_width_dilation) {
14147*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14148*4bdc9457SAndroid Build Coastguard Worker for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) {
14149*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14150*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
14151*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
14152*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14153*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14154*4bdc9457SAndroid Build Coastguard Worker .dilation_width(dilation_width)
14155*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14156*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
14157*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14158*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14159*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14160*4bdc9457SAndroid Build Coastguard Worker }
14161*4bdc9457SAndroid Build Coastguard Worker }
14162*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3_with_height_dilation_and_stride)14163*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3_with_height_dilation_and_stride) {
14164*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14165*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14166*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
14167*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
14168*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14169*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14170*4bdc9457SAndroid Build Coastguard Worker .dilation_height(3)
14171*4bdc9457SAndroid Build Coastguard Worker .stride_width(2)
14172*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14173*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
14174*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14175*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14176*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14177*4bdc9457SAndroid Build Coastguard Worker }
14178*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3_with_width_dilation_and_stride)14179*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3_with_width_dilation_and_stride) {
14180*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14181*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14182*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
14183*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
14184*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14185*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14186*4bdc9457SAndroid Build Coastguard Worker .dilation_width(3)
14187*4bdc9457SAndroid Build Coastguard Worker .stride_width(2)
14188*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14189*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
14190*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14191*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14192*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14193*4bdc9457SAndroid Build Coastguard Worker }
14194*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3_with_input_stride)14195*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3_with_input_stride) {
14196*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14197*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14198*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
14199*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
14200*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14201*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14202*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14203*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
14204*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14205*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(47)
14206*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14207*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14208*4bdc9457SAndroid Build Coastguard Worker }
14209*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3_with_output_stride)14210*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3_with_output_stride) {
14211*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14212*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14213*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
14214*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
14215*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14216*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14217*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14218*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
14219*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr + 3)
14220*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f32.gemm.nr * 2 + 13)
14221*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14222*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14223*4bdc9457SAndroid Build Coastguard Worker }
14224*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3_with_qmin)14225*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3_with_qmin) {
14226*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14227*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14228*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
14229*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
14230*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14231*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14232*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14233*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
14234*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14235*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
14236*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14237*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14238*4bdc9457SAndroid Build Coastguard Worker }
14239*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3_with_qmax)14240*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3_with_qmax) {
14241*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14242*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14243*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
14244*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
14245*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14246*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14247*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14248*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
14249*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14250*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
14251*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14252*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14253*4bdc9457SAndroid Build Coastguard Worker }
14254*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3_without_bias)14255*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3_without_bias) {
14256*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14257*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14258*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
14259*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
14260*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
14261*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14262*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14263*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14264*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
14265*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14266*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14267*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14268*4bdc9457SAndroid Build Coastguard Worker }
14269*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,weights_cache_batched_grouped_3x3)14270*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, weights_cache_batched_grouped_3x3) {
14271*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14272*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14273*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
14274*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
14275*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14276*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14277*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14278*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
14279*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14280*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
14281*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14282*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14283*4bdc9457SAndroid Build Coastguard Worker }
14284*4bdc9457SAndroid Build Coastguard Worker
14285*4bdc9457SAndroid Build Coastguard Worker /**************************** CONV path, setup ****************************/
14286*4bdc9457SAndroid Build Coastguard Worker
14287*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3_setup_changing_batch) {
14288*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14289*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14290*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
14291*4bdc9457SAndroid Build Coastguard Worker .next_batch_size(5)
14292*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
14293*4bdc9457SAndroid Build Coastguard Worker .kernel_height(3)
14294*4bdc9457SAndroid Build Coastguard Worker .kernel_width(5)
14295*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14296*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
14297*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
14298*4bdc9457SAndroid Build Coastguard Worker .TestSetupF32();
14299*4bdc9457SAndroid Build Coastguard Worker }
14300*4bdc9457SAndroid Build Coastguard Worker
14301*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3_setup_changing_height) {
14302*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14303*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14304*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
14305*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
14306*4bdc9457SAndroid Build Coastguard Worker .next_input_height(kUnstridedInputHeight + 3)
14307*4bdc9457SAndroid Build Coastguard Worker .kernel_height(3)
14308*4bdc9457SAndroid Build Coastguard Worker .kernel_width(5)
14309*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14310*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
14311*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
14312*4bdc9457SAndroid Build Coastguard Worker .TestSetupF32();
14313*4bdc9457SAndroid Build Coastguard Worker }
14314*4bdc9457SAndroid Build Coastguard Worker
14315*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3_setup_changing_width) {
14316*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14317*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14318*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
14319*4bdc9457SAndroid Build Coastguard Worker .input_size(kUnstridedInputHeight, kUnstridedInputWidth)
14320*4bdc9457SAndroid Build Coastguard Worker .next_input_width(kUnstridedInputWidth + 3)
14321*4bdc9457SAndroid Build Coastguard Worker .kernel_height(3)
14322*4bdc9457SAndroid Build Coastguard Worker .kernel_width(5)
14323*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14324*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
14325*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
14326*4bdc9457SAndroid Build Coastguard Worker .TestSetupF32();
14327*4bdc9457SAndroid Build Coastguard Worker }
14328*4bdc9457SAndroid Build Coastguard Worker
14329*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/IGEMM path ****************************/
14330*4bdc9457SAndroid Build Coastguard Worker
14331*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3s2) {
14332*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14333*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14334*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14335*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14336*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14337*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14338*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
14339*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14340*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14341*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14342*4bdc9457SAndroid Build Coastguard Worker }
14343*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,Kx3s2)14344*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, Kx3s2) {
14345*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14346*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 2; kernel_height <= 5; kernel_height++) {
14347*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14348*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14349*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
14350*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
14351*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14352*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
14353*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14354*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14355*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14356*4bdc9457SAndroid Build Coastguard Worker }
14357*4bdc9457SAndroid Build Coastguard Worker }
14358*4bdc9457SAndroid Build Coastguard Worker
14359*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3xKs2) {
14360*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14361*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 2; kernel_width <= 5; kernel_width++) {
14362*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14363*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14364*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
14365*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
14366*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14367*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
14368*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14369*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14370*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14371*4bdc9457SAndroid Build Coastguard Worker }
14372*4bdc9457SAndroid Build Coastguard Worker }
14373*4bdc9457SAndroid Build Coastguard Worker
14374*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3sSx1) {
14375*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14376*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_height = 2; stride_height <= 3; stride_height++) {
14377*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14378*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14379*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14380*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
14381*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14382*4bdc9457SAndroid Build Coastguard Worker .stride_height(stride_height)
14383*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
14384*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14385*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14386*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14387*4bdc9457SAndroid Build Coastguard Worker }
14388*4bdc9457SAndroid Build Coastguard Worker }
14389*4bdc9457SAndroid Build Coastguard Worker
14390*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3s1xS) {
14391*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14392*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_width = 2; stride_width <= 3; stride_width++) {
14393*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14394*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14395*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14396*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
14397*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14398*4bdc9457SAndroid Build Coastguard Worker .stride_width(stride_width)
14399*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
14400*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14401*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14402*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14403*4bdc9457SAndroid Build Coastguard Worker }
14404*4bdc9457SAndroid Build Coastguard Worker }
14405*4bdc9457SAndroid Build Coastguard Worker
14406*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3s2_varying_height_padding) {
14407*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14408*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
14409*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
14410*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14411*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14412*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
14413*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
14414*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
14415*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14416*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14417*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
14418*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14419*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
14420*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14421*4bdc9457SAndroid Build Coastguard Worker }
14422*4bdc9457SAndroid Build Coastguard Worker }
14423*4bdc9457SAndroid Build Coastguard Worker }
14424*4bdc9457SAndroid Build Coastguard Worker
14425*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3s2_varying_width_padding) {
14426*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14427*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
14428*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
14429*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14430*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14431*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
14432*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
14433*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
14434*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14435*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14436*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
14437*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14438*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
14439*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14440*4bdc9457SAndroid Build Coastguard Worker }
14441*4bdc9457SAndroid Build Coastguard Worker }
14442*4bdc9457SAndroid Build Coastguard Worker }
14443*4bdc9457SAndroid Build Coastguard Worker
14444*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3s2_varying_height_adjustment) {
14445*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14446*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 0; adjustment_height <= 1; adjustment_height++) {
14447*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14448*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14449*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14450*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
14451*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14452*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14453*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
14454*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14455*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
14456*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14457*4bdc9457SAndroid Build Coastguard Worker }
14458*4bdc9457SAndroid Build Coastguard Worker }
14459*4bdc9457SAndroid Build Coastguard Worker
14460*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3s2_varying_width_adjustment) {
14461*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14462*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 0; adjustment_width <= 1; adjustment_width++) {
14463*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14464*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14465*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14466*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
14467*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14468*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14469*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
14470*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14471*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
14472*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14473*4bdc9457SAndroid Build Coastguard Worker }
14474*4bdc9457SAndroid Build Coastguard Worker }
14475*4bdc9457SAndroid Build Coastguard Worker
14476*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3s2_varying_input_height) {
14477*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14478*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
14479*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14480*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
14481*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14482*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14483*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14484*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
14485*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14486*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
14487*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14488*4bdc9457SAndroid Build Coastguard Worker }
14489*4bdc9457SAndroid Build Coastguard Worker }
14490*4bdc9457SAndroid Build Coastguard Worker
14491*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3s2_varying_input_width) {
14492*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14493*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
14494*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14495*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14496*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14497*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14498*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14499*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
14500*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14501*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
14502*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14503*4bdc9457SAndroid Build Coastguard Worker }
14504*4bdc9457SAndroid Build Coastguard Worker }
14505*4bdc9457SAndroid Build Coastguard Worker
14506*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3s2_varying_input_channels) {
14507*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14508*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
14509*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14510*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14511*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14512*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14513*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14514*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
14515*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14516*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
14517*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14518*4bdc9457SAndroid Build Coastguard Worker }
14519*4bdc9457SAndroid Build Coastguard Worker }
14520*4bdc9457SAndroid Build Coastguard Worker
14521*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3s2_varying_output_channels) {
14522*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14523*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f32.gemm.nr * 2; output_channels *= 2) {
14524*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14525*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14526*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14527*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14528*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14529*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
14530*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
14531*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
14532*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14533*4bdc9457SAndroid Build Coastguard Worker }
14534*4bdc9457SAndroid Build Coastguard Worker }
14535*4bdc9457SAndroid Build Coastguard Worker
14536*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3s2_with_input_stride) {
14537*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14538*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14539*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14540*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14541*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14542*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14543*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
14544*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14545*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
14546*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14547*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14548*4bdc9457SAndroid Build Coastguard Worker }
14549*4bdc9457SAndroid Build Coastguard Worker
14550*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3s2_with_output_stride) {
14551*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14552*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14553*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14554*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14555*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14556*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14557*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
14558*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14559*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f32.gemm.nr * 2 + 13)
14560*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14561*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14562*4bdc9457SAndroid Build Coastguard Worker }
14563*4bdc9457SAndroid Build Coastguard Worker
14564*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3s2_with_qmin) {
14565*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14566*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14567*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14568*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14569*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14570*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14571*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
14572*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14573*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
14574*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14575*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14576*4bdc9457SAndroid Build Coastguard Worker }
14577*4bdc9457SAndroid Build Coastguard Worker
14578*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3s2_with_qmax) {
14579*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14580*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14581*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14582*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14583*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14584*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14585*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
14586*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14587*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
14588*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14589*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14590*4bdc9457SAndroid Build Coastguard Worker }
14591*4bdc9457SAndroid Build Coastguard Worker
14592*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3s2_without_bias) {
14593*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14594*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14595*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
14596*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14597*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14598*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14599*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14600*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
14601*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14602*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14603*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14604*4bdc9457SAndroid Build Coastguard Worker }
14605*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,weights_cache_3x3s2)14606*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, weights_cache_3x3s2) {
14607*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14608*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14609*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14610*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14611*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14612*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14613*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
14614*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14615*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
14616*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14617*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14618*4bdc9457SAndroid Build Coastguard Worker }
14619*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,stress_weights_cache_5x5s4)14620*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, stress_weights_cache_5x5s4) {
14621*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14622*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14623*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14624*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14625*4bdc9457SAndroid Build Coastguard Worker .kernel_size(5, 5)
14626*4bdc9457SAndroid Build Coastguard Worker .stride(4)
14627*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
14628*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14629*4bdc9457SAndroid Build Coastguard Worker .iterations(60) // Higher number of iterations to write more weights.
14630*4bdc9457SAndroid Build Coastguard Worker .StressWeightsCacheTestF32();
14631*4bdc9457SAndroid Build Coastguard Worker }
14632*4bdc9457SAndroid Build Coastguard Worker
14633*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/IGEMM path, grouped ****************************/
14634*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3s2)14635*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3s2) {
14636*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14637*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14638*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14639*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14640*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14641*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14642*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14643*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
14644*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14645*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14646*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14647*4bdc9457SAndroid Build Coastguard Worker }
14648*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_Kx3s2)14649*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_Kx3s2) {
14650*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14651*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 2; kernel_height <= 5; kernel_height++) {
14652*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14653*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14654*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
14655*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
14656*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14657*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14658*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
14659*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14660*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14661*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14662*4bdc9457SAndroid Build Coastguard Worker }
14663*4bdc9457SAndroid Build Coastguard Worker }
14664*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3xKs2)14665*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3xKs2) {
14666*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14667*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 2; kernel_width <= 5; kernel_width++) {
14668*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14669*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14670*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
14671*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
14672*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14673*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14674*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
14675*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14676*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14677*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14678*4bdc9457SAndroid Build Coastguard Worker }
14679*4bdc9457SAndroid Build Coastguard Worker }
14680*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3sSx1)14681*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3sSx1) {
14682*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14683*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_height = 2; stride_height <= 3; stride_height++) {
14684*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14685*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14686*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14687*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
14688*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14689*4bdc9457SAndroid Build Coastguard Worker .stride_height(stride_height)
14690*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14691*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
14692*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14693*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14694*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14695*4bdc9457SAndroid Build Coastguard Worker }
14696*4bdc9457SAndroid Build Coastguard Worker }
14697*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3s1xS)14698*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3s1xS) {
14699*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14700*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_width = 2; stride_width <= 3; stride_width++) {
14701*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14702*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14703*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14704*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
14705*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14706*4bdc9457SAndroid Build Coastguard Worker .stride_width(stride_width)
14707*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14708*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
14709*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14710*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14711*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14712*4bdc9457SAndroid Build Coastguard Worker }
14713*4bdc9457SAndroid Build Coastguard Worker }
14714*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3s2_varying_height_padding)14715*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3s2_varying_height_padding) {
14716*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14717*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
14718*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
14719*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14720*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14721*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
14722*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
14723*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
14724*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14725*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14726*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14727*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
14728*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14729*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
14730*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14731*4bdc9457SAndroid Build Coastguard Worker }
14732*4bdc9457SAndroid Build Coastguard Worker }
14733*4bdc9457SAndroid Build Coastguard Worker }
14734*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3s2_varying_width_padding)14735*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3s2_varying_width_padding) {
14736*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14737*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
14738*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
14739*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14740*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14741*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
14742*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
14743*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
14744*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14745*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14746*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14747*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
14748*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14749*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
14750*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14751*4bdc9457SAndroid Build Coastguard Worker }
14752*4bdc9457SAndroid Build Coastguard Worker }
14753*4bdc9457SAndroid Build Coastguard Worker }
14754*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3s2_varying_height_adjustment)14755*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3s2_varying_height_adjustment) {
14756*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14757*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 0; adjustment_height <= 1; adjustment_height++) {
14758*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14759*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14760*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14761*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
14762*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14763*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14764*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14765*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
14766*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14767*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
14768*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14769*4bdc9457SAndroid Build Coastguard Worker }
14770*4bdc9457SAndroid Build Coastguard Worker }
14771*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3s2_varying_width_adjustment)14772*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3s2_varying_width_adjustment) {
14773*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14774*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 0; adjustment_width <= 1; adjustment_width++) {
14775*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14776*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14777*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14778*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
14779*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14780*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14781*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14782*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
14783*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14784*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
14785*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14786*4bdc9457SAndroid Build Coastguard Worker }
14787*4bdc9457SAndroid Build Coastguard Worker }
14788*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3s2_varying_input_height)14789*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3s2_varying_input_height) {
14790*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14791*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
14792*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14793*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
14794*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14795*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14796*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14797*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14798*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
14799*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14800*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
14801*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14802*4bdc9457SAndroid Build Coastguard Worker }
14803*4bdc9457SAndroid Build Coastguard Worker }
14804*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3s2_varying_input_width)14805*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3s2_varying_input_width) {
14806*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14807*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
14808*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14809*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14810*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14811*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14812*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14813*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14814*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
14815*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14816*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
14817*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14818*4bdc9457SAndroid Build Coastguard Worker }
14819*4bdc9457SAndroid Build Coastguard Worker }
14820*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3s2_varying_input_channels)14821*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3s2_varying_input_channels) {
14822*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14823*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 14; input_channels <= 20; input_channels++) {
14824*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14825*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14826*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14827*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14828*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14829*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14830*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
14831*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14832*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
14833*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14834*4bdc9457SAndroid Build Coastguard Worker }
14835*4bdc9457SAndroid Build Coastguard Worker }
14836*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3s2_varying_output_channels)14837*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3s2_varying_output_channels) {
14838*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14839*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f32.gemm.nr * 2; output_channels *= 2) {
14840*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14841*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14842*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14843*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14844*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14845*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14846*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
14847*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
14848*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
14849*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14850*4bdc9457SAndroid Build Coastguard Worker }
14851*4bdc9457SAndroid Build Coastguard Worker }
14852*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3s2_with_input_stride)14853*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3s2_with_input_stride) {
14854*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14855*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14856*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14857*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14858*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14859*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14860*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14861*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
14862*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14863*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(37)
14864*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14865*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14866*4bdc9457SAndroid Build Coastguard Worker }
14867*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3s2_with_output_stride)14868*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3s2_with_output_stride) {
14869*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14870*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14871*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14872*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14873*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14874*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14875*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14876*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
14877*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr + 3)
14878*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f32.gemm.nr * 2 + 13)
14879*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14880*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14881*4bdc9457SAndroid Build Coastguard Worker }
14882*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3s2_with_qmin)14883*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3s2_with_qmin) {
14884*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14885*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14886*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14887*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14888*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14889*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14890*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14891*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
14892*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14893*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
14894*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14895*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14896*4bdc9457SAndroid Build Coastguard Worker }
14897*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3s2_with_qmax)14898*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3s2_with_qmax) {
14899*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14900*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14901*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14902*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14903*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14904*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14905*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14906*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
14907*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14908*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
14909*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14910*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14911*4bdc9457SAndroid Build Coastguard Worker }
14912*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_3x3s2_without_bias)14913*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_3x3s2_without_bias) {
14914*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14915*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14916*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
14917*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14918*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14919*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14920*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14921*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14922*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
14923*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14924*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14925*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14926*4bdc9457SAndroid Build Coastguard Worker }
14927*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,weights_cache_grouped_3x3s2)14928*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, weights_cache_grouped_3x3s2) {
14929*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14930*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14931*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14932*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14933*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14934*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14935*4bdc9457SAndroid Build Coastguard Worker .groups(2)
14936*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
14937*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14938*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
14939*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14940*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14941*4bdc9457SAndroid Build Coastguard Worker }
14942*4bdc9457SAndroid Build Coastguard Worker
14943*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/IGEMM path, batched ****************************/
14944*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3s2)14945*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3s2) {
14946*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14947*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14948*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
14949*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14950*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14951*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
14952*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14953*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
14954*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14955*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14956*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14957*4bdc9457SAndroid Build Coastguard Worker }
14958*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_Kx3s2)14959*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_Kx3s2) {
14960*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14961*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 2; kernel_height <= 5; kernel_height++) {
14962*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14963*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
14964*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14965*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
14966*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
14967*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14968*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
14969*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14970*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14971*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14972*4bdc9457SAndroid Build Coastguard Worker }
14973*4bdc9457SAndroid Build Coastguard Worker }
14974*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3xKs2)14975*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3xKs2) {
14976*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14977*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 2; kernel_width <= 5; kernel_width++) {
14978*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14979*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
14980*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14981*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
14982*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
14983*4bdc9457SAndroid Build Coastguard Worker .stride(2)
14984*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
14985*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
14986*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
14987*4bdc9457SAndroid Build Coastguard Worker .TestF32();
14988*4bdc9457SAndroid Build Coastguard Worker }
14989*4bdc9457SAndroid Build Coastguard Worker }
14990*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3sSx1)14991*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3sSx1) {
14992*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
14993*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_height = 2; stride_height <= 3; stride_height++) {
14994*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
14995*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
14996*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
14997*4bdc9457SAndroid Build Coastguard Worker .padding(1)
14998*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
14999*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15000*4bdc9457SAndroid Build Coastguard Worker .stride_height(stride_height)
15001*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15002*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15003*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15004*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15005*4bdc9457SAndroid Build Coastguard Worker }
15006*4bdc9457SAndroid Build Coastguard Worker }
15007*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3s1xS)15008*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3s1xS) {
15009*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15010*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_width = 2; stride_width <= 3; stride_width++) {
15011*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15012*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15013*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15014*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15015*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
15016*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15017*4bdc9457SAndroid Build Coastguard Worker .stride_width(stride_width)
15018*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15019*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15020*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15021*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15022*4bdc9457SAndroid Build Coastguard Worker }
15023*4bdc9457SAndroid Build Coastguard Worker }
15024*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3s2_varying_height_padding)15025*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3s2_varying_height_padding) {
15026*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15027*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
15028*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
15029*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15030*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15031*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15032*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
15033*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
15034*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
15035*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15036*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15037*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
15038*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15039*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15040*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15041*4bdc9457SAndroid Build Coastguard Worker }
15042*4bdc9457SAndroid Build Coastguard Worker }
15043*4bdc9457SAndroid Build Coastguard Worker }
15044*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3s2_varying_width_padding)15045*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3s2_varying_width_padding) {
15046*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15047*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
15048*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
15049*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15050*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15051*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15052*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
15053*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
15054*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
15055*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15056*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15057*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
15058*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15059*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15060*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15061*4bdc9457SAndroid Build Coastguard Worker }
15062*4bdc9457SAndroid Build Coastguard Worker }
15063*4bdc9457SAndroid Build Coastguard Worker }
15064*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3s2_varying_height_adjustment)15065*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3s2_varying_height_adjustment) {
15066*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15067*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 0; adjustment_height <= 1; adjustment_height++) {
15068*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15069*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15070*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15071*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15072*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
15073*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15074*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15075*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
15076*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15077*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15078*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15079*4bdc9457SAndroid Build Coastguard Worker }
15080*4bdc9457SAndroid Build Coastguard Worker }
15081*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3s2_varying_width_adjustment)15082*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3s2_varying_width_adjustment) {
15083*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15084*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 0; adjustment_width <= 1; adjustment_width++) {
15085*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15086*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15087*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15088*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15089*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
15090*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15091*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15092*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
15093*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15094*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15095*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15096*4bdc9457SAndroid Build Coastguard Worker }
15097*4bdc9457SAndroid Build Coastguard Worker }
15098*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3s2_varying_input_height)15099*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3s2_varying_input_height) {
15100*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15101*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
15102*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15103*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15104*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
15105*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15106*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15107*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15108*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
15109*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15110*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15111*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15112*4bdc9457SAndroid Build Coastguard Worker }
15113*4bdc9457SAndroid Build Coastguard Worker }
15114*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3s2_varying_input_width)15115*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3s2_varying_input_width) {
15116*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15117*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
15118*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15119*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15120*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15121*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15122*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15123*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15124*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
15125*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15126*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15127*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15128*4bdc9457SAndroid Build Coastguard Worker }
15129*4bdc9457SAndroid Build Coastguard Worker }
15130*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3s2_varying_input_channels)15131*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3s2_varying_input_channels) {
15132*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15133*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
15134*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15135*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15136*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15137*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15138*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15139*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15140*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
15141*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15142*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15143*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15144*4bdc9457SAndroid Build Coastguard Worker }
15145*4bdc9457SAndroid Build Coastguard Worker }
15146*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3s2_varying_output_channels)15147*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3s2_varying_output_channels) {
15148*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15149*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f32.gemm.nr * 2; output_channels *= 2) {
15150*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15151*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15152*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15153*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15154*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15155*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15156*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
15157*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
15158*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15159*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15160*4bdc9457SAndroid Build Coastguard Worker }
15161*4bdc9457SAndroid Build Coastguard Worker }
15162*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3s2_with_input_stride)15163*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3s2_with_input_stride) {
15164*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15165*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15166*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15167*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15168*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15169*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15170*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15171*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
15172*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15173*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
15174*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15175*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15176*4bdc9457SAndroid Build Coastguard Worker }
15177*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3s2_with_output_stride)15178*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3s2_with_output_stride) {
15179*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15180*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15181*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15182*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15183*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15184*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15185*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15186*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
15187*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15188*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f32.gemm.nr * 2 + 13)
15189*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15190*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15191*4bdc9457SAndroid Build Coastguard Worker }
15192*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3s2_with_qmin)15193*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3s2_with_qmin) {
15194*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15195*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15196*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15197*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15198*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15199*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15200*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15201*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
15202*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15203*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
15204*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15205*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15206*4bdc9457SAndroid Build Coastguard Worker }
15207*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3s2_with_qmax)15208*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3s2_with_qmax) {
15209*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15210*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15211*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15212*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15213*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15214*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15215*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15216*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
15217*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15218*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
15219*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15220*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15221*4bdc9457SAndroid Build Coastguard Worker }
15222*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_3x3s2_without_bias)15223*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_3x3s2_without_bias) {
15224*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15225*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15226*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
15227*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15228*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15229*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15230*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15231*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15232*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
15233*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15234*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15235*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15236*4bdc9457SAndroid Build Coastguard Worker }
15237*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,weights_cache_batched_3x3s2)15238*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, weights_cache_batched_3x3s2) {
15239*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15240*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15241*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15242*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15243*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15244*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15245*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15246*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
15247*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15248*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
15249*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15250*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15251*4bdc9457SAndroid Build Coastguard Worker }
15252*4bdc9457SAndroid Build Coastguard Worker
15253*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/IGEMM path, grouped, batched ****************************/
15254*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3s2)15255*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3s2) {
15256*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15257*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15258*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15259*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15260*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15261*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15262*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15263*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15264*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15265*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15266*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15267*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15268*4bdc9457SAndroid Build Coastguard Worker }
15269*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_Kx3s2)15270*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_Kx3s2) {
15271*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15272*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 2; kernel_height <= 5; kernel_height++) {
15273*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15274*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15275*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15276*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
15277*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 3)
15278*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15279*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15280*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15281*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15282*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15283*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15284*4bdc9457SAndroid Build Coastguard Worker }
15285*4bdc9457SAndroid Build Coastguard Worker }
15286*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3xKs2)15287*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3xKs2) {
15288*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15289*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 2; kernel_width <= 5; kernel_width++) {
15290*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15291*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15292*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15293*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
15294*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, kernel_width)
15295*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15296*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15297*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15298*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15299*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15300*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15301*4bdc9457SAndroid Build Coastguard Worker }
15302*4bdc9457SAndroid Build Coastguard Worker }
15303*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3sSx1)15304*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3sSx1) {
15305*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15306*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_height = 2; stride_height <= 3; stride_height++) {
15307*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15308*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15309*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15310*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15311*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
15312*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15313*4bdc9457SAndroid Build Coastguard Worker .stride_height(stride_height)
15314*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15315*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15316*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15317*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15318*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15319*4bdc9457SAndroid Build Coastguard Worker }
15320*4bdc9457SAndroid Build Coastguard Worker }
15321*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3s1xS)15322*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3s1xS) {
15323*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15324*4bdc9457SAndroid Build Coastguard Worker for (size_t stride_width = 2; stride_width <= 3; stride_width++) {
15325*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15326*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15327*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15328*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15329*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
15330*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15331*4bdc9457SAndroid Build Coastguard Worker .stride_width(stride_width)
15332*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15333*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15334*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15335*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15336*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15337*4bdc9457SAndroid Build Coastguard Worker }
15338*4bdc9457SAndroid Build Coastguard Worker }
15339*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3s2_varying_height_padding)15340*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3s2_varying_height_padding) {
15341*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15342*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_top = 0; padding_top <= 2; padding_top++) {
15343*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_bottom = 0; padding_bottom <= 2; padding_bottom++) {
15344*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15345*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15346*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15347*4bdc9457SAndroid Build Coastguard Worker .padding_width(1)
15348*4bdc9457SAndroid Build Coastguard Worker .padding_top(padding_top)
15349*4bdc9457SAndroid Build Coastguard Worker .padding_bottom(padding_bottom)
15350*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15351*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15352*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15353*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15354*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15355*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15356*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15357*4bdc9457SAndroid Build Coastguard Worker }
15358*4bdc9457SAndroid Build Coastguard Worker }
15359*4bdc9457SAndroid Build Coastguard Worker }
15360*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3s2_varying_width_padding)15361*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3s2_varying_width_padding) {
15362*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15363*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_left = 0; padding_left <= 2; padding_left++) {
15364*4bdc9457SAndroid Build Coastguard Worker for (size_t padding_right = 0; padding_right <= 2; padding_right++) {
15365*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15366*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15367*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15368*4bdc9457SAndroid Build Coastguard Worker .padding_height(1)
15369*4bdc9457SAndroid Build Coastguard Worker .padding_left(padding_left)
15370*4bdc9457SAndroid Build Coastguard Worker .padding_right(padding_right)
15371*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15372*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15373*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15374*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15375*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15376*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15377*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15378*4bdc9457SAndroid Build Coastguard Worker }
15379*4bdc9457SAndroid Build Coastguard Worker }
15380*4bdc9457SAndroid Build Coastguard Worker }
15381*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3s2_varying_height_adjustment)15382*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3s2_varying_height_adjustment) {
15383*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15384*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_height = 0; adjustment_height <= 1; adjustment_height++) {
15385*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15386*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15387*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15388*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15389*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(adjustment_height)
15390*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15391*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15392*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15393*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15394*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15395*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15396*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15397*4bdc9457SAndroid Build Coastguard Worker }
15398*4bdc9457SAndroid Build Coastguard Worker }
15399*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3s2_varying_width_adjustment)15400*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3s2_varying_width_adjustment) {
15401*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15402*4bdc9457SAndroid Build Coastguard Worker for (size_t adjustment_width = 0; adjustment_width <= 1; adjustment_width++) {
15403*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15404*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15405*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15406*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15407*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(adjustment_width)
15408*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15409*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15410*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15411*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15412*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15413*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15414*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15415*4bdc9457SAndroid Build Coastguard Worker }
15416*4bdc9457SAndroid Build Coastguard Worker }
15417*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3s2_varying_input_height)15418*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3s2_varying_input_height) {
15419*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15420*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
15421*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15422*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15423*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
15424*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15425*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15426*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15427*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15428*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15429*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15430*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15431*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15432*4bdc9457SAndroid Build Coastguard Worker }
15433*4bdc9457SAndroid Build Coastguard Worker }
15434*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3s2_varying_input_width)15435*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3s2_varying_input_width) {
15436*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15437*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
15438*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15439*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15440*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15441*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15442*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15443*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15444*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15445*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15446*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15447*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15448*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15449*4bdc9457SAndroid Build Coastguard Worker }
15450*4bdc9457SAndroid Build Coastguard Worker }
15451*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3s2_varying_input_channels)15452*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3s2_varying_input_channels) {
15453*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15454*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 14; input_channels <= 20; input_channels++) {
15455*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15456*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15457*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15458*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15459*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15460*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15461*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15462*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
15463*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15464*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15465*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15466*4bdc9457SAndroid Build Coastguard Worker }
15467*4bdc9457SAndroid Build Coastguard Worker }
15468*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3s2_varying_output_channels)15469*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3s2_varying_output_channels) {
15470*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15471*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f32.gemm.nr * 2; output_channels *= 2) {
15472*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15473*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15474*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15475*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15476*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15477*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15478*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15479*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15480*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
15481*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15482*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15483*4bdc9457SAndroid Build Coastguard Worker }
15484*4bdc9457SAndroid Build Coastguard Worker }
15485*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3s2_with_input_stride)15486*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3s2_with_input_stride) {
15487*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15488*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15489*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15490*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15491*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15492*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15493*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15494*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15495*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15496*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15497*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(37)
15498*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15499*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15500*4bdc9457SAndroid Build Coastguard Worker }
15501*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3s2_with_output_stride)15502*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3s2_with_output_stride) {
15503*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15504*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15505*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15506*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15507*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15508*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15509*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15510*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15511*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15512*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr + 3)
15513*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f32.gemm.nr * 2 + 13)
15514*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15515*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15516*4bdc9457SAndroid Build Coastguard Worker }
15517*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3s2_with_qmin)15518*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3s2_with_qmin) {
15519*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15520*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15521*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15522*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15523*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15524*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15525*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15526*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15527*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15528*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15529*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
15530*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15531*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15532*4bdc9457SAndroid Build Coastguard Worker }
15533*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3s2_with_qmax)15534*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3s2_with_qmax) {
15535*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15536*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15537*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15538*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15539*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15540*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15541*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15542*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15543*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15544*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15545*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
15546*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15547*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15548*4bdc9457SAndroid Build Coastguard Worker }
15549*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_3x3s2_without_bias)15550*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_3x3s2_without_bias) {
15551*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15552*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15553*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
15554*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15555*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15556*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15557*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15558*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15559*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15560*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15561*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15562*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15563*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15564*4bdc9457SAndroid Build Coastguard Worker }
15565*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,weights_cache_batched_grouped_3x3s2)15566*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, weights_cache_batched_grouped_3x3s2) {
15567*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15568*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15569*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15570*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15571*4bdc9457SAndroid Build Coastguard Worker .padding(1)
15572*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15573*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15574*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15575*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15576*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15577*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
15578*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15579*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15580*4bdc9457SAndroid Build Coastguard Worker }
15581*4bdc9457SAndroid Build Coastguard Worker
15582*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/IGEMM path, setup ****************************/
15583*4bdc9457SAndroid Build Coastguard Worker
15584*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3s2_setup_changing_batch) {
15585*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15586*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15587*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15588*4bdc9457SAndroid Build Coastguard Worker .next_batch_size(5)
15589*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15590*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15591*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15592*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15593*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
15594*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
15595*4bdc9457SAndroid Build Coastguard Worker .TestSetupF32();
15596*4bdc9457SAndroid Build Coastguard Worker }
15597*4bdc9457SAndroid Build Coastguard Worker
15598*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3s2_setup_changing_height) {
15599*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15600*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15601*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15602*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15603*4bdc9457SAndroid Build Coastguard Worker .next_input_height(kStridedInputHeight + 3)
15604*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15605*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15606*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15607*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
15608*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
15609*4bdc9457SAndroid Build Coastguard Worker .TestSetupF32();
15610*4bdc9457SAndroid Build Coastguard Worker }
15611*4bdc9457SAndroid Build Coastguard Worker
15612*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 3x3s2_setup_changing_width) {
15613*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15614*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15615*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
15616*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15617*4bdc9457SAndroid Build Coastguard Worker .next_input_width(kStridedInputWidth + 3)
15618*4bdc9457SAndroid Build Coastguard Worker .kernel_size(3, 3)
15619*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15620*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15621*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
15622*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
15623*4bdc9457SAndroid Build Coastguard Worker .TestSetupF32();
15624*4bdc9457SAndroid Build Coastguard Worker }
15625*4bdc9457SAndroid Build Coastguard Worker
15626*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/GEMM path ****************************/
15627*4bdc9457SAndroid Build Coastguard Worker
15628*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 2x2s2) {
15629*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15630*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15631*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15632*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
15633*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15634*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
15635*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15636*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15637*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15638*4bdc9457SAndroid Build Coastguard Worker }
15639*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,Kx2sKx2)15640*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, Kx2sKx2) {
15641*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15642*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 3; kernel_height <= 5; kernel_height++) {
15643*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15644*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15645*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 2)
15646*4bdc9457SAndroid Build Coastguard Worker .stride(kernel_height, 2)
15647*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15648*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15649*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15650*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15651*4bdc9457SAndroid Build Coastguard Worker }
15652*4bdc9457SAndroid Build Coastguard Worker }
15653*4bdc9457SAndroid Build Coastguard Worker
15654*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 2xKs2xK) {
15655*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15656*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 3; kernel_width <= 5; kernel_width++) {
15657*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15658*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15659*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, kernel_width)
15660*4bdc9457SAndroid Build Coastguard Worker .stride(2, kernel_width)
15661*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15662*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15663*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15664*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15665*4bdc9457SAndroid Build Coastguard Worker }
15666*4bdc9457SAndroid Build Coastguard Worker }
15667*4bdc9457SAndroid Build Coastguard Worker
15668*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 2x2s2_height_adjustment) {
15669*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15670*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15671*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15672*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(1)
15673*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
15674*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15675*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
15676*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15677*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15678*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15679*4bdc9457SAndroid Build Coastguard Worker }
15680*4bdc9457SAndroid Build Coastguard Worker
15681*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 2x2s2_width_adjustment) {
15682*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15683*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15684*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15685*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(1)
15686*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
15687*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15688*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
15689*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15690*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15691*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15692*4bdc9457SAndroid Build Coastguard Worker }
15693*4bdc9457SAndroid Build Coastguard Worker
15694*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 2x2s2_varying_input_height) {
15695*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15696*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
15697*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15698*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
15699*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
15700*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15701*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
15702*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15703*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15704*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15705*4bdc9457SAndroid Build Coastguard Worker }
15706*4bdc9457SAndroid Build Coastguard Worker }
15707*4bdc9457SAndroid Build Coastguard Worker
15708*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 2x2s2_varying_input_width) {
15709*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15710*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
15711*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15712*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15713*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
15714*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15715*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
15716*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15717*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15718*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15719*4bdc9457SAndroid Build Coastguard Worker }
15720*4bdc9457SAndroid Build Coastguard Worker }
15721*4bdc9457SAndroid Build Coastguard Worker
15722*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 2x2s2_varying_input_channels) {
15723*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15724*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
15725*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15726*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15727*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
15728*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15729*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
15730*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15731*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15732*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15733*4bdc9457SAndroid Build Coastguard Worker }
15734*4bdc9457SAndroid Build Coastguard Worker }
15735*4bdc9457SAndroid Build Coastguard Worker
15736*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 2x2s2_varying_output_channels) {
15737*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15738*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f32.gemm.nr * 2; output_channels *= 2) {
15739*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15740*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15741*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
15742*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15743*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
15744*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
15745*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15746*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15747*4bdc9457SAndroid Build Coastguard Worker }
15748*4bdc9457SAndroid Build Coastguard Worker }
15749*4bdc9457SAndroid Build Coastguard Worker
15750*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 2x2s2_with_input_stride) {
15751*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15752*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15753*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15754*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
15755*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15756*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
15757*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15758*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
15759*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15760*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15761*4bdc9457SAndroid Build Coastguard Worker }
15762*4bdc9457SAndroid Build Coastguard Worker
15763*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 2x2s2_with_output_stride) {
15764*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15765*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15766*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15767*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
15768*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15769*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
15770*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15771*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f32.gemm.nr * 2 + 13)
15772*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15773*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15774*4bdc9457SAndroid Build Coastguard Worker }
15775*4bdc9457SAndroid Build Coastguard Worker
15776*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 2x2s2_with_qmin) {
15777*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15778*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15779*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15780*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
15781*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15782*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
15783*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15784*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
15785*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15786*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15787*4bdc9457SAndroid Build Coastguard Worker }
15788*4bdc9457SAndroid Build Coastguard Worker
15789*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 2x2s2_with_qmax) {
15790*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15791*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15792*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15793*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
15794*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15795*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
15796*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15797*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
15798*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15799*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15800*4bdc9457SAndroid Build Coastguard Worker }
15801*4bdc9457SAndroid Build Coastguard Worker
15802*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 2x2s2_without_bias) {
15803*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15804*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15805*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
15806*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15807*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
15808*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15809*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
15810*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15811*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15812*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15813*4bdc9457SAndroid Build Coastguard Worker }
15814*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,weights_cache_2x2s2)15815*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, weights_cache_2x2s2) {
15816*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15817*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15818*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15819*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
15820*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15821*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
15822*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15823*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
15824*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15825*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15826*4bdc9457SAndroid Build Coastguard Worker }
15827*4bdc9457SAndroid Build Coastguard Worker
15828*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/GEMM path, grouped ****************************/
15829*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_2x2s2)15830*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_2x2s2) {
15831*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15832*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15833*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15834*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
15835*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15836*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15837*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15838*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15839*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15840*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15841*4bdc9457SAndroid Build Coastguard Worker }
15842*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_Kx2sKx2)15843*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_Kx2sKx2) {
15844*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15845*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 3; kernel_height <= 5; kernel_height++) {
15846*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15847*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15848*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 2)
15849*4bdc9457SAndroid Build Coastguard Worker .stride(kernel_height, 2)
15850*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15851*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15852*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15853*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15854*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15855*4bdc9457SAndroid Build Coastguard Worker }
15856*4bdc9457SAndroid Build Coastguard Worker }
15857*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_2xKs2xK)15858*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_2xKs2xK) {
15859*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15860*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 3; kernel_width <= 5; kernel_width++) {
15861*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15862*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15863*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, kernel_width)
15864*4bdc9457SAndroid Build Coastguard Worker .stride(2, kernel_width)
15865*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15866*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15867*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15868*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15869*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15870*4bdc9457SAndroid Build Coastguard Worker }
15871*4bdc9457SAndroid Build Coastguard Worker }
15872*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_2x2s2_height_adjustment)15873*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_2x2s2_height_adjustment) {
15874*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15875*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15876*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15877*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(1)
15878*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
15879*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15880*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15881*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15882*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15883*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15884*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15885*4bdc9457SAndroid Build Coastguard Worker }
15886*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_2x2s2_width_adjustment)15887*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_2x2s2_width_adjustment) {
15888*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15889*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15890*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15891*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(1)
15892*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
15893*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15894*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15895*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15896*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15897*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15898*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15899*4bdc9457SAndroid Build Coastguard Worker }
15900*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_2x2s2_varying_input_height)15901*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_2x2s2_varying_input_height) {
15902*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15903*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
15904*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15905*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
15906*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
15907*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15908*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15909*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15910*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15911*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15912*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15913*4bdc9457SAndroid Build Coastguard Worker }
15914*4bdc9457SAndroid Build Coastguard Worker }
15915*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_2x2s2_varying_input_width)15916*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_2x2s2_varying_input_width) {
15917*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15918*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
15919*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15920*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15921*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
15922*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15923*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15924*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15925*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15926*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15927*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15928*4bdc9457SAndroid Build Coastguard Worker }
15929*4bdc9457SAndroid Build Coastguard Worker }
15930*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_2x2s2_varying_input_channels)15931*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_2x2s2_varying_input_channels) {
15932*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15933*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 14; input_channels <= 20; input_channels++) {
15934*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15935*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15936*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
15937*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15938*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15939*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
15940*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15941*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15942*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15943*4bdc9457SAndroid Build Coastguard Worker }
15944*4bdc9457SAndroid Build Coastguard Worker }
15945*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_2x2s2_varying_output_channels)15946*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_2x2s2_varying_output_channels) {
15947*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15948*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f32.gemm.nr * 2; output_channels *= 2) {
15949*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15950*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15951*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
15952*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15953*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15954*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15955*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
15956*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
15957*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15958*4bdc9457SAndroid Build Coastguard Worker }
15959*4bdc9457SAndroid Build Coastguard Worker }
15960*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_2x2s2_with_input_stride)15961*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_2x2s2_with_input_stride) {
15962*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15963*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15964*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15965*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
15966*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15967*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15968*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15969*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15970*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(37)
15971*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15972*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15973*4bdc9457SAndroid Build Coastguard Worker }
15974*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_2x2s2_with_output_stride)15975*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_2x2s2_with_output_stride) {
15976*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15977*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15978*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15979*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
15980*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15981*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15982*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15983*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr + 3)
15984*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f32.gemm.nr * 2 + 13)
15985*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
15986*4bdc9457SAndroid Build Coastguard Worker .TestF32();
15987*4bdc9457SAndroid Build Coastguard Worker }
15988*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_2x2s2_with_qmin)15989*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_2x2s2_with_qmin) {
15990*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
15991*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
15992*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
15993*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
15994*4bdc9457SAndroid Build Coastguard Worker .stride(2)
15995*4bdc9457SAndroid Build Coastguard Worker .groups(2)
15996*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
15997*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
15998*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
15999*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
16000*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16001*4bdc9457SAndroid Build Coastguard Worker }
16002*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_2x2s2_with_qmax)16003*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_2x2s2_with_qmax) {
16004*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16005*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16006*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16007*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16008*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16009*4bdc9457SAndroid Build Coastguard Worker .groups(2)
16010*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
16011*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16012*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
16013*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
16014*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16015*4bdc9457SAndroid Build Coastguard Worker }
16016*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,grouped_2x2s2_without_bias)16017*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, grouped_2x2s2_without_bias) {
16018*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16019*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16020*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
16021*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16022*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16023*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16024*4bdc9457SAndroid Build Coastguard Worker .groups(2)
16025*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
16026*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16027*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
16028*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16029*4bdc9457SAndroid Build Coastguard Worker }
16030*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,weights_cache_grouped_2x2s2)16031*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, weights_cache_grouped_2x2s2) {
16032*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16033*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16034*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16035*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16036*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16037*4bdc9457SAndroid Build Coastguard Worker .groups(2)
16038*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
16039*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16040*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
16041*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
16042*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16043*4bdc9457SAndroid Build Coastguard Worker }
16044*4bdc9457SAndroid Build Coastguard Worker
16045*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/GEMM path, batched ****************************/
16046*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_2x2s2)16047*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_2x2s2) {
16048*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16049*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16050*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16051*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16052*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16053*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16054*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
16055*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16056*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
16057*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16058*4bdc9457SAndroid Build Coastguard Worker }
16059*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_Kx2sKx2)16060*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_Kx2sKx2) {
16061*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16062*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 3; kernel_height <= 5; kernel_height++) {
16063*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16064*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16065*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16066*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 2)
16067*4bdc9457SAndroid Build Coastguard Worker .stride(kernel_height, 2)
16068*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
16069*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16070*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
16071*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16072*4bdc9457SAndroid Build Coastguard Worker }
16073*4bdc9457SAndroid Build Coastguard Worker }
16074*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_2xKs2xK)16075*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_2xKs2xK) {
16076*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16077*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 3; kernel_width <= 5; kernel_width++) {
16078*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16079*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16080*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16081*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, kernel_width)
16082*4bdc9457SAndroid Build Coastguard Worker .stride(2, kernel_width)
16083*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
16084*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16085*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
16086*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16087*4bdc9457SAndroid Build Coastguard Worker }
16088*4bdc9457SAndroid Build Coastguard Worker }
16089*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_2x2s2_height_adjustment)16090*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_2x2s2_height_adjustment) {
16091*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16092*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16093*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16094*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16095*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(1)
16096*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16097*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16098*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
16099*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16100*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
16101*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16102*4bdc9457SAndroid Build Coastguard Worker }
16103*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_2x2s2_width_adjustment)16104*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_2x2s2_width_adjustment) {
16105*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16106*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16107*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16108*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16109*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(1)
16110*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16111*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16112*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
16113*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16114*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
16115*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16116*4bdc9457SAndroid Build Coastguard Worker }
16117*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_2x2s2_varying_input_height)16118*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_2x2s2_varying_input_height) {
16119*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16120*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
16121*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16122*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16123*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
16124*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16125*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16126*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
16127*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16128*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
16129*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16130*4bdc9457SAndroid Build Coastguard Worker }
16131*4bdc9457SAndroid Build Coastguard Worker }
16132*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_2x2s2_varying_input_width)16133*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_2x2s2_varying_input_width) {
16134*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16135*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
16136*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16137*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16138*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16139*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16140*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16141*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
16142*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16143*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
16144*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16145*4bdc9457SAndroid Build Coastguard Worker }
16146*4bdc9457SAndroid Build Coastguard Worker }
16147*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_2x2s2_varying_input_channels)16148*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_2x2s2_varying_input_channels) {
16149*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16150*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) {
16151*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16152*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16153*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16154*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16155*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16156*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
16157*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16158*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
16159*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16160*4bdc9457SAndroid Build Coastguard Worker }
16161*4bdc9457SAndroid Build Coastguard Worker }
16162*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_2x2s2_varying_output_channels)16163*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_2x2s2_varying_output_channels) {
16164*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16165*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f32.gemm.nr * 2; output_channels *= 2) {
16166*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16167*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16168*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16169*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16170*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16171*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
16172*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
16173*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
16174*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16175*4bdc9457SAndroid Build Coastguard Worker }
16176*4bdc9457SAndroid Build Coastguard Worker }
16177*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_2x2s2_with_input_stride)16178*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_2x2s2_with_input_stride) {
16179*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16180*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16181*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16182*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16183*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16184*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16185*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
16186*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16187*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(28)
16188*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
16189*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16190*4bdc9457SAndroid Build Coastguard Worker }
16191*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_2x2s2_with_output_stride)16192*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_2x2s2_with_output_stride) {
16193*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16194*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16195*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16196*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16197*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16198*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16199*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
16200*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16201*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f32.gemm.nr * 2 + 13)
16202*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
16203*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16204*4bdc9457SAndroid Build Coastguard Worker }
16205*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_2x2s2_with_qmin)16206*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_2x2s2_with_qmin) {
16207*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16208*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16209*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16210*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16211*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16212*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16213*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
16214*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16215*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
16216*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
16217*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16218*4bdc9457SAndroid Build Coastguard Worker }
16219*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_2x2s2_with_qmax)16220*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_2x2s2_with_qmax) {
16221*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16222*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16223*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16224*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16225*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16226*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16227*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
16228*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16229*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
16230*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
16231*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16232*4bdc9457SAndroid Build Coastguard Worker }
16233*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_2x2s2_without_bias)16234*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_2x2s2_without_bias) {
16235*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16236*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16237*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
16238*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16239*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16240*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16241*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16242*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(23)
16243*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16244*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
16245*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16246*4bdc9457SAndroid Build Coastguard Worker }
16247*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,weights_cache_batched_2x2s2)16248*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, weights_cache_batched_2x2s2) {
16249*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16250*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16251*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16252*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16253*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16254*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16255*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
16256*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16257*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
16258*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
16259*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16260*4bdc9457SAndroid Build Coastguard Worker }
16261*4bdc9457SAndroid Build Coastguard Worker
16262*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/GEMM path, grouped, batched ****************************/
16263*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_2x2s2)16264*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_2x2s2) {
16265*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16266*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16267*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16268*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16269*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16270*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16271*4bdc9457SAndroid Build Coastguard Worker .groups(2)
16272*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
16273*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16274*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
16275*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16276*4bdc9457SAndroid Build Coastguard Worker }
16277*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_Kx2sKx2)16278*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_Kx2sKx2) {
16279*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16280*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_height = 3; kernel_height <= 5; kernel_height++) {
16281*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16282*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16283*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16284*4bdc9457SAndroid Build Coastguard Worker .kernel_size(kernel_height, 2)
16285*4bdc9457SAndroid Build Coastguard Worker .stride(kernel_height, 2)
16286*4bdc9457SAndroid Build Coastguard Worker .groups(2)
16287*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
16288*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16289*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
16290*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16291*4bdc9457SAndroid Build Coastguard Worker }
16292*4bdc9457SAndroid Build Coastguard Worker }
16293*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_2xKs2xK)16294*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_2xKs2xK) {
16295*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16296*4bdc9457SAndroid Build Coastguard Worker for (size_t kernel_width = 3; kernel_width <= 5; kernel_width++) {
16297*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16298*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16299*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16300*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, kernel_width)
16301*4bdc9457SAndroid Build Coastguard Worker .stride(2, kernel_width)
16302*4bdc9457SAndroid Build Coastguard Worker .groups(2)
16303*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
16304*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16305*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
16306*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16307*4bdc9457SAndroid Build Coastguard Worker }
16308*4bdc9457SAndroid Build Coastguard Worker }
16309*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_2x2s2_height_adjustment)16310*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_2x2s2_height_adjustment) {
16311*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16312*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16313*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16314*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16315*4bdc9457SAndroid Build Coastguard Worker .adjustment_height(1)
16316*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16317*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16318*4bdc9457SAndroid Build Coastguard Worker .groups(2)
16319*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
16320*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16321*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
16322*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16323*4bdc9457SAndroid Build Coastguard Worker }
16324*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_2x2s2_width_adjustment)16325*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_2x2s2_width_adjustment) {
16326*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16327*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16328*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16329*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16330*4bdc9457SAndroid Build Coastguard Worker .adjustment_width(1)
16331*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16332*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16333*4bdc9457SAndroid Build Coastguard Worker .groups(2)
16334*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
16335*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16336*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
16337*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16338*4bdc9457SAndroid Build Coastguard Worker }
16339*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_2x2s2_varying_input_height)16340*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_2x2s2_varying_input_height) {
16341*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16342*4bdc9457SAndroid Build Coastguard Worker for (size_t input_height = kStridedInputHeight - 2; input_height <= kStridedInputHeight + 2; input_height++) {
16343*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16344*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16345*4bdc9457SAndroid Build Coastguard Worker .input_size(input_height, kStridedInputWidth)
16346*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16347*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16348*4bdc9457SAndroid Build Coastguard Worker .groups(2)
16349*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
16350*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16351*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
16352*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16353*4bdc9457SAndroid Build Coastguard Worker }
16354*4bdc9457SAndroid Build Coastguard Worker }
16355*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_2x2s2_varying_input_width)16356*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_2x2s2_varying_input_width) {
16357*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16358*4bdc9457SAndroid Build Coastguard Worker for (size_t input_width = kStridedInputWidth - 2; input_width <= kStridedInputWidth + 2; input_width++) {
16359*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16360*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16361*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16362*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16363*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16364*4bdc9457SAndroid Build Coastguard Worker .groups(2)
16365*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
16366*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16367*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
16368*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16369*4bdc9457SAndroid Build Coastguard Worker }
16370*4bdc9457SAndroid Build Coastguard Worker }
16371*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_2x2s2_varying_input_channels)16372*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_2x2s2_varying_input_channels) {
16373*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16374*4bdc9457SAndroid Build Coastguard Worker for (size_t input_channels = 14; input_channels <= 20; input_channels++) {
16375*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16376*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16377*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16378*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16379*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16380*4bdc9457SAndroid Build Coastguard Worker .groups(2)
16381*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(input_channels)
16382*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16383*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
16384*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16385*4bdc9457SAndroid Build Coastguard Worker }
16386*4bdc9457SAndroid Build Coastguard Worker }
16387*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_2x2s2_varying_output_channels)16388*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_2x2s2_varying_output_channels) {
16389*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16390*4bdc9457SAndroid Build Coastguard Worker for (size_t output_channels = 1; output_channels <= xnn_params.f32.gemm.nr * 2; output_channels *= 2) {
16391*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16392*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16393*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16394*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16395*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16396*4bdc9457SAndroid Build Coastguard Worker .groups(2)
16397*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
16398*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(output_channels)
16399*4bdc9457SAndroid Build Coastguard Worker .iterations(1)
16400*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16401*4bdc9457SAndroid Build Coastguard Worker }
16402*4bdc9457SAndroid Build Coastguard Worker }
16403*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_2x2s2_with_input_stride)16404*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_2x2s2_with_input_stride) {
16405*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16406*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16407*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16408*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16409*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16410*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16411*4bdc9457SAndroid Build Coastguard Worker .groups(2)
16412*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
16413*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16414*4bdc9457SAndroid Build Coastguard Worker .input_pixel_stride(37)
16415*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
16416*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16417*4bdc9457SAndroid Build Coastguard Worker }
16418*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_2x2s2_with_output_stride)16419*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_2x2s2_with_output_stride) {
16420*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16421*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16422*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16423*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16424*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16425*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16426*4bdc9457SAndroid Build Coastguard Worker .groups(2)
16427*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
16428*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr + 3)
16429*4bdc9457SAndroid Build Coastguard Worker .output_pixel_stride(xnn_params.f32.gemm.nr * 2 + 13)
16430*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
16431*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16432*4bdc9457SAndroid Build Coastguard Worker }
16433*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_2x2s2_with_qmin)16434*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_2x2s2_with_qmin) {
16435*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16436*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16437*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16438*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16439*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16440*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16441*4bdc9457SAndroid Build Coastguard Worker .groups(2)
16442*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
16443*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16444*4bdc9457SAndroid Build Coastguard Worker .qmin(128)
16445*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
16446*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16447*4bdc9457SAndroid Build Coastguard Worker }
16448*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_2x2s2_with_qmax)16449*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_2x2s2_with_qmax) {
16450*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16451*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16452*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16453*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16454*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16455*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16456*4bdc9457SAndroid Build Coastguard Worker .groups(2)
16457*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
16458*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16459*4bdc9457SAndroid Build Coastguard Worker .qmax(128)
16460*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
16461*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16462*4bdc9457SAndroid Build Coastguard Worker }
16463*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,batched_grouped_2x2s2_without_bias)16464*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, batched_grouped_2x2s2_without_bias) {
16465*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16466*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16467*4bdc9457SAndroid Build Coastguard Worker .has_bias(false)
16468*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16469*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16470*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16471*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16472*4bdc9457SAndroid Build Coastguard Worker .groups(2)
16473*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
16474*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16475*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
16476*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16477*4bdc9457SAndroid Build Coastguard Worker }
16478*4bdc9457SAndroid Build Coastguard Worker
TEST(DECONVOLUTION_NHWC_F32,weights_cache_batched_grouped_2x2s2)16479*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, weights_cache_batched_grouped_2x2s2) {
16480*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16481*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16482*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16483*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16484*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16485*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16486*4bdc9457SAndroid Build Coastguard Worker .groups(2)
16487*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(17)
16488*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(xnn_params.f32.gemm.nr * 2 + 3)
16489*4bdc9457SAndroid Build Coastguard Worker .use_weights_cache(true)
16490*4bdc9457SAndroid Build Coastguard Worker .iterations(3)
16491*4bdc9457SAndroid Build Coastguard Worker .TestF32();
16492*4bdc9457SAndroid Build Coastguard Worker }
16493*4bdc9457SAndroid Build Coastguard Worker
16494*4bdc9457SAndroid Build Coastguard Worker /**************************** SUBCONV2D/GEMM path, setup ****************************/
16495*4bdc9457SAndroid Build Coastguard Worker
16496*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 2x2s2_setup_changing_batch) {
16497*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16498*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16499*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16500*4bdc9457SAndroid Build Coastguard Worker .next_batch_size(5)
16501*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16502*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16503*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16504*4bdc9457SAndroid Build Coastguard Worker .groups(2)
16505*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
16506*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
16507*4bdc9457SAndroid Build Coastguard Worker .TestSetupF32();
16508*4bdc9457SAndroid Build Coastguard Worker }
16509*4bdc9457SAndroid Build Coastguard Worker
16510*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 2x2s2_setup_changing_height) {
16511*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16512*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16513*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16514*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16515*4bdc9457SAndroid Build Coastguard Worker .next_input_height(kStridedInputHeight + 3)
16516*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16517*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16518*4bdc9457SAndroid Build Coastguard Worker .groups(2)
16519*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
16520*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
16521*4bdc9457SAndroid Build Coastguard Worker .TestSetupF32();
16522*4bdc9457SAndroid Build Coastguard Worker }
16523*4bdc9457SAndroid Build Coastguard Worker
16524*4bdc9457SAndroid Build Coastguard Worker TEST(DECONVOLUTION_NHWC_F32, 2x2s2_setup_changing_width) {
16525*4bdc9457SAndroid Build Coastguard Worker ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
16526*4bdc9457SAndroid Build Coastguard Worker DeconvolutionOperatorTester()
16527*4bdc9457SAndroid Build Coastguard Worker .batch_size(2)
16528*4bdc9457SAndroid Build Coastguard Worker .input_size(kStridedInputHeight, kStridedInputWidth)
16529*4bdc9457SAndroid Build Coastguard Worker .next_input_width(kStridedInputWidth + 3)
16530*4bdc9457SAndroid Build Coastguard Worker .kernel_size(2, 2)
16531*4bdc9457SAndroid Build Coastguard Worker .stride(2)
16532*4bdc9457SAndroid Build Coastguard Worker .groups(2)
16533*4bdc9457SAndroid Build Coastguard Worker .group_input_channels(15)
16534*4bdc9457SAndroid Build Coastguard Worker .group_output_channels(17)
16535*4bdc9457SAndroid Build Coastguard Worker .TestSetupF32();
16536*4bdc9457SAndroid Build Coastguard Worker }
16537