1*4bdc9457SAndroid Build Coastguard Worker // Auto-generated file. Do not edit!
2*4bdc9457SAndroid Build Coastguard Worker // Template: src/qu8-dwconv/unipass-neon-mul8.c.in
3*4bdc9457SAndroid Build Coastguard Worker // Generator: tools/xngen
4*4bdc9457SAndroid Build Coastguard Worker //
5*4bdc9457SAndroid Build Coastguard Worker // Copyright 2020 Google LLC
6*4bdc9457SAndroid Build Coastguard Worker //
7*4bdc9457SAndroid Build Coastguard Worker // This source code is licensed under the BSD-style license found in the
8*4bdc9457SAndroid Build Coastguard Worker // LICENSE file in the root directory of this source tree.
9*4bdc9457SAndroid Build Coastguard Worker
10*4bdc9457SAndroid Build Coastguard Worker #include <assert.h>
11*4bdc9457SAndroid Build Coastguard Worker
12*4bdc9457SAndroid Build Coastguard Worker #include <arm_neon.h>
13*4bdc9457SAndroid Build Coastguard Worker
14*4bdc9457SAndroid Build Coastguard Worker #include <xnnpack/dwconv.h>
15*4bdc9457SAndroid Build Coastguard Worker
16*4bdc9457SAndroid Build Coastguard Worker
xnn_qu8_dwconv_minmax_rndnu_ukernel_up8x25__neon_mul8(size_t channels,size_t output_width,const uint8_t ** input,const void * weights,uint8_t * output,size_t input_stride,size_t output_increment,size_t input_offset,const uint8_t * zero,const union xnn_qu8_conv_minmax_params params[restrict XNN_MIN_ELEMENTS (1)])17*4bdc9457SAndroid Build Coastguard Worker void xnn_qu8_dwconv_minmax_rndnu_ukernel_up8x25__neon_mul8(
18*4bdc9457SAndroid Build Coastguard Worker size_t channels,
19*4bdc9457SAndroid Build Coastguard Worker size_t output_width,
20*4bdc9457SAndroid Build Coastguard Worker const uint8_t** input,
21*4bdc9457SAndroid Build Coastguard Worker const void* weights,
22*4bdc9457SAndroid Build Coastguard Worker uint8_t* output,
23*4bdc9457SAndroid Build Coastguard Worker size_t input_stride,
24*4bdc9457SAndroid Build Coastguard Worker size_t output_increment,
25*4bdc9457SAndroid Build Coastguard Worker size_t input_offset,
26*4bdc9457SAndroid Build Coastguard Worker const uint8_t* zero,
27*4bdc9457SAndroid Build Coastguard Worker const union xnn_qu8_conv_minmax_params params[restrict XNN_MIN_ELEMENTS(1)]) XNN_OOB_READS
28*4bdc9457SAndroid Build Coastguard Worker {
29*4bdc9457SAndroid Build Coastguard Worker assert(channels != 0);
30*4bdc9457SAndroid Build Coastguard Worker assert(output_width != 0);
31*4bdc9457SAndroid Build Coastguard Worker
32*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vkernel_zero_point = vld1_dup_u8(params->rndnu_neon.kernel_zero_point);
33*4bdc9457SAndroid Build Coastguard Worker const uint16x8_t vkernel_zero_point16 = vmovl_u8(vkernel_zero_point);
34*4bdc9457SAndroid Build Coastguard Worker const int32x4_t vright_pre_shift = vld1q_dup_s32(¶ms->rndnu_neon.right_pre_shift);
35*4bdc9457SAndroid Build Coastguard Worker const int32x4_t vmultiplier = vld1q_dup_s32(¶ms->rndnu_neon.multiplier);
36*4bdc9457SAndroid Build Coastguard Worker const int32x4_t vright_post_shift = vld1q_dup_s32(¶ms->rndnu_neon.right_post_shift);
37*4bdc9457SAndroid Build Coastguard Worker const int16x8_t voutput_zero_point = vld1q_dup_s16(¶ms->rndnu_neon.output_zero_point);
38*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t voutput_min = vld1_dup_u8(¶ms->rndnu_neon.output_min);
39*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t voutput_max = vld1_dup_u8(¶ms->rndnu_neon.output_max);
40*4bdc9457SAndroid Build Coastguard Worker do {
41*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i0 = input[0];
42*4bdc9457SAndroid Build Coastguard Worker assert(i0 != NULL);
43*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i0 != zero) {
44*4bdc9457SAndroid Build Coastguard Worker i0 = (const uint8_t*) ((uintptr_t) i0 + input_offset);
45*4bdc9457SAndroid Build Coastguard Worker }
46*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i1 = input[1];
47*4bdc9457SAndroid Build Coastguard Worker assert(i1 != NULL);
48*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i1 != zero) {
49*4bdc9457SAndroid Build Coastguard Worker i1 = (const uint8_t*) ((uintptr_t) i1 + input_offset);
50*4bdc9457SAndroid Build Coastguard Worker }
51*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i2 = input[2];
52*4bdc9457SAndroid Build Coastguard Worker assert(i2 != NULL);
53*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i2 != zero) {
54*4bdc9457SAndroid Build Coastguard Worker i2 = (const uint8_t*) ((uintptr_t) i2 + input_offset);
55*4bdc9457SAndroid Build Coastguard Worker }
56*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i3 = input[3];
57*4bdc9457SAndroid Build Coastguard Worker assert(i3 != NULL);
58*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i3 != zero) {
59*4bdc9457SAndroid Build Coastguard Worker i3 = (const uint8_t*) ((uintptr_t) i3 + input_offset);
60*4bdc9457SAndroid Build Coastguard Worker }
61*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i4 = input[4];
62*4bdc9457SAndroid Build Coastguard Worker assert(i4 != NULL);
63*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i4 != zero) {
64*4bdc9457SAndroid Build Coastguard Worker i4 = (const uint8_t*) ((uintptr_t) i4 + input_offset);
65*4bdc9457SAndroid Build Coastguard Worker }
66*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i5 = input[5];
67*4bdc9457SAndroid Build Coastguard Worker assert(i5 != NULL);
68*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i5 != zero) {
69*4bdc9457SAndroid Build Coastguard Worker i5 = (const uint8_t*) ((uintptr_t) i5 + input_offset);
70*4bdc9457SAndroid Build Coastguard Worker }
71*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i6 = input[6];
72*4bdc9457SAndroid Build Coastguard Worker assert(i6 != NULL);
73*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i6 != zero) {
74*4bdc9457SAndroid Build Coastguard Worker i6 = (const uint8_t*) ((uintptr_t) i6 + input_offset);
75*4bdc9457SAndroid Build Coastguard Worker }
76*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i7 = input[7];
77*4bdc9457SAndroid Build Coastguard Worker assert(i7 != NULL);
78*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i7 != zero) {
79*4bdc9457SAndroid Build Coastguard Worker i7 = (const uint8_t*) ((uintptr_t) i7 + input_offset);
80*4bdc9457SAndroid Build Coastguard Worker }
81*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i8 = input[8];
82*4bdc9457SAndroid Build Coastguard Worker assert(i8 != NULL);
83*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i8 != zero) {
84*4bdc9457SAndroid Build Coastguard Worker i8 = (const uint8_t*) ((uintptr_t) i8 + input_offset);
85*4bdc9457SAndroid Build Coastguard Worker }
86*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i9 = input[9];
87*4bdc9457SAndroid Build Coastguard Worker assert(i9 != NULL);
88*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i9 != zero) {
89*4bdc9457SAndroid Build Coastguard Worker i9 = (const uint8_t*) ((uintptr_t) i9 + input_offset);
90*4bdc9457SAndroid Build Coastguard Worker }
91*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i10 = input[10];
92*4bdc9457SAndroid Build Coastguard Worker assert(i10 != NULL);
93*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i10 != zero) {
94*4bdc9457SAndroid Build Coastguard Worker i10 = (const uint8_t*) ((uintptr_t) i10 + input_offset);
95*4bdc9457SAndroid Build Coastguard Worker }
96*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i11 = input[11];
97*4bdc9457SAndroid Build Coastguard Worker assert(i11 != NULL);
98*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i11 != zero) {
99*4bdc9457SAndroid Build Coastguard Worker i11 = (const uint8_t*) ((uintptr_t) i11 + input_offset);
100*4bdc9457SAndroid Build Coastguard Worker }
101*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i12 = input[12];
102*4bdc9457SAndroid Build Coastguard Worker assert(i12 != NULL);
103*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i12 != zero) {
104*4bdc9457SAndroid Build Coastguard Worker i12 = (const uint8_t*) ((uintptr_t) i12 + input_offset);
105*4bdc9457SAndroid Build Coastguard Worker }
106*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i13 = input[13];
107*4bdc9457SAndroid Build Coastguard Worker assert(i13 != NULL);
108*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i13 != zero) {
109*4bdc9457SAndroid Build Coastguard Worker i13 = (const uint8_t*) ((uintptr_t) i13 + input_offset);
110*4bdc9457SAndroid Build Coastguard Worker }
111*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i14 = input[14];
112*4bdc9457SAndroid Build Coastguard Worker assert(i14 != NULL);
113*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i14 != zero) {
114*4bdc9457SAndroid Build Coastguard Worker i14 = (const uint8_t*) ((uintptr_t) i14 + input_offset);
115*4bdc9457SAndroid Build Coastguard Worker }
116*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i15 = input[15];
117*4bdc9457SAndroid Build Coastguard Worker assert(i15 != NULL);
118*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i15 != zero) {
119*4bdc9457SAndroid Build Coastguard Worker i15 = (const uint8_t*) ((uintptr_t) i15 + input_offset);
120*4bdc9457SAndroid Build Coastguard Worker }
121*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i16 = input[16];
122*4bdc9457SAndroid Build Coastguard Worker assert(i16 != NULL);
123*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i16 != zero) {
124*4bdc9457SAndroid Build Coastguard Worker i16 = (const uint8_t*) ((uintptr_t) i16 + input_offset);
125*4bdc9457SAndroid Build Coastguard Worker }
126*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i17 = input[17];
127*4bdc9457SAndroid Build Coastguard Worker assert(i17 != NULL);
128*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i17 != zero) {
129*4bdc9457SAndroid Build Coastguard Worker i17 = (const uint8_t*) ((uintptr_t) i17 + input_offset);
130*4bdc9457SAndroid Build Coastguard Worker }
131*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i18 = input[18];
132*4bdc9457SAndroid Build Coastguard Worker assert(i18 != NULL);
133*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i18 != zero) {
134*4bdc9457SAndroid Build Coastguard Worker i18 = (const uint8_t*) ((uintptr_t) i18 + input_offset);
135*4bdc9457SAndroid Build Coastguard Worker }
136*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i19 = input[19];
137*4bdc9457SAndroid Build Coastguard Worker assert(i19 != NULL);
138*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i19 != zero) {
139*4bdc9457SAndroid Build Coastguard Worker i19 = (const uint8_t*) ((uintptr_t) i19 + input_offset);
140*4bdc9457SAndroid Build Coastguard Worker }
141*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i20 = input[20];
142*4bdc9457SAndroid Build Coastguard Worker assert(i20 != NULL);
143*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i20 != zero) {
144*4bdc9457SAndroid Build Coastguard Worker i20 = (const uint8_t*) ((uintptr_t) i20 + input_offset);
145*4bdc9457SAndroid Build Coastguard Worker }
146*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i21 = input[21];
147*4bdc9457SAndroid Build Coastguard Worker assert(i21 != NULL);
148*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i21 != zero) {
149*4bdc9457SAndroid Build Coastguard Worker i21 = (const uint8_t*) ((uintptr_t) i21 + input_offset);
150*4bdc9457SAndroid Build Coastguard Worker }
151*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i22 = input[22];
152*4bdc9457SAndroid Build Coastguard Worker assert(i22 != NULL);
153*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i22 != zero) {
154*4bdc9457SAndroid Build Coastguard Worker i22 = (const uint8_t*) ((uintptr_t) i22 + input_offset);
155*4bdc9457SAndroid Build Coastguard Worker }
156*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i23 = input[23];
157*4bdc9457SAndroid Build Coastguard Worker assert(i23 != NULL);
158*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i23 != zero) {
159*4bdc9457SAndroid Build Coastguard Worker i23 = (const uint8_t*) ((uintptr_t) i23 + input_offset);
160*4bdc9457SAndroid Build Coastguard Worker }
161*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i24 = input[24];
162*4bdc9457SAndroid Build Coastguard Worker assert(i24 != NULL);
163*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i24 != zero) {
164*4bdc9457SAndroid Build Coastguard Worker i24 = (const uint8_t*) ((uintptr_t) i24 + input_offset);
165*4bdc9457SAndroid Build Coastguard Worker }
166*4bdc9457SAndroid Build Coastguard Worker input = (const uint8_t**) ((uintptr_t) input + input_stride);
167*4bdc9457SAndroid Build Coastguard Worker
168*4bdc9457SAndroid Build Coastguard Worker
169*4bdc9457SAndroid Build Coastguard Worker size_t c = channels;
170*4bdc9457SAndroid Build Coastguard Worker const void* w = weights;
171*4bdc9457SAndroid Build Coastguard Worker for (; c >= 8; c -= 8) {
172*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc0123 = vld1q_s32(w); w = (const void*) ((const int32_t*) w + 4);
173*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc4567 = vld1q_s32(w); w = (const void*) ((const int32_t*) w + 4);
174*4bdc9457SAndroid Build Coastguard Worker
175*4bdc9457SAndroid Build Coastguard Worker
176*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vi0x01234567 = vld1_u8(i0); i0 += 8;
177*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vk0x01234567 = vld1_u8(w); w = (const void*) ((const int8_t*) w + 8);
178*4bdc9457SAndroid Build Coastguard Worker
179*4bdc9457SAndroid Build Coastguard Worker uint16x8_t vprod01234567 = vmull_u8(vi0x01234567, vk0x01234567);
180*4bdc9457SAndroid Build Coastguard Worker
181*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc0123), vget_low_u16(vprod01234567)));
182*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc4567), vget_high_u16(vprod01234567)));
183*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vi1x01234567 = vld1_u8(i1); i1 += 8;
184*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vk1x01234567 = vld1_u8(w); w = (const void*) ((const int8_t*) w + 8);
185*4bdc9457SAndroid Build Coastguard Worker
186*4bdc9457SAndroid Build Coastguard Worker vprod01234567 = vmull_u8(vi1x01234567, vk1x01234567);
187*4bdc9457SAndroid Build Coastguard Worker uint16x8_t vsum01234567 = vaddl_u8(vi0x01234567, vi1x01234567);
188*4bdc9457SAndroid Build Coastguard Worker
189*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc0123), vget_low_u16(vprod01234567)));
190*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc4567), vget_high_u16(vprod01234567)));
191*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vi2x01234567 = vld1_u8(i2); i2 += 8;
192*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vk2x01234567 = vld1_u8(w); w = (const void*) ((const int8_t*) w + 8);
193*4bdc9457SAndroid Build Coastguard Worker
194*4bdc9457SAndroid Build Coastguard Worker vprod01234567 = vmull_u8(vi2x01234567, vk2x01234567);
195*4bdc9457SAndroid Build Coastguard Worker vsum01234567 = vaddw_u8(vsum01234567, vi2x01234567);
196*4bdc9457SAndroid Build Coastguard Worker
197*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc0123), vget_low_u16(vprod01234567)));
198*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc4567), vget_high_u16(vprod01234567)));
199*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vi3x01234567 = vld1_u8(i3); i3 += 8;
200*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vk3x01234567 = vld1_u8(w); w = (const void*) ((const int8_t*) w + 8);
201*4bdc9457SAndroid Build Coastguard Worker
202*4bdc9457SAndroid Build Coastguard Worker vprod01234567 = vmull_u8(vi3x01234567, vk3x01234567);
203*4bdc9457SAndroid Build Coastguard Worker vsum01234567 = vaddw_u8(vsum01234567, vi3x01234567);
204*4bdc9457SAndroid Build Coastguard Worker
205*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc0123), vget_low_u16(vprod01234567)));
206*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc4567), vget_high_u16(vprod01234567)));
207*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vi4x01234567 = vld1_u8(i4); i4 += 8;
208*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vk4x01234567 = vld1_u8(w); w = (const void*) ((const int8_t*) w + 8);
209*4bdc9457SAndroid Build Coastguard Worker
210*4bdc9457SAndroid Build Coastguard Worker vprod01234567 = vmull_u8(vi4x01234567, vk4x01234567);
211*4bdc9457SAndroid Build Coastguard Worker vsum01234567 = vaddw_u8(vsum01234567, vi4x01234567);
212*4bdc9457SAndroid Build Coastguard Worker
213*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc0123), vget_low_u16(vprod01234567)));
214*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc4567), vget_high_u16(vprod01234567)));
215*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vi5x01234567 = vld1_u8(i5); i5 += 8;
216*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vk5x01234567 = vld1_u8(w); w = (const void*) ((const int8_t*) w + 8);
217*4bdc9457SAndroid Build Coastguard Worker
218*4bdc9457SAndroid Build Coastguard Worker vprod01234567 = vmull_u8(vi5x01234567, vk5x01234567);
219*4bdc9457SAndroid Build Coastguard Worker vsum01234567 = vaddw_u8(vsum01234567, vi5x01234567);
220*4bdc9457SAndroid Build Coastguard Worker
221*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc0123), vget_low_u16(vprod01234567)));
222*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc4567), vget_high_u16(vprod01234567)));
223*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vi6x01234567 = vld1_u8(i6); i6 += 8;
224*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vk6x01234567 = vld1_u8(w); w = (const void*) ((const int8_t*) w + 8);
225*4bdc9457SAndroid Build Coastguard Worker
226*4bdc9457SAndroid Build Coastguard Worker vprod01234567 = vmull_u8(vi6x01234567, vk6x01234567);
227*4bdc9457SAndroid Build Coastguard Worker vsum01234567 = vaddw_u8(vsum01234567, vi6x01234567);
228*4bdc9457SAndroid Build Coastguard Worker
229*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc0123), vget_low_u16(vprod01234567)));
230*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc4567), vget_high_u16(vprod01234567)));
231*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vi7x01234567 = vld1_u8(i7); i7 += 8;
232*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vk7x01234567 = vld1_u8(w); w = (const void*) ((const int8_t*) w + 8);
233*4bdc9457SAndroid Build Coastguard Worker
234*4bdc9457SAndroid Build Coastguard Worker vprod01234567 = vmull_u8(vi7x01234567, vk7x01234567);
235*4bdc9457SAndroid Build Coastguard Worker vsum01234567 = vaddw_u8(vsum01234567, vi7x01234567);
236*4bdc9457SAndroid Build Coastguard Worker
237*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc0123), vget_low_u16(vprod01234567)));
238*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc4567), vget_high_u16(vprod01234567)));
239*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vi8x01234567 = vld1_u8(i8); i8 += 8;
240*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vk8x01234567 = vld1_u8(w); w = (const void*) ((const int8_t*) w + 8);
241*4bdc9457SAndroid Build Coastguard Worker
242*4bdc9457SAndroid Build Coastguard Worker vprod01234567 = vmull_u8(vi8x01234567, vk8x01234567);
243*4bdc9457SAndroid Build Coastguard Worker vsum01234567 = vaddw_u8(vsum01234567, vi8x01234567);
244*4bdc9457SAndroid Build Coastguard Worker
245*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc0123), vget_low_u16(vprod01234567)));
246*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc4567), vget_high_u16(vprod01234567)));
247*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vi9x01234567 = vld1_u8(i9); i9 += 8;
248*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vk9x01234567 = vld1_u8(w); w = (const void*) ((const int8_t*) w + 8);
249*4bdc9457SAndroid Build Coastguard Worker
250*4bdc9457SAndroid Build Coastguard Worker vprod01234567 = vmull_u8(vi9x01234567, vk9x01234567);
251*4bdc9457SAndroid Build Coastguard Worker vsum01234567 = vaddw_u8(vsum01234567, vi9x01234567);
252*4bdc9457SAndroid Build Coastguard Worker
253*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc0123), vget_low_u16(vprod01234567)));
254*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc4567), vget_high_u16(vprod01234567)));
255*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vi10x01234567 = vld1_u8(i10); i10 += 8;
256*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vk10x01234567 = vld1_u8(w); w = (const void*) ((const int8_t*) w + 8);
257*4bdc9457SAndroid Build Coastguard Worker
258*4bdc9457SAndroid Build Coastguard Worker vprod01234567 = vmull_u8(vi10x01234567, vk10x01234567);
259*4bdc9457SAndroid Build Coastguard Worker vsum01234567 = vaddw_u8(vsum01234567, vi10x01234567);
260*4bdc9457SAndroid Build Coastguard Worker
261*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc0123), vget_low_u16(vprod01234567)));
262*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc4567), vget_high_u16(vprod01234567)));
263*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vi11x01234567 = vld1_u8(i11); i11 += 8;
264*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vk11x01234567 = vld1_u8(w); w = (const void*) ((const int8_t*) w + 8);
265*4bdc9457SAndroid Build Coastguard Worker
266*4bdc9457SAndroid Build Coastguard Worker vprod01234567 = vmull_u8(vi11x01234567, vk11x01234567);
267*4bdc9457SAndroid Build Coastguard Worker vsum01234567 = vaddw_u8(vsum01234567, vi11x01234567);
268*4bdc9457SAndroid Build Coastguard Worker
269*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc0123), vget_low_u16(vprod01234567)));
270*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc4567), vget_high_u16(vprod01234567)));
271*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vi12x01234567 = vld1_u8(i12); i12 += 8;
272*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vk12x01234567 = vld1_u8(w); w = (const void*) ((const int8_t*) w + 8);
273*4bdc9457SAndroid Build Coastguard Worker
274*4bdc9457SAndroid Build Coastguard Worker vprod01234567 = vmull_u8(vi12x01234567, vk12x01234567);
275*4bdc9457SAndroid Build Coastguard Worker vsum01234567 = vaddw_u8(vsum01234567, vi12x01234567);
276*4bdc9457SAndroid Build Coastguard Worker
277*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc0123), vget_low_u16(vprod01234567)));
278*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc4567), vget_high_u16(vprod01234567)));
279*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vi13x01234567 = vld1_u8(i13); i13 += 8;
280*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vk13x01234567 = vld1_u8(w); w = (const void*) ((const int8_t*) w + 8);
281*4bdc9457SAndroid Build Coastguard Worker
282*4bdc9457SAndroid Build Coastguard Worker vprod01234567 = vmull_u8(vi13x01234567, vk13x01234567);
283*4bdc9457SAndroid Build Coastguard Worker vsum01234567 = vaddw_u8(vsum01234567, vi13x01234567);
284*4bdc9457SAndroid Build Coastguard Worker
285*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc0123), vget_low_u16(vprod01234567)));
286*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc4567), vget_high_u16(vprod01234567)));
287*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vi14x01234567 = vld1_u8(i14); i14 += 8;
288*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vk14x01234567 = vld1_u8(w); w = (const void*) ((const int8_t*) w + 8);
289*4bdc9457SAndroid Build Coastguard Worker
290*4bdc9457SAndroid Build Coastguard Worker vprod01234567 = vmull_u8(vi14x01234567, vk14x01234567);
291*4bdc9457SAndroid Build Coastguard Worker vsum01234567 = vaddw_u8(vsum01234567, vi14x01234567);
292*4bdc9457SAndroid Build Coastguard Worker
293*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc0123), vget_low_u16(vprod01234567)));
294*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc4567), vget_high_u16(vprod01234567)));
295*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vi15x01234567 = vld1_u8(i15); i15 += 8;
296*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vk15x01234567 = vld1_u8(w); w = (const void*) ((const int8_t*) w + 8);
297*4bdc9457SAndroid Build Coastguard Worker
298*4bdc9457SAndroid Build Coastguard Worker vprod01234567 = vmull_u8(vi15x01234567, vk15x01234567);
299*4bdc9457SAndroid Build Coastguard Worker vsum01234567 = vaddw_u8(vsum01234567, vi15x01234567);
300*4bdc9457SAndroid Build Coastguard Worker
301*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc0123), vget_low_u16(vprod01234567)));
302*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc4567), vget_high_u16(vprod01234567)));
303*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vi16x01234567 = vld1_u8(i16); i16 += 8;
304*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vk16x01234567 = vld1_u8(w); w = (const void*) ((const int8_t*) w + 8);
305*4bdc9457SAndroid Build Coastguard Worker
306*4bdc9457SAndroid Build Coastguard Worker vprod01234567 = vmull_u8(vi16x01234567, vk16x01234567);
307*4bdc9457SAndroid Build Coastguard Worker vsum01234567 = vaddw_u8(vsum01234567, vi16x01234567);
308*4bdc9457SAndroid Build Coastguard Worker
309*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc0123), vget_low_u16(vprod01234567)));
310*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc4567), vget_high_u16(vprod01234567)));
311*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vi17x01234567 = vld1_u8(i17); i17 += 8;
312*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vk17x01234567 = vld1_u8(w); w = (const void*) ((const int8_t*) w + 8);
313*4bdc9457SAndroid Build Coastguard Worker
314*4bdc9457SAndroid Build Coastguard Worker vprod01234567 = vmull_u8(vi17x01234567, vk17x01234567);
315*4bdc9457SAndroid Build Coastguard Worker vsum01234567 = vaddw_u8(vsum01234567, vi17x01234567);
316*4bdc9457SAndroid Build Coastguard Worker
317*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc0123), vget_low_u16(vprod01234567)));
318*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc4567), vget_high_u16(vprod01234567)));
319*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vi18x01234567 = vld1_u8(i18); i18 += 8;
320*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vk18x01234567 = vld1_u8(w); w = (const void*) ((const int8_t*) w + 8);
321*4bdc9457SAndroid Build Coastguard Worker
322*4bdc9457SAndroid Build Coastguard Worker vprod01234567 = vmull_u8(vi18x01234567, vk18x01234567);
323*4bdc9457SAndroid Build Coastguard Worker vsum01234567 = vaddw_u8(vsum01234567, vi18x01234567);
324*4bdc9457SAndroid Build Coastguard Worker
325*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc0123), vget_low_u16(vprod01234567)));
326*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc4567), vget_high_u16(vprod01234567)));
327*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vi19x01234567 = vld1_u8(i19); i19 += 8;
328*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vk19x01234567 = vld1_u8(w); w = (const void*) ((const int8_t*) w + 8);
329*4bdc9457SAndroid Build Coastguard Worker
330*4bdc9457SAndroid Build Coastguard Worker vprod01234567 = vmull_u8(vi19x01234567, vk19x01234567);
331*4bdc9457SAndroid Build Coastguard Worker vsum01234567 = vaddw_u8(vsum01234567, vi19x01234567);
332*4bdc9457SAndroid Build Coastguard Worker
333*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc0123), vget_low_u16(vprod01234567)));
334*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc4567), vget_high_u16(vprod01234567)));
335*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vi20x01234567 = vld1_u8(i20); i20 += 8;
336*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vk20x01234567 = vld1_u8(w); w = (const void*) ((const int8_t*) w + 8);
337*4bdc9457SAndroid Build Coastguard Worker
338*4bdc9457SAndroid Build Coastguard Worker vprod01234567 = vmull_u8(vi20x01234567, vk20x01234567);
339*4bdc9457SAndroid Build Coastguard Worker vsum01234567 = vaddw_u8(vsum01234567, vi20x01234567);
340*4bdc9457SAndroid Build Coastguard Worker
341*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc0123), vget_low_u16(vprod01234567)));
342*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc4567), vget_high_u16(vprod01234567)));
343*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vi21x01234567 = vld1_u8(i21); i21 += 8;
344*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vk21x01234567 = vld1_u8(w); w = (const void*) ((const int8_t*) w + 8);
345*4bdc9457SAndroid Build Coastguard Worker
346*4bdc9457SAndroid Build Coastguard Worker vprod01234567 = vmull_u8(vi21x01234567, vk21x01234567);
347*4bdc9457SAndroid Build Coastguard Worker vsum01234567 = vaddw_u8(vsum01234567, vi21x01234567);
348*4bdc9457SAndroid Build Coastguard Worker
349*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc0123), vget_low_u16(vprod01234567)));
350*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc4567), vget_high_u16(vprod01234567)));
351*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vi22x01234567 = vld1_u8(i22); i22 += 8;
352*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vk22x01234567 = vld1_u8(w); w = (const void*) ((const int8_t*) w + 8);
353*4bdc9457SAndroid Build Coastguard Worker
354*4bdc9457SAndroid Build Coastguard Worker vprod01234567 = vmull_u8(vi22x01234567, vk22x01234567);
355*4bdc9457SAndroid Build Coastguard Worker vsum01234567 = vaddw_u8(vsum01234567, vi22x01234567);
356*4bdc9457SAndroid Build Coastguard Worker
357*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc0123), vget_low_u16(vprod01234567)));
358*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc4567), vget_high_u16(vprod01234567)));
359*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vi23x01234567 = vld1_u8(i23); i23 += 8;
360*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vk23x01234567 = vld1_u8(w); w = (const void*) ((const int8_t*) w + 8);
361*4bdc9457SAndroid Build Coastguard Worker
362*4bdc9457SAndroid Build Coastguard Worker vprod01234567 = vmull_u8(vi23x01234567, vk23x01234567);
363*4bdc9457SAndroid Build Coastguard Worker vsum01234567 = vaddw_u8(vsum01234567, vi23x01234567);
364*4bdc9457SAndroid Build Coastguard Worker
365*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc0123), vget_low_u16(vprod01234567)));
366*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc4567), vget_high_u16(vprod01234567)));
367*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vi24x01234567 = vld1_u8(i24); i24 += 8;
368*4bdc9457SAndroid Build Coastguard Worker const uint8x8_t vk24x01234567 = vld1_u8(w); w = (const void*) ((const int8_t*) w + 8);
369*4bdc9457SAndroid Build Coastguard Worker
370*4bdc9457SAndroid Build Coastguard Worker vprod01234567 = vmull_u8(vi24x01234567, vk24x01234567);
371*4bdc9457SAndroid Build Coastguard Worker vsum01234567 = vaddw_u8(vsum01234567, vi24x01234567);
372*4bdc9457SAndroid Build Coastguard Worker
373*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc0123), vget_low_u16(vprod01234567)));
374*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vreinterpretq_s32_u32(vaddw_u16(vreinterpretq_u32_s32(vacc4567), vget_high_u16(vprod01234567)));
375*4bdc9457SAndroid Build Coastguard Worker
376*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vreinterpretq_s32_u32(vmlsl_u16(vreinterpretq_u32_s32(vacc0123), vget_low_u16(vsum01234567), vget_low_u16(vkernel_zero_point16)));
377*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vreinterpretq_s32_u32(vmlsl_u16(vreinterpretq_u32_s32(vacc4567), vget_high_u16(vsum01234567), vget_high_u16(vkernel_zero_point16)));
378*4bdc9457SAndroid Build Coastguard Worker
379*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vshlq_s32(vacc0123, vright_pre_shift);
380*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vshlq_s32(vacc4567, vright_pre_shift);
381*4bdc9457SAndroid Build Coastguard Worker
382*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vqdmulhq_s32(vacc0123, vmultiplier);
383*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vqdmulhq_s32(vacc4567, vmultiplier);
384*4bdc9457SAndroid Build Coastguard Worker
385*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vrshlq_s32(vacc0123, vright_post_shift);
386*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vrshlq_s32(vacc4567, vright_post_shift);
387*4bdc9457SAndroid Build Coastguard Worker
388*4bdc9457SAndroid Build Coastguard Worker #if XNN_ARCH_ARM64
389*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vacc01234567 = vqaddq_s16(vqmovn_high_s32(vqmovn_s32(vacc0123), vacc4567), voutput_zero_point);
390*4bdc9457SAndroid Build Coastguard Worker
391*4bdc9457SAndroid Build Coastguard Worker uint8x8_t vout01234567 = vqmovun_s16(vacc01234567);
392*4bdc9457SAndroid Build Coastguard Worker #else
393*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vacc01234567 = vqaddq_s16(vcombine_s16(vqmovn_s32(vacc0123), vqmovn_s32(vacc4567)), voutput_zero_point);
394*4bdc9457SAndroid Build Coastguard Worker
395*4bdc9457SAndroid Build Coastguard Worker uint8x8_t vout01234567 = vqmovun_s16(vacc01234567);
396*4bdc9457SAndroid Build Coastguard Worker #endif
397*4bdc9457SAndroid Build Coastguard Worker
398*4bdc9457SAndroid Build Coastguard Worker vout01234567 = vmax_u8(vout01234567, voutput_min);
399*4bdc9457SAndroid Build Coastguard Worker
400*4bdc9457SAndroid Build Coastguard Worker vout01234567 = vmin_u8(vout01234567, voutput_max);
401*4bdc9457SAndroid Build Coastguard Worker
402*4bdc9457SAndroid Build Coastguard Worker vst1_u8(output, vout01234567); output += 8;
403*4bdc9457SAndroid Build Coastguard Worker }
404*4bdc9457SAndroid Build Coastguard Worker if XNN_UNLIKELY(c != 0) {
405*4bdc9457SAndroid Build Coastguard Worker {
406*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc0123 = vld1q_s32(w); w = (const void*) ((const int32_t*) w + 4);
407*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc4567 = vld1q_s32(w); w = (const void*) ((const int32_t*) w + 4);
408*4bdc9457SAndroid Build Coastguard Worker
409*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vi0x01234567 = vreinterpretq_s16_u16(vmovl_u8(vld1_u8(i0)));
410*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vk0x01234567 = vreinterpretq_s16_u16(vsubl_u8(vld1_u8(w), vkernel_zero_point));
411*4bdc9457SAndroid Build Coastguard Worker
412*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vmlal_s16(vacc0123, vget_low_s16(vi0x01234567), vget_low_s16(vk0x01234567));
413*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vmlal_s16(vacc4567, vget_high_s16(vi0x01234567), vget_high_s16(vk0x01234567));
414*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vi1x01234567 = vreinterpretq_s16_u16(vmovl_u8(vld1_u8(i1)));
415*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vk1x01234567 = vreinterpretq_s16_u16(vsubl_u8(vld1_u8((const void*) ((const uint8_t*) w + 8)), vkernel_zero_point));
416*4bdc9457SAndroid Build Coastguard Worker
417*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vmlal_s16(vacc0123, vget_low_s16(vi1x01234567), vget_low_s16(vk1x01234567));
418*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vmlal_s16(vacc4567, vget_high_s16(vi1x01234567), vget_high_s16(vk1x01234567));
419*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vi2x01234567 = vreinterpretq_s16_u16(vmovl_u8(vld1_u8(i2)));
420*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vk2x01234567 = vreinterpretq_s16_u16(vsubl_u8(vld1_u8((const void*) ((const uint8_t*) w + 16)), vkernel_zero_point));
421*4bdc9457SAndroid Build Coastguard Worker
422*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vmlal_s16(vacc0123, vget_low_s16(vi2x01234567), vget_low_s16(vk2x01234567));
423*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vmlal_s16(vacc4567, vget_high_s16(vi2x01234567), vget_high_s16(vk2x01234567));
424*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vi3x01234567 = vreinterpretq_s16_u16(vmovl_u8(vld1_u8(i3)));
425*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vk3x01234567 = vreinterpretq_s16_u16(vsubl_u8(vld1_u8((const void*) ((const uint8_t*) w + 24)), vkernel_zero_point));
426*4bdc9457SAndroid Build Coastguard Worker
427*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vmlal_s16(vacc0123, vget_low_s16(vi3x01234567), vget_low_s16(vk3x01234567));
428*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vmlal_s16(vacc4567, vget_high_s16(vi3x01234567), vget_high_s16(vk3x01234567));
429*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vi4x01234567 = vreinterpretq_s16_u16(vmovl_u8(vld1_u8(i4)));
430*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vk4x01234567 = vreinterpretq_s16_u16(vsubl_u8(vld1_u8((const void*) ((const uint8_t*) w + 32)), vkernel_zero_point));
431*4bdc9457SAndroid Build Coastguard Worker
432*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vmlal_s16(vacc0123, vget_low_s16(vi4x01234567), vget_low_s16(vk4x01234567));
433*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vmlal_s16(vacc4567, vget_high_s16(vi4x01234567), vget_high_s16(vk4x01234567));
434*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vi5x01234567 = vreinterpretq_s16_u16(vmovl_u8(vld1_u8(i5)));
435*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vk5x01234567 = vreinterpretq_s16_u16(vsubl_u8(vld1_u8((const void*) ((const uint8_t*) w + 40)), vkernel_zero_point));
436*4bdc9457SAndroid Build Coastguard Worker
437*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vmlal_s16(vacc0123, vget_low_s16(vi5x01234567), vget_low_s16(vk5x01234567));
438*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vmlal_s16(vacc4567, vget_high_s16(vi5x01234567), vget_high_s16(vk5x01234567));
439*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vi6x01234567 = vreinterpretq_s16_u16(vmovl_u8(vld1_u8(i6)));
440*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vk6x01234567 = vreinterpretq_s16_u16(vsubl_u8(vld1_u8((const void*) ((const uint8_t*) w + 48)), vkernel_zero_point));
441*4bdc9457SAndroid Build Coastguard Worker
442*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vmlal_s16(vacc0123, vget_low_s16(vi6x01234567), vget_low_s16(vk6x01234567));
443*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vmlal_s16(vacc4567, vget_high_s16(vi6x01234567), vget_high_s16(vk6x01234567));
444*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vi7x01234567 = vreinterpretq_s16_u16(vmovl_u8(vld1_u8(i7)));
445*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vk7x01234567 = vreinterpretq_s16_u16(vsubl_u8(vld1_u8((const void*) ((const uint8_t*) w + 56)), vkernel_zero_point));
446*4bdc9457SAndroid Build Coastguard Worker
447*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vmlal_s16(vacc0123, vget_low_s16(vi7x01234567), vget_low_s16(vk7x01234567));
448*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vmlal_s16(vacc4567, vget_high_s16(vi7x01234567), vget_high_s16(vk7x01234567));
449*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vi8x01234567 = vreinterpretq_s16_u16(vmovl_u8(vld1_u8(i8)));
450*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vk8x01234567 = vreinterpretq_s16_u16(vsubl_u8(vld1_u8((const void*) ((const uint8_t*) w + 64)), vkernel_zero_point));
451*4bdc9457SAndroid Build Coastguard Worker
452*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vmlal_s16(vacc0123, vget_low_s16(vi8x01234567), vget_low_s16(vk8x01234567));
453*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vmlal_s16(vacc4567, vget_high_s16(vi8x01234567), vget_high_s16(vk8x01234567));
454*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vi9x01234567 = vreinterpretq_s16_u16(vmovl_u8(vld1_u8(i9)));
455*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vk9x01234567 = vreinterpretq_s16_u16(vsubl_u8(vld1_u8((const void*) ((const uint8_t*) w + 72)), vkernel_zero_point));
456*4bdc9457SAndroid Build Coastguard Worker
457*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vmlal_s16(vacc0123, vget_low_s16(vi9x01234567), vget_low_s16(vk9x01234567));
458*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vmlal_s16(vacc4567, vget_high_s16(vi9x01234567), vget_high_s16(vk9x01234567));
459*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vi10x01234567 = vreinterpretq_s16_u16(vmovl_u8(vld1_u8(i10)));
460*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vk10x01234567 = vreinterpretq_s16_u16(vsubl_u8(vld1_u8((const void*) ((const uint8_t*) w + 80)), vkernel_zero_point));
461*4bdc9457SAndroid Build Coastguard Worker
462*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vmlal_s16(vacc0123, vget_low_s16(vi10x01234567), vget_low_s16(vk10x01234567));
463*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vmlal_s16(vacc4567, vget_high_s16(vi10x01234567), vget_high_s16(vk10x01234567));
464*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vi11x01234567 = vreinterpretq_s16_u16(vmovl_u8(vld1_u8(i11)));
465*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vk11x01234567 = vreinterpretq_s16_u16(vsubl_u8(vld1_u8((const void*) ((const uint8_t*) w + 88)), vkernel_zero_point));
466*4bdc9457SAndroid Build Coastguard Worker
467*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vmlal_s16(vacc0123, vget_low_s16(vi11x01234567), vget_low_s16(vk11x01234567));
468*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vmlal_s16(vacc4567, vget_high_s16(vi11x01234567), vget_high_s16(vk11x01234567));
469*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vi12x01234567 = vreinterpretq_s16_u16(vmovl_u8(vld1_u8(i12)));
470*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vk12x01234567 = vreinterpretq_s16_u16(vsubl_u8(vld1_u8((const void*) ((const uint8_t*) w + 96)), vkernel_zero_point));
471*4bdc9457SAndroid Build Coastguard Worker
472*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vmlal_s16(vacc0123, vget_low_s16(vi12x01234567), vget_low_s16(vk12x01234567));
473*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vmlal_s16(vacc4567, vget_high_s16(vi12x01234567), vget_high_s16(vk12x01234567));
474*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vi13x01234567 = vreinterpretq_s16_u16(vmovl_u8(vld1_u8(i13)));
475*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vk13x01234567 = vreinterpretq_s16_u16(vsubl_u8(vld1_u8((const void*) ((const uint8_t*) w + 104)), vkernel_zero_point));
476*4bdc9457SAndroid Build Coastguard Worker
477*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vmlal_s16(vacc0123, vget_low_s16(vi13x01234567), vget_low_s16(vk13x01234567));
478*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vmlal_s16(vacc4567, vget_high_s16(vi13x01234567), vget_high_s16(vk13x01234567));
479*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vi14x01234567 = vreinterpretq_s16_u16(vmovl_u8(vld1_u8(i14)));
480*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vk14x01234567 = vreinterpretq_s16_u16(vsubl_u8(vld1_u8((const void*) ((const uint8_t*) w + 112)), vkernel_zero_point));
481*4bdc9457SAndroid Build Coastguard Worker
482*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vmlal_s16(vacc0123, vget_low_s16(vi14x01234567), vget_low_s16(vk14x01234567));
483*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vmlal_s16(vacc4567, vget_high_s16(vi14x01234567), vget_high_s16(vk14x01234567));
484*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vi15x01234567 = vreinterpretq_s16_u16(vmovl_u8(vld1_u8(i15)));
485*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vk15x01234567 = vreinterpretq_s16_u16(vsubl_u8(vld1_u8((const void*) ((const uint8_t*) w + 120)), vkernel_zero_point));
486*4bdc9457SAndroid Build Coastguard Worker
487*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vmlal_s16(vacc0123, vget_low_s16(vi15x01234567), vget_low_s16(vk15x01234567));
488*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vmlal_s16(vacc4567, vget_high_s16(vi15x01234567), vget_high_s16(vk15x01234567));
489*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vi16x01234567 = vreinterpretq_s16_u16(vmovl_u8(vld1_u8(i16)));
490*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vk16x01234567 = vreinterpretq_s16_u16(vsubl_u8(vld1_u8((const void*) ((const uint8_t*) w + 128)), vkernel_zero_point));
491*4bdc9457SAndroid Build Coastguard Worker
492*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vmlal_s16(vacc0123, vget_low_s16(vi16x01234567), vget_low_s16(vk16x01234567));
493*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vmlal_s16(vacc4567, vget_high_s16(vi16x01234567), vget_high_s16(vk16x01234567));
494*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vi17x01234567 = vreinterpretq_s16_u16(vmovl_u8(vld1_u8(i17)));
495*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vk17x01234567 = vreinterpretq_s16_u16(vsubl_u8(vld1_u8((const void*) ((const uint8_t*) w + 136)), vkernel_zero_point));
496*4bdc9457SAndroid Build Coastguard Worker
497*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vmlal_s16(vacc0123, vget_low_s16(vi17x01234567), vget_low_s16(vk17x01234567));
498*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vmlal_s16(vacc4567, vget_high_s16(vi17x01234567), vget_high_s16(vk17x01234567));
499*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vi18x01234567 = vreinterpretq_s16_u16(vmovl_u8(vld1_u8(i18)));
500*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vk18x01234567 = vreinterpretq_s16_u16(vsubl_u8(vld1_u8((const void*) ((const uint8_t*) w + 144)), vkernel_zero_point));
501*4bdc9457SAndroid Build Coastguard Worker
502*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vmlal_s16(vacc0123, vget_low_s16(vi18x01234567), vget_low_s16(vk18x01234567));
503*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vmlal_s16(vacc4567, vget_high_s16(vi18x01234567), vget_high_s16(vk18x01234567));
504*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vi19x01234567 = vreinterpretq_s16_u16(vmovl_u8(vld1_u8(i19)));
505*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vk19x01234567 = vreinterpretq_s16_u16(vsubl_u8(vld1_u8((const void*) ((const uint8_t*) w + 152)), vkernel_zero_point));
506*4bdc9457SAndroid Build Coastguard Worker
507*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vmlal_s16(vacc0123, vget_low_s16(vi19x01234567), vget_low_s16(vk19x01234567));
508*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vmlal_s16(vacc4567, vget_high_s16(vi19x01234567), vget_high_s16(vk19x01234567));
509*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vi20x01234567 = vreinterpretq_s16_u16(vmovl_u8(vld1_u8(i20)));
510*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vk20x01234567 = vreinterpretq_s16_u16(vsubl_u8(vld1_u8((const void*) ((const uint8_t*) w + 160)), vkernel_zero_point));
511*4bdc9457SAndroid Build Coastguard Worker
512*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vmlal_s16(vacc0123, vget_low_s16(vi20x01234567), vget_low_s16(vk20x01234567));
513*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vmlal_s16(vacc4567, vget_high_s16(vi20x01234567), vget_high_s16(vk20x01234567));
514*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vi21x01234567 = vreinterpretq_s16_u16(vmovl_u8(vld1_u8(i21)));
515*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vk21x01234567 = vreinterpretq_s16_u16(vsubl_u8(vld1_u8((const void*) ((const uint8_t*) w + 168)), vkernel_zero_point));
516*4bdc9457SAndroid Build Coastguard Worker
517*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vmlal_s16(vacc0123, vget_low_s16(vi21x01234567), vget_low_s16(vk21x01234567));
518*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vmlal_s16(vacc4567, vget_high_s16(vi21x01234567), vget_high_s16(vk21x01234567));
519*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vi22x01234567 = vreinterpretq_s16_u16(vmovl_u8(vld1_u8(i22)));
520*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vk22x01234567 = vreinterpretq_s16_u16(vsubl_u8(vld1_u8((const void*) ((const uint8_t*) w + 176)), vkernel_zero_point));
521*4bdc9457SAndroid Build Coastguard Worker
522*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vmlal_s16(vacc0123, vget_low_s16(vi22x01234567), vget_low_s16(vk22x01234567));
523*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vmlal_s16(vacc4567, vget_high_s16(vi22x01234567), vget_high_s16(vk22x01234567));
524*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vi23x01234567 = vreinterpretq_s16_u16(vmovl_u8(vld1_u8(i23)));
525*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vk23x01234567 = vreinterpretq_s16_u16(vsubl_u8(vld1_u8((const void*) ((const uint8_t*) w + 184)), vkernel_zero_point));
526*4bdc9457SAndroid Build Coastguard Worker
527*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vmlal_s16(vacc0123, vget_low_s16(vi23x01234567), vget_low_s16(vk23x01234567));
528*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vmlal_s16(vacc4567, vget_high_s16(vi23x01234567), vget_high_s16(vk23x01234567));
529*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vi24x01234567 = vreinterpretq_s16_u16(vmovl_u8(vld1_u8(i24)));
530*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vk24x01234567 = vreinterpretq_s16_u16(vsubl_u8(vld1_u8((const void*) ((const uint8_t*) w + 192)), vkernel_zero_point));
531*4bdc9457SAndroid Build Coastguard Worker
532*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vmlal_s16(vacc0123, vget_low_s16(vi24x01234567), vget_low_s16(vk24x01234567));
533*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vmlal_s16(vacc4567, vget_high_s16(vi24x01234567), vget_high_s16(vk24x01234567));
534*4bdc9457SAndroid Build Coastguard Worker
535*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vrshlq_s32(vacc0123, vright_pre_shift);
536*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vrshlq_s32(vacc4567, vright_pre_shift);
537*4bdc9457SAndroid Build Coastguard Worker
538*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vqdmulhq_s32(vacc0123, vmultiplier);
539*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vqdmulhq_s32(vacc4567, vmultiplier);
540*4bdc9457SAndroid Build Coastguard Worker
541*4bdc9457SAndroid Build Coastguard Worker vacc0123 = vrshlq_s32(vacc0123, vright_post_shift);
542*4bdc9457SAndroid Build Coastguard Worker vacc4567 = vrshlq_s32(vacc4567, vright_post_shift);
543*4bdc9457SAndroid Build Coastguard Worker
544*4bdc9457SAndroid Build Coastguard Worker #if XNN_ARCH_ARM64
545*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vacc01234567 = vqaddq_s16(vqmovn_high_s32(vqmovn_s32(vacc0123), vacc4567), voutput_zero_point);
546*4bdc9457SAndroid Build Coastguard Worker uint8x8_t vout01234567 = vqmovun_s16(vacc01234567);
547*4bdc9457SAndroid Build Coastguard Worker #else
548*4bdc9457SAndroid Build Coastguard Worker const int16x8_t vacc01234567 = vqaddq_s16(vcombine_s16(vqmovn_s32(vacc0123), vqmovn_s32(vacc4567)), voutput_zero_point);
549*4bdc9457SAndroid Build Coastguard Worker uint8x8_t vout01234567 = vqmovun_s16(vacc01234567);
550*4bdc9457SAndroid Build Coastguard Worker #endif
551*4bdc9457SAndroid Build Coastguard Worker
552*4bdc9457SAndroid Build Coastguard Worker vout01234567 = vmax_u8(vout01234567, voutput_min);
553*4bdc9457SAndroid Build Coastguard Worker vout01234567 = vmin_u8(vout01234567, voutput_max);
554*4bdc9457SAndroid Build Coastguard Worker
555*4bdc9457SAndroid Build Coastguard Worker if (c & 4) {
556*4bdc9457SAndroid Build Coastguard Worker vst1_lane_u32((void*) output, vreinterpret_u32_u8(vout01234567), 0); output += 4;
557*4bdc9457SAndroid Build Coastguard Worker vout01234567 = vext_u8(vout01234567, vout01234567, 4);
558*4bdc9457SAndroid Build Coastguard Worker }
559*4bdc9457SAndroid Build Coastguard Worker if (c & 2) {
560*4bdc9457SAndroid Build Coastguard Worker vst1_lane_u16((void*) output, vreinterpret_u16_u8(vout01234567), 0); output += 2;
561*4bdc9457SAndroid Build Coastguard Worker vout01234567 = vext_u8(vout01234567, vout01234567, 2);
562*4bdc9457SAndroid Build Coastguard Worker }
563*4bdc9457SAndroid Build Coastguard Worker if (c & 1) {
564*4bdc9457SAndroid Build Coastguard Worker vst1_lane_u8(output, vout01234567, 0); output += 1;
565*4bdc9457SAndroid Build Coastguard Worker }
566*4bdc9457SAndroid Build Coastguard Worker }
567*4bdc9457SAndroid Build Coastguard Worker }
568*4bdc9457SAndroid Build Coastguard Worker
569*4bdc9457SAndroid Build Coastguard Worker output = (uint8_t*) ((uintptr_t) output + output_increment);
570*4bdc9457SAndroid Build Coastguard Worker } while (--output_width != 0);
571*4bdc9457SAndroid Build Coastguard Worker }
572