1*4bdc9457SAndroid Build Coastguard Worker // Auto-generated file. Do not edit!
2*4bdc9457SAndroid Build Coastguard Worker // Template: src/qs8-igemm/c4-neon-mull-shuffle.c.in
3*4bdc9457SAndroid Build Coastguard Worker // Generator: tools/xngen
4*4bdc9457SAndroid Build Coastguard Worker //
5*4bdc9457SAndroid Build Coastguard Worker // Copyright 2021 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/gemm.h>
15*4bdc9457SAndroid Build Coastguard Worker #include <xnnpack/math.h>
16*4bdc9457SAndroid Build Coastguard Worker
17*4bdc9457SAndroid Build Coastguard Worker
xnn_qs8_igemm_minmax_rndnu_ukernel_3x16c4s2__neon_mlal(size_t mr,size_t nc,size_t kc,size_t ks,const int8_t ** restrict a,const void * restrict w,int8_t * restrict c,size_t cm_stride,size_t cn_stride,size_t a_offset,const int8_t * zero,const union xnn_qs8_conv_minmax_params params[restrict XNN_MIN_ELEMENTS (1)])18*4bdc9457SAndroid Build Coastguard Worker void xnn_qs8_igemm_minmax_rndnu_ukernel_3x16c4s2__neon_mlal(
19*4bdc9457SAndroid Build Coastguard Worker size_t mr,
20*4bdc9457SAndroid Build Coastguard Worker size_t nc,
21*4bdc9457SAndroid Build Coastguard Worker size_t kc,
22*4bdc9457SAndroid Build Coastguard Worker size_t ks,
23*4bdc9457SAndroid Build Coastguard Worker const int8_t** restrict a,
24*4bdc9457SAndroid Build Coastguard Worker const void* restrict w,
25*4bdc9457SAndroid Build Coastguard Worker int8_t* restrict c,
26*4bdc9457SAndroid Build Coastguard Worker size_t cm_stride,
27*4bdc9457SAndroid Build Coastguard Worker size_t cn_stride,
28*4bdc9457SAndroid Build Coastguard Worker size_t a_offset,
29*4bdc9457SAndroid Build Coastguard Worker const int8_t* zero,
30*4bdc9457SAndroid Build Coastguard Worker const union xnn_qs8_conv_minmax_params params[restrict XNN_MIN_ELEMENTS(1)]) XNN_OOB_READS
31*4bdc9457SAndroid Build Coastguard Worker {
32*4bdc9457SAndroid Build Coastguard Worker assert(mr != 0);
33*4bdc9457SAndroid Build Coastguard Worker assert(mr <= 3);
34*4bdc9457SAndroid Build Coastguard Worker assert(nc != 0);
35*4bdc9457SAndroid Build Coastguard Worker assert(kc != 0);
36*4bdc9457SAndroid Build Coastguard Worker assert(ks != 0);
37*4bdc9457SAndroid Build Coastguard Worker assert(ks % (3 * sizeof(void*)) == 0);
38*4bdc9457SAndroid Build Coastguard Worker assert(a_offset % sizeof(int8_t) == 0);
39*4bdc9457SAndroid Build Coastguard Worker assert(a != NULL);
40*4bdc9457SAndroid Build Coastguard Worker assert(w != NULL);
41*4bdc9457SAndroid Build Coastguard Worker assert(c != NULL);
42*4bdc9457SAndroid Build Coastguard Worker
43*4bdc9457SAndroid Build Coastguard Worker int8_t* c0 = c;
44*4bdc9457SAndroid Build Coastguard Worker int8_t* c1 = (int8_t*) ((uintptr_t) c0 + cm_stride);
45*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(mr < 2) {
46*4bdc9457SAndroid Build Coastguard Worker c1 = c0;
47*4bdc9457SAndroid Build Coastguard Worker }
48*4bdc9457SAndroid Build Coastguard Worker int8_t* c2 = (int8_t*) ((uintptr_t) c1 + cm_stride);
49*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(mr <= 2) {
50*4bdc9457SAndroid Build Coastguard Worker c2 = c1;
51*4bdc9457SAndroid Build Coastguard Worker }
52*4bdc9457SAndroid Build Coastguard Worker
53*4bdc9457SAndroid Build Coastguard Worker kc = round_up_po2(kc, 8 * sizeof(int8_t));
54*4bdc9457SAndroid Build Coastguard Worker do {
55*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc0x01 = vreinterpretq_s32_u64(vmovl_u32(vld1_u32(w))); w = (const int32_t*) w + 2;
56*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc0x23 = vreinterpretq_s32_u64(vmovl_u32(vld1_u32(w))); w = (const int32_t*) w + 2;
57*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc0x45 = vreinterpretq_s32_u64(vmovl_u32(vld1_u32(w))); w = (const int32_t*) w + 2;
58*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc0x67 = vreinterpretq_s32_u64(vmovl_u32(vld1_u32(w))); w = (const int32_t*) w + 2;
59*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc0x89 = vreinterpretq_s32_u64(vmovl_u32(vld1_u32(w))); w = (const int32_t*) w + 2;
60*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc0xAB = vreinterpretq_s32_u64(vmovl_u32(vld1_u32(w))); w = (const int32_t*) w + 2;
61*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc0xCD = vreinterpretq_s32_u64(vmovl_u32(vld1_u32(w))); w = (const int32_t*) w + 2;
62*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc0xEF = vreinterpretq_s32_u64(vmovl_u32(vld1_u32(w))); w = (const int32_t*) w + 2;
63*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc1x01 = vacc0x01;
64*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc1x23 = vacc0x23;
65*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc1x45 = vacc0x45;
66*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc1x67 = vacc0x67;
67*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc1x89 = vacc0x89;
68*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc1xAB = vacc0xAB;
69*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc1xCD = vacc0xCD;
70*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc1xEF = vacc0xEF;
71*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc2x01 = vacc0x01;
72*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc2x23 = vacc0x23;
73*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc2x45 = vacc0x45;
74*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc2x67 = vacc0x67;
75*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc2x89 = vacc0x89;
76*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc2xAB = vacc0xAB;
77*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc2xCD = vacc0xCD;
78*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc2xEF = vacc0xEF;
79*4bdc9457SAndroid Build Coastguard Worker
80*4bdc9457SAndroid Build Coastguard Worker size_t p = ks;
81*4bdc9457SAndroid Build Coastguard Worker do {
82*4bdc9457SAndroid Build Coastguard Worker const int8_t* restrict a0 = a[0];
83*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(a0 != zero) {
84*4bdc9457SAndroid Build Coastguard Worker a0 = (const int8_t*) ((uintptr_t) a0 + a_offset);
85*4bdc9457SAndroid Build Coastguard Worker }
86*4bdc9457SAndroid Build Coastguard Worker const int8_t* restrict a1 = a[1];
87*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(a1 != zero) {
88*4bdc9457SAndroid Build Coastguard Worker a1 = (const int8_t*) ((uintptr_t) a1 + a_offset);
89*4bdc9457SAndroid Build Coastguard Worker }
90*4bdc9457SAndroid Build Coastguard Worker const int8_t* restrict a2 = a[2];
91*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(a2 != zero) {
92*4bdc9457SAndroid Build Coastguard Worker a2 = (const int8_t*) ((uintptr_t) a2 + a_offset);
93*4bdc9457SAndroid Build Coastguard Worker }
94*4bdc9457SAndroid Build Coastguard Worker a += 3;
95*4bdc9457SAndroid Build Coastguard Worker
96*4bdc9457SAndroid Build Coastguard Worker size_t k = kc;
97*4bdc9457SAndroid Build Coastguard Worker while (k >= 16 * sizeof(int8_t)) {
98*4bdc9457SAndroid Build Coastguard Worker int8x8_t va0x0 = vld1_s8(a0); a0 += 8;
99*4bdc9457SAndroid Build Coastguard Worker int8x8_t va0x1 = vld1_s8(a0); a0 += 8;
100*4bdc9457SAndroid Build Coastguard Worker int8x8_t va1x0 = vld1_s8(a1); a1 += 8;
101*4bdc9457SAndroid Build Coastguard Worker int8x8_t va1x1 = vld1_s8(a1); a1 += 8;
102*4bdc9457SAndroid Build Coastguard Worker int8x8_t va2x0 = vld1_s8(a2); a2 += 8;
103*4bdc9457SAndroid Build Coastguard Worker int8x8_t va2x1 = vld1_s8(a2); a2 += 8;
104*4bdc9457SAndroid Build Coastguard Worker
105*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb01c0x0 = vld1_s8(w); w = (const int8_t*) w + 8;
106*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb23c0x0 = vld1_s8(w); w = (const int8_t*) w + 8;
107*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb45c0x0 = vld1_s8(w); w = (const int8_t*) w + 8;
108*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb67c0x0 = vld1_s8(w); w = (const int8_t*) w + 8;
109*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb89c0x0 = vld1_s8(w); w = (const int8_t*) w + 8;
110*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vbABc0x0 = vld1_s8(w); w = (const int8_t*) w + 8;
111*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vbCDc0x0 = vld1_s8(w); w = (const int8_t*) w + 8;
112*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vbEFc0x0 = vld1_s8(w); w = (const int8_t*) w + 8;
113*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb01c1x0 = vld1_s8(w); w = (const int8_t*) w + 8;
114*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb23c1x0 = vld1_s8(w); w = (const int8_t*) w + 8;
115*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb45c1x0 = vld1_s8(w); w = (const int8_t*) w + 8;
116*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb67c1x0 = vld1_s8(w); w = (const int8_t*) w + 8;
117*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb89c1x0 = vld1_s8(w); w = (const int8_t*) w + 8;
118*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vbABc1x0 = vld1_s8(w); w = (const int8_t*) w + 8;
119*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vbCDc1x0 = vld1_s8(w); w = (const int8_t*) w + 8;
120*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vbEFc1x0 = vld1_s8(w); w = (const int8_t*) w + 8;
121*4bdc9457SAndroid Build Coastguard Worker
122*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0x01c0 = vmull_s8(vb01c0x0, va0x0);
123*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1x01c0 = vmull_s8(vb01c0x0, va1x0);
124*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2x01c0 = vmull_s8(vb01c0x0, va2x0);
125*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb01c0x1 = vld1_s8(w); w = (const int8_t*) w + 8;
126*4bdc9457SAndroid Build Coastguard Worker vprod0x01c0 = vmlal_s8(vprod0x01c0, vb01c0x1, va0x1);
127*4bdc9457SAndroid Build Coastguard Worker vprod1x01c0 = vmlal_s8(vprod1x01c0, vb01c0x1, va1x1);
128*4bdc9457SAndroid Build Coastguard Worker vprod2x01c0 = vmlal_s8(vprod2x01c0, vb01c0x1, va2x1);
129*4bdc9457SAndroid Build Coastguard Worker vacc0x01 = vpadalq_s16(vacc0x01, vprod0x01c0);
130*4bdc9457SAndroid Build Coastguard Worker vacc1x01 = vpadalq_s16(vacc1x01, vprod1x01c0);
131*4bdc9457SAndroid Build Coastguard Worker vacc2x01 = vpadalq_s16(vacc2x01, vprod2x01c0);
132*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0x23c0 = vmull_s8(vb23c0x0, va0x0);
133*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1x23c0 = vmull_s8(vb23c0x0, va1x0);
134*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2x23c0 = vmull_s8(vb23c0x0, va2x0);
135*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb23c0x1 = vld1_s8(w); w = (const int8_t*) w + 8;
136*4bdc9457SAndroid Build Coastguard Worker vprod0x23c0 = vmlal_s8(vprod0x23c0, vb23c0x1, va0x1);
137*4bdc9457SAndroid Build Coastguard Worker vprod1x23c0 = vmlal_s8(vprod1x23c0, vb23c0x1, va1x1);
138*4bdc9457SAndroid Build Coastguard Worker vprod2x23c0 = vmlal_s8(vprod2x23c0, vb23c0x1, va2x1);
139*4bdc9457SAndroid Build Coastguard Worker vacc0x23 = vpadalq_s16(vacc0x23, vprod0x23c0);
140*4bdc9457SAndroid Build Coastguard Worker vacc1x23 = vpadalq_s16(vacc1x23, vprod1x23c0);
141*4bdc9457SAndroid Build Coastguard Worker vacc2x23 = vpadalq_s16(vacc2x23, vprod2x23c0);
142*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0x45c0 = vmull_s8(vb45c0x0, va0x0);
143*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1x45c0 = vmull_s8(vb45c0x0, va1x0);
144*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2x45c0 = vmull_s8(vb45c0x0, va2x0);
145*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb45c0x1 = vld1_s8(w); w = (const int8_t*) w + 8;
146*4bdc9457SAndroid Build Coastguard Worker vprod0x45c0 = vmlal_s8(vprod0x45c0, vb45c0x1, va0x1);
147*4bdc9457SAndroid Build Coastguard Worker vprod1x45c0 = vmlal_s8(vprod1x45c0, vb45c0x1, va1x1);
148*4bdc9457SAndroid Build Coastguard Worker vprod2x45c0 = vmlal_s8(vprod2x45c0, vb45c0x1, va2x1);
149*4bdc9457SAndroid Build Coastguard Worker vacc0x45 = vpadalq_s16(vacc0x45, vprod0x45c0);
150*4bdc9457SAndroid Build Coastguard Worker vacc1x45 = vpadalq_s16(vacc1x45, vprod1x45c0);
151*4bdc9457SAndroid Build Coastguard Worker vacc2x45 = vpadalq_s16(vacc2x45, vprod2x45c0);
152*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0x67c0 = vmull_s8(vb67c0x0, va0x0);
153*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1x67c0 = vmull_s8(vb67c0x0, va1x0);
154*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2x67c0 = vmull_s8(vb67c0x0, va2x0);
155*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb67c0x1 = vld1_s8(w); w = (const int8_t*) w + 8;
156*4bdc9457SAndroid Build Coastguard Worker vprod0x67c0 = vmlal_s8(vprod0x67c0, vb67c0x1, va0x1);
157*4bdc9457SAndroid Build Coastguard Worker vprod1x67c0 = vmlal_s8(vprod1x67c0, vb67c0x1, va1x1);
158*4bdc9457SAndroid Build Coastguard Worker vprod2x67c0 = vmlal_s8(vprod2x67c0, vb67c0x1, va2x1);
159*4bdc9457SAndroid Build Coastguard Worker vacc0x67 = vpadalq_s16(vacc0x67, vprod0x67c0);
160*4bdc9457SAndroid Build Coastguard Worker vacc1x67 = vpadalq_s16(vacc1x67, vprod1x67c0);
161*4bdc9457SAndroid Build Coastguard Worker vacc2x67 = vpadalq_s16(vacc2x67, vprod2x67c0);
162*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0x89c0 = vmull_s8(vb89c0x0, va0x0);
163*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1x89c0 = vmull_s8(vb89c0x0, va1x0);
164*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2x89c0 = vmull_s8(vb89c0x0, va2x0);
165*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb89c0x1 = vld1_s8(w); w = (const int8_t*) w + 8;
166*4bdc9457SAndroid Build Coastguard Worker vprod0x89c0 = vmlal_s8(vprod0x89c0, vb89c0x1, va0x1);
167*4bdc9457SAndroid Build Coastguard Worker vprod1x89c0 = vmlal_s8(vprod1x89c0, vb89c0x1, va1x1);
168*4bdc9457SAndroid Build Coastguard Worker vprod2x89c0 = vmlal_s8(vprod2x89c0, vb89c0x1, va2x1);
169*4bdc9457SAndroid Build Coastguard Worker vacc0x89 = vpadalq_s16(vacc0x89, vprod0x89c0);
170*4bdc9457SAndroid Build Coastguard Worker vacc1x89 = vpadalq_s16(vacc1x89, vprod1x89c0);
171*4bdc9457SAndroid Build Coastguard Worker vacc2x89 = vpadalq_s16(vacc2x89, vprod2x89c0);
172*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0xABc0 = vmull_s8(vbABc0x0, va0x0);
173*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1xABc0 = vmull_s8(vbABc0x0, va1x0);
174*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2xABc0 = vmull_s8(vbABc0x0, va2x0);
175*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vbABc0x1 = vld1_s8(w); w = (const int8_t*) w + 8;
176*4bdc9457SAndroid Build Coastguard Worker vprod0xABc0 = vmlal_s8(vprod0xABc0, vbABc0x1, va0x1);
177*4bdc9457SAndroid Build Coastguard Worker vprod1xABc0 = vmlal_s8(vprod1xABc0, vbABc0x1, va1x1);
178*4bdc9457SAndroid Build Coastguard Worker vprod2xABc0 = vmlal_s8(vprod2xABc0, vbABc0x1, va2x1);
179*4bdc9457SAndroid Build Coastguard Worker vacc0xAB = vpadalq_s16(vacc0xAB, vprod0xABc0);
180*4bdc9457SAndroid Build Coastguard Worker vacc1xAB = vpadalq_s16(vacc1xAB, vprod1xABc0);
181*4bdc9457SAndroid Build Coastguard Worker vacc2xAB = vpadalq_s16(vacc2xAB, vprod2xABc0);
182*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0xCDc0 = vmull_s8(vbCDc0x0, va0x0);
183*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1xCDc0 = vmull_s8(vbCDc0x0, va1x0);
184*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2xCDc0 = vmull_s8(vbCDc0x0, va2x0);
185*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vbCDc0x1 = vld1_s8(w); w = (const int8_t*) w + 8;
186*4bdc9457SAndroid Build Coastguard Worker vprod0xCDc0 = vmlal_s8(vprod0xCDc0, vbCDc0x1, va0x1);
187*4bdc9457SAndroid Build Coastguard Worker vprod1xCDc0 = vmlal_s8(vprod1xCDc0, vbCDc0x1, va1x1);
188*4bdc9457SAndroid Build Coastguard Worker vprod2xCDc0 = vmlal_s8(vprod2xCDc0, vbCDc0x1, va2x1);
189*4bdc9457SAndroid Build Coastguard Worker vacc0xCD = vpadalq_s16(vacc0xCD, vprod0xCDc0);
190*4bdc9457SAndroid Build Coastguard Worker vacc1xCD = vpadalq_s16(vacc1xCD, vprod1xCDc0);
191*4bdc9457SAndroid Build Coastguard Worker vacc2xCD = vpadalq_s16(vacc2xCD, vprod2xCDc0);
192*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0xEFc0 = vmull_s8(vbEFc0x0, va0x0);
193*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1xEFc0 = vmull_s8(vbEFc0x0, va1x0);
194*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2xEFc0 = vmull_s8(vbEFc0x0, va2x0);
195*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vbEFc0x1 = vld1_s8(w); w = (const int8_t*) w + 8;
196*4bdc9457SAndroid Build Coastguard Worker vprod0xEFc0 = vmlal_s8(vprod0xEFc0, vbEFc0x1, va0x1);
197*4bdc9457SAndroid Build Coastguard Worker vprod1xEFc0 = vmlal_s8(vprod1xEFc0, vbEFc0x1, va1x1);
198*4bdc9457SAndroid Build Coastguard Worker vprod2xEFc0 = vmlal_s8(vprod2xEFc0, vbEFc0x1, va2x1);
199*4bdc9457SAndroid Build Coastguard Worker vacc0xEF = vpadalq_s16(vacc0xEF, vprod0xEFc0);
200*4bdc9457SAndroid Build Coastguard Worker vacc1xEF = vpadalq_s16(vacc1xEF, vprod1xEFc0);
201*4bdc9457SAndroid Build Coastguard Worker vacc2xEF = vpadalq_s16(vacc2xEF, vprod2xEFc0);
202*4bdc9457SAndroid Build Coastguard Worker va0x0 = vext_s8(va0x0, va0x0, 4);
203*4bdc9457SAndroid Build Coastguard Worker va0x1 = vext_s8(va0x1, va0x1, 4);
204*4bdc9457SAndroid Build Coastguard Worker va1x0 = vext_s8(va1x0, va1x0, 4);
205*4bdc9457SAndroid Build Coastguard Worker va1x1 = vext_s8(va1x1, va1x1, 4);
206*4bdc9457SAndroid Build Coastguard Worker va2x0 = vext_s8(va2x0, va2x0, 4);
207*4bdc9457SAndroid Build Coastguard Worker va2x1 = vext_s8(va2x1, va2x1, 4);
208*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0x01c1 = vmull_s8(vb01c1x0, va0x0);
209*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1x01c1 = vmull_s8(vb01c1x0, va1x0);
210*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2x01c1 = vmull_s8(vb01c1x0, va2x0);
211*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb01c1x1 = vld1_s8(w); w = (const int8_t*) w + 8;
212*4bdc9457SAndroid Build Coastguard Worker vprod0x01c1 = vmlal_s8(vprod0x01c1, vb01c1x1, va0x1);
213*4bdc9457SAndroid Build Coastguard Worker vprod1x01c1 = vmlal_s8(vprod1x01c1, vb01c1x1, va1x1);
214*4bdc9457SAndroid Build Coastguard Worker vprod2x01c1 = vmlal_s8(vprod2x01c1, vb01c1x1, va2x1);
215*4bdc9457SAndroid Build Coastguard Worker vacc0x01 = vpadalq_s16(vacc0x01, vprod0x01c1);
216*4bdc9457SAndroid Build Coastguard Worker vacc1x01 = vpadalq_s16(vacc1x01, vprod1x01c1);
217*4bdc9457SAndroid Build Coastguard Worker vacc2x01 = vpadalq_s16(vacc2x01, vprod2x01c1);
218*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0x23c1 = vmull_s8(vb23c1x0, va0x0);
219*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1x23c1 = vmull_s8(vb23c1x0, va1x0);
220*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2x23c1 = vmull_s8(vb23c1x0, va2x0);
221*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb23c1x1 = vld1_s8(w); w = (const int8_t*) w + 8;
222*4bdc9457SAndroid Build Coastguard Worker vprod0x23c1 = vmlal_s8(vprod0x23c1, vb23c1x1, va0x1);
223*4bdc9457SAndroid Build Coastguard Worker vprod1x23c1 = vmlal_s8(vprod1x23c1, vb23c1x1, va1x1);
224*4bdc9457SAndroid Build Coastguard Worker vprod2x23c1 = vmlal_s8(vprod2x23c1, vb23c1x1, va2x1);
225*4bdc9457SAndroid Build Coastguard Worker vacc0x23 = vpadalq_s16(vacc0x23, vprod0x23c1);
226*4bdc9457SAndroid Build Coastguard Worker vacc1x23 = vpadalq_s16(vacc1x23, vprod1x23c1);
227*4bdc9457SAndroid Build Coastguard Worker vacc2x23 = vpadalq_s16(vacc2x23, vprod2x23c1);
228*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0x45c1 = vmull_s8(vb45c1x0, va0x0);
229*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1x45c1 = vmull_s8(vb45c1x0, va1x0);
230*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2x45c1 = vmull_s8(vb45c1x0, va2x0);
231*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb45c1x1 = vld1_s8(w); w = (const int8_t*) w + 8;
232*4bdc9457SAndroid Build Coastguard Worker vprod0x45c1 = vmlal_s8(vprod0x45c1, vb45c1x1, va0x1);
233*4bdc9457SAndroid Build Coastguard Worker vprod1x45c1 = vmlal_s8(vprod1x45c1, vb45c1x1, va1x1);
234*4bdc9457SAndroid Build Coastguard Worker vprod2x45c1 = vmlal_s8(vprod2x45c1, vb45c1x1, va2x1);
235*4bdc9457SAndroid Build Coastguard Worker vacc0x45 = vpadalq_s16(vacc0x45, vprod0x45c1);
236*4bdc9457SAndroid Build Coastguard Worker vacc1x45 = vpadalq_s16(vacc1x45, vprod1x45c1);
237*4bdc9457SAndroid Build Coastguard Worker vacc2x45 = vpadalq_s16(vacc2x45, vprod2x45c1);
238*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0x67c1 = vmull_s8(vb67c1x0, va0x0);
239*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1x67c1 = vmull_s8(vb67c1x0, va1x0);
240*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2x67c1 = vmull_s8(vb67c1x0, va2x0);
241*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb67c1x1 = vld1_s8(w); w = (const int8_t*) w + 8;
242*4bdc9457SAndroid Build Coastguard Worker vprod0x67c1 = vmlal_s8(vprod0x67c1, vb67c1x1, va0x1);
243*4bdc9457SAndroid Build Coastguard Worker vprod1x67c1 = vmlal_s8(vprod1x67c1, vb67c1x1, va1x1);
244*4bdc9457SAndroid Build Coastguard Worker vprod2x67c1 = vmlal_s8(vprod2x67c1, vb67c1x1, va2x1);
245*4bdc9457SAndroid Build Coastguard Worker vacc0x67 = vpadalq_s16(vacc0x67, vprod0x67c1);
246*4bdc9457SAndroid Build Coastguard Worker vacc1x67 = vpadalq_s16(vacc1x67, vprod1x67c1);
247*4bdc9457SAndroid Build Coastguard Worker vacc2x67 = vpadalq_s16(vacc2x67, vprod2x67c1);
248*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0x89c1 = vmull_s8(vb89c1x0, va0x0);
249*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1x89c1 = vmull_s8(vb89c1x0, va1x0);
250*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2x89c1 = vmull_s8(vb89c1x0, va2x0);
251*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb89c1x1 = vld1_s8(w); w = (const int8_t*) w + 8;
252*4bdc9457SAndroid Build Coastguard Worker vprod0x89c1 = vmlal_s8(vprod0x89c1, vb89c1x1, va0x1);
253*4bdc9457SAndroid Build Coastguard Worker vprod1x89c1 = vmlal_s8(vprod1x89c1, vb89c1x1, va1x1);
254*4bdc9457SAndroid Build Coastguard Worker vprod2x89c1 = vmlal_s8(vprod2x89c1, vb89c1x1, va2x1);
255*4bdc9457SAndroid Build Coastguard Worker vacc0x89 = vpadalq_s16(vacc0x89, vprod0x89c1);
256*4bdc9457SAndroid Build Coastguard Worker vacc1x89 = vpadalq_s16(vacc1x89, vprod1x89c1);
257*4bdc9457SAndroid Build Coastguard Worker vacc2x89 = vpadalq_s16(vacc2x89, vprod2x89c1);
258*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0xABc1 = vmull_s8(vbABc1x0, va0x0);
259*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1xABc1 = vmull_s8(vbABc1x0, va1x0);
260*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2xABc1 = vmull_s8(vbABc1x0, va2x0);
261*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vbABc1x1 = vld1_s8(w); w = (const int8_t*) w + 8;
262*4bdc9457SAndroid Build Coastguard Worker vprod0xABc1 = vmlal_s8(vprod0xABc1, vbABc1x1, va0x1);
263*4bdc9457SAndroid Build Coastguard Worker vprod1xABc1 = vmlal_s8(vprod1xABc1, vbABc1x1, va1x1);
264*4bdc9457SAndroid Build Coastguard Worker vprod2xABc1 = vmlal_s8(vprod2xABc1, vbABc1x1, va2x1);
265*4bdc9457SAndroid Build Coastguard Worker vacc0xAB = vpadalq_s16(vacc0xAB, vprod0xABc1);
266*4bdc9457SAndroid Build Coastguard Worker vacc1xAB = vpadalq_s16(vacc1xAB, vprod1xABc1);
267*4bdc9457SAndroid Build Coastguard Worker vacc2xAB = vpadalq_s16(vacc2xAB, vprod2xABc1);
268*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0xCDc1 = vmull_s8(vbCDc1x0, va0x0);
269*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1xCDc1 = vmull_s8(vbCDc1x0, va1x0);
270*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2xCDc1 = vmull_s8(vbCDc1x0, va2x0);
271*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vbCDc1x1 = vld1_s8(w); w = (const int8_t*) w + 8;
272*4bdc9457SAndroid Build Coastguard Worker vprod0xCDc1 = vmlal_s8(vprod0xCDc1, vbCDc1x1, va0x1);
273*4bdc9457SAndroid Build Coastguard Worker vprod1xCDc1 = vmlal_s8(vprod1xCDc1, vbCDc1x1, va1x1);
274*4bdc9457SAndroid Build Coastguard Worker vprod2xCDc1 = vmlal_s8(vprod2xCDc1, vbCDc1x1, va2x1);
275*4bdc9457SAndroid Build Coastguard Worker vacc0xCD = vpadalq_s16(vacc0xCD, vprod0xCDc1);
276*4bdc9457SAndroid Build Coastguard Worker vacc1xCD = vpadalq_s16(vacc1xCD, vprod1xCDc1);
277*4bdc9457SAndroid Build Coastguard Worker vacc2xCD = vpadalq_s16(vacc2xCD, vprod2xCDc1);
278*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0xEFc1 = vmull_s8(vbEFc1x0, va0x0);
279*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1xEFc1 = vmull_s8(vbEFc1x0, va1x0);
280*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2xEFc1 = vmull_s8(vbEFc1x0, va2x0);
281*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vbEFc1x1 = vld1_s8(w); w = (const int8_t*) w + 8;
282*4bdc9457SAndroid Build Coastguard Worker vprod0xEFc1 = vmlal_s8(vprod0xEFc1, vbEFc1x1, va0x1);
283*4bdc9457SAndroid Build Coastguard Worker vprod1xEFc1 = vmlal_s8(vprod1xEFc1, vbEFc1x1, va1x1);
284*4bdc9457SAndroid Build Coastguard Worker vprod2xEFc1 = vmlal_s8(vprod2xEFc1, vbEFc1x1, va2x1);
285*4bdc9457SAndroid Build Coastguard Worker vacc0xEF = vpadalq_s16(vacc0xEF, vprod0xEFc1);
286*4bdc9457SAndroid Build Coastguard Worker vacc1xEF = vpadalq_s16(vacc1xEF, vprod1xEFc1);
287*4bdc9457SAndroid Build Coastguard Worker vacc2xEF = vpadalq_s16(vacc2xEF, vprod2xEFc1);
288*4bdc9457SAndroid Build Coastguard Worker
289*4bdc9457SAndroid Build Coastguard Worker k -= 16 * sizeof(int8_t);
290*4bdc9457SAndroid Build Coastguard Worker }
291*4bdc9457SAndroid Build Coastguard Worker if (k != 0) {
292*4bdc9457SAndroid Build Coastguard Worker int8x8_t va0x0 = vld1_s8(a0); a0 += 8;
293*4bdc9457SAndroid Build Coastguard Worker int8x8_t va1x0 = vld1_s8(a1); a1 += 8;
294*4bdc9457SAndroid Build Coastguard Worker int8x8_t va2x0 = vld1_s8(a2); a2 += 8;
295*4bdc9457SAndroid Build Coastguard Worker
296*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb01c0x0 = vld1_s8(w); w = (const int8_t*) w + 8;
297*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb23c0x0 = vld1_s8(w); w = (const int8_t*) w + 8;
298*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb45c0x0 = vld1_s8(w); w = (const int8_t*) w + 8;
299*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb67c0x0 = vld1_s8(w); w = (const int8_t*) w + 8;
300*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb89c0x0 = vld1_s8(w); w = (const int8_t*) w + 8;
301*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vbABc0x0 = vld1_s8(w); w = (const int8_t*) w + 8;
302*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vbCDc0x0 = vld1_s8(w); w = (const int8_t*) w + 8;
303*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vbEFc0x0 = vld1_s8(w); w = (const int8_t*) w + 8;
304*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb01c1x0 = vld1_s8(w); w = (const int8_t*) w + 8;
305*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb23c1x0 = vld1_s8(w); w = (const int8_t*) w + 8;
306*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb45c1x0 = vld1_s8(w); w = (const int8_t*) w + 8;
307*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb67c1x0 = vld1_s8(w); w = (const int8_t*) w + 8;
308*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vb89c1x0 = vld1_s8(w); w = (const int8_t*) w + 8;
309*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vbABc1x0 = vld1_s8(w); w = (const int8_t*) w + 8;
310*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vbCDc1x0 = vld1_s8(w); w = (const int8_t*) w + 8;
311*4bdc9457SAndroid Build Coastguard Worker const int8x8_t vbEFc1x0 = vld1_s8(w); w = (const int8_t*) w + 8;
312*4bdc9457SAndroid Build Coastguard Worker
313*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0x01c0 = vmull_s8(vb01c0x0, va0x0);
314*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1x01c0 = vmull_s8(vb01c0x0, va1x0);
315*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2x01c0 = vmull_s8(vb01c0x0, va2x0);
316*4bdc9457SAndroid Build Coastguard Worker vacc0x01 = vpadalq_s16(vacc0x01, vprod0x01c0);
317*4bdc9457SAndroid Build Coastguard Worker vacc1x01 = vpadalq_s16(vacc1x01, vprod1x01c0);
318*4bdc9457SAndroid Build Coastguard Worker vacc2x01 = vpadalq_s16(vacc2x01, vprod2x01c0);
319*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0x23c0 = vmull_s8(vb23c0x0, va0x0);
320*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1x23c0 = vmull_s8(vb23c0x0, va1x0);
321*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2x23c0 = vmull_s8(vb23c0x0, va2x0);
322*4bdc9457SAndroid Build Coastguard Worker vacc0x23 = vpadalq_s16(vacc0x23, vprod0x23c0);
323*4bdc9457SAndroid Build Coastguard Worker vacc1x23 = vpadalq_s16(vacc1x23, vprod1x23c0);
324*4bdc9457SAndroid Build Coastguard Worker vacc2x23 = vpadalq_s16(vacc2x23, vprod2x23c0);
325*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0x45c0 = vmull_s8(vb45c0x0, va0x0);
326*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1x45c0 = vmull_s8(vb45c0x0, va1x0);
327*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2x45c0 = vmull_s8(vb45c0x0, va2x0);
328*4bdc9457SAndroid Build Coastguard Worker vacc0x45 = vpadalq_s16(vacc0x45, vprod0x45c0);
329*4bdc9457SAndroid Build Coastguard Worker vacc1x45 = vpadalq_s16(vacc1x45, vprod1x45c0);
330*4bdc9457SAndroid Build Coastguard Worker vacc2x45 = vpadalq_s16(vacc2x45, vprod2x45c0);
331*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0x67c0 = vmull_s8(vb67c0x0, va0x0);
332*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1x67c0 = vmull_s8(vb67c0x0, va1x0);
333*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2x67c0 = vmull_s8(vb67c0x0, va2x0);
334*4bdc9457SAndroid Build Coastguard Worker vacc0x67 = vpadalq_s16(vacc0x67, vprod0x67c0);
335*4bdc9457SAndroid Build Coastguard Worker vacc1x67 = vpadalq_s16(vacc1x67, vprod1x67c0);
336*4bdc9457SAndroid Build Coastguard Worker vacc2x67 = vpadalq_s16(vacc2x67, vprod2x67c0);
337*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0x89c0 = vmull_s8(vb89c0x0, va0x0);
338*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1x89c0 = vmull_s8(vb89c0x0, va1x0);
339*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2x89c0 = vmull_s8(vb89c0x0, va2x0);
340*4bdc9457SAndroid Build Coastguard Worker vacc0x89 = vpadalq_s16(vacc0x89, vprod0x89c0);
341*4bdc9457SAndroid Build Coastguard Worker vacc1x89 = vpadalq_s16(vacc1x89, vprod1x89c0);
342*4bdc9457SAndroid Build Coastguard Worker vacc2x89 = vpadalq_s16(vacc2x89, vprod2x89c0);
343*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0xABc0 = vmull_s8(vbABc0x0, va0x0);
344*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1xABc0 = vmull_s8(vbABc0x0, va1x0);
345*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2xABc0 = vmull_s8(vbABc0x0, va2x0);
346*4bdc9457SAndroid Build Coastguard Worker vacc0xAB = vpadalq_s16(vacc0xAB, vprod0xABc0);
347*4bdc9457SAndroid Build Coastguard Worker vacc1xAB = vpadalq_s16(vacc1xAB, vprod1xABc0);
348*4bdc9457SAndroid Build Coastguard Worker vacc2xAB = vpadalq_s16(vacc2xAB, vprod2xABc0);
349*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0xCDc0 = vmull_s8(vbCDc0x0, va0x0);
350*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1xCDc0 = vmull_s8(vbCDc0x0, va1x0);
351*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2xCDc0 = vmull_s8(vbCDc0x0, va2x0);
352*4bdc9457SAndroid Build Coastguard Worker vacc0xCD = vpadalq_s16(vacc0xCD, vprod0xCDc0);
353*4bdc9457SAndroid Build Coastguard Worker vacc1xCD = vpadalq_s16(vacc1xCD, vprod1xCDc0);
354*4bdc9457SAndroid Build Coastguard Worker vacc2xCD = vpadalq_s16(vacc2xCD, vprod2xCDc0);
355*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0xEFc0 = vmull_s8(vbEFc0x0, va0x0);
356*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1xEFc0 = vmull_s8(vbEFc0x0, va1x0);
357*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2xEFc0 = vmull_s8(vbEFc0x0, va2x0);
358*4bdc9457SAndroid Build Coastguard Worker vacc0xEF = vpadalq_s16(vacc0xEF, vprod0xEFc0);
359*4bdc9457SAndroid Build Coastguard Worker vacc1xEF = vpadalq_s16(vacc1xEF, vprod1xEFc0);
360*4bdc9457SAndroid Build Coastguard Worker vacc2xEF = vpadalq_s16(vacc2xEF, vprod2xEFc0);
361*4bdc9457SAndroid Build Coastguard Worker va0x0 = vext_s8(va0x0, va0x0, 4);
362*4bdc9457SAndroid Build Coastguard Worker va1x0 = vext_s8(va1x0, va1x0, 4);
363*4bdc9457SAndroid Build Coastguard Worker va2x0 = vext_s8(va2x0, va2x0, 4);
364*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0x01c1 = vmull_s8(vb01c1x0, va0x0);
365*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1x01c1 = vmull_s8(vb01c1x0, va1x0);
366*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2x01c1 = vmull_s8(vb01c1x0, va2x0);
367*4bdc9457SAndroid Build Coastguard Worker vacc0x01 = vpadalq_s16(vacc0x01, vprod0x01c1);
368*4bdc9457SAndroid Build Coastguard Worker vacc1x01 = vpadalq_s16(vacc1x01, vprod1x01c1);
369*4bdc9457SAndroid Build Coastguard Worker vacc2x01 = vpadalq_s16(vacc2x01, vprod2x01c1);
370*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0x23c1 = vmull_s8(vb23c1x0, va0x0);
371*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1x23c1 = vmull_s8(vb23c1x0, va1x0);
372*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2x23c1 = vmull_s8(vb23c1x0, va2x0);
373*4bdc9457SAndroid Build Coastguard Worker vacc0x23 = vpadalq_s16(vacc0x23, vprod0x23c1);
374*4bdc9457SAndroid Build Coastguard Worker vacc1x23 = vpadalq_s16(vacc1x23, vprod1x23c1);
375*4bdc9457SAndroid Build Coastguard Worker vacc2x23 = vpadalq_s16(vacc2x23, vprod2x23c1);
376*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0x45c1 = vmull_s8(vb45c1x0, va0x0);
377*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1x45c1 = vmull_s8(vb45c1x0, va1x0);
378*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2x45c1 = vmull_s8(vb45c1x0, va2x0);
379*4bdc9457SAndroid Build Coastguard Worker vacc0x45 = vpadalq_s16(vacc0x45, vprod0x45c1);
380*4bdc9457SAndroid Build Coastguard Worker vacc1x45 = vpadalq_s16(vacc1x45, vprod1x45c1);
381*4bdc9457SAndroid Build Coastguard Worker vacc2x45 = vpadalq_s16(vacc2x45, vprod2x45c1);
382*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0x67c1 = vmull_s8(vb67c1x0, va0x0);
383*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1x67c1 = vmull_s8(vb67c1x0, va1x0);
384*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2x67c1 = vmull_s8(vb67c1x0, va2x0);
385*4bdc9457SAndroid Build Coastguard Worker vacc0x67 = vpadalq_s16(vacc0x67, vprod0x67c1);
386*4bdc9457SAndroid Build Coastguard Worker vacc1x67 = vpadalq_s16(vacc1x67, vprod1x67c1);
387*4bdc9457SAndroid Build Coastguard Worker vacc2x67 = vpadalq_s16(vacc2x67, vprod2x67c1);
388*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0x89c1 = vmull_s8(vb89c1x0, va0x0);
389*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1x89c1 = vmull_s8(vb89c1x0, va1x0);
390*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2x89c1 = vmull_s8(vb89c1x0, va2x0);
391*4bdc9457SAndroid Build Coastguard Worker vacc0x89 = vpadalq_s16(vacc0x89, vprod0x89c1);
392*4bdc9457SAndroid Build Coastguard Worker vacc1x89 = vpadalq_s16(vacc1x89, vprod1x89c1);
393*4bdc9457SAndroid Build Coastguard Worker vacc2x89 = vpadalq_s16(vacc2x89, vprod2x89c1);
394*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0xABc1 = vmull_s8(vbABc1x0, va0x0);
395*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1xABc1 = vmull_s8(vbABc1x0, va1x0);
396*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2xABc1 = vmull_s8(vbABc1x0, va2x0);
397*4bdc9457SAndroid Build Coastguard Worker vacc0xAB = vpadalq_s16(vacc0xAB, vprod0xABc1);
398*4bdc9457SAndroid Build Coastguard Worker vacc1xAB = vpadalq_s16(vacc1xAB, vprod1xABc1);
399*4bdc9457SAndroid Build Coastguard Worker vacc2xAB = vpadalq_s16(vacc2xAB, vprod2xABc1);
400*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0xCDc1 = vmull_s8(vbCDc1x0, va0x0);
401*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1xCDc1 = vmull_s8(vbCDc1x0, va1x0);
402*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2xCDc1 = vmull_s8(vbCDc1x0, va2x0);
403*4bdc9457SAndroid Build Coastguard Worker vacc0xCD = vpadalq_s16(vacc0xCD, vprod0xCDc1);
404*4bdc9457SAndroid Build Coastguard Worker vacc1xCD = vpadalq_s16(vacc1xCD, vprod1xCDc1);
405*4bdc9457SAndroid Build Coastguard Worker vacc2xCD = vpadalq_s16(vacc2xCD, vprod2xCDc1);
406*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod0xEFc1 = vmull_s8(vbEFc1x0, va0x0);
407*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod1xEFc1 = vmull_s8(vbEFc1x0, va1x0);
408*4bdc9457SAndroid Build Coastguard Worker int16x8_t vprod2xEFc1 = vmull_s8(vbEFc1x0, va2x0);
409*4bdc9457SAndroid Build Coastguard Worker vacc0xEF = vpadalq_s16(vacc0xEF, vprod0xEFc1);
410*4bdc9457SAndroid Build Coastguard Worker vacc1xEF = vpadalq_s16(vacc1xEF, vprod1xEFc1);
411*4bdc9457SAndroid Build Coastguard Worker vacc2xEF = vpadalq_s16(vacc2xEF, vprod2xEFc1);
412*4bdc9457SAndroid Build Coastguard Worker
413*4bdc9457SAndroid Build Coastguard Worker }
414*4bdc9457SAndroid Build Coastguard Worker
415*4bdc9457SAndroid Build Coastguard Worker p -= 3 * sizeof(void*);
416*4bdc9457SAndroid Build Coastguard Worker } while (p != 0);
417*4bdc9457SAndroid Build Coastguard Worker
418*4bdc9457SAndroid Build Coastguard Worker #if XNN_ARCH_ARM64
419*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc0x0123 = vpaddq_s32(vacc0x01, vacc0x23);
420*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc0x4567 = vpaddq_s32(vacc0x45, vacc0x67);
421*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc0x89AB = vpaddq_s32(vacc0x89, vacc0xAB);
422*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc0xCDEF = vpaddq_s32(vacc0xCD, vacc0xEF);
423*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc1x0123 = vpaddq_s32(vacc1x01, vacc1x23);
424*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc1x4567 = vpaddq_s32(vacc1x45, vacc1x67);
425*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc1x89AB = vpaddq_s32(vacc1x89, vacc1xAB);
426*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc1xCDEF = vpaddq_s32(vacc1xCD, vacc1xEF);
427*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc2x0123 = vpaddq_s32(vacc2x01, vacc2x23);
428*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc2x4567 = vpaddq_s32(vacc2x45, vacc2x67);
429*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc2x89AB = vpaddq_s32(vacc2x89, vacc2xAB);
430*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc2xCDEF = vpaddq_s32(vacc2xCD, vacc2xEF);
431*4bdc9457SAndroid Build Coastguard Worker #else
432*4bdc9457SAndroid Build Coastguard Worker const int32x2_t vsum0x01 = vpadd_s32(vget_low_s32(vacc0x01), vget_high_s32(vacc0x01));
433*4bdc9457SAndroid Build Coastguard Worker const int32x2_t vsum0x23 = vpadd_s32(vget_low_s32(vacc0x23), vget_high_s32(vacc0x23));
434*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc0x0123 = vcombine_s32(vsum0x01, vsum0x23);
435*4bdc9457SAndroid Build Coastguard Worker const int32x2_t vsum0x45 = vpadd_s32(vget_low_s32(vacc0x45), vget_high_s32(vacc0x45));
436*4bdc9457SAndroid Build Coastguard Worker const int32x2_t vsum0x67 = vpadd_s32(vget_low_s32(vacc0x67), vget_high_s32(vacc0x67));
437*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc0x4567 = vcombine_s32(vsum0x45, vsum0x67);
438*4bdc9457SAndroid Build Coastguard Worker const int32x2_t vsum0x89 = vpadd_s32(vget_low_s32(vacc0x89), vget_high_s32(vacc0x89));
439*4bdc9457SAndroid Build Coastguard Worker const int32x2_t vsum0xAB = vpadd_s32(vget_low_s32(vacc0xAB), vget_high_s32(vacc0xAB));
440*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc0x89AB = vcombine_s32(vsum0x89, vsum0xAB);
441*4bdc9457SAndroid Build Coastguard Worker const int32x2_t vsum0xCD = vpadd_s32(vget_low_s32(vacc0xCD), vget_high_s32(vacc0xCD));
442*4bdc9457SAndroid Build Coastguard Worker const int32x2_t vsum0xEF = vpadd_s32(vget_low_s32(vacc0xEF), vget_high_s32(vacc0xEF));
443*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc0xCDEF = vcombine_s32(vsum0xCD, vsum0xEF);
444*4bdc9457SAndroid Build Coastguard Worker const int32x2_t vsum1x01 = vpadd_s32(vget_low_s32(vacc1x01), vget_high_s32(vacc1x01));
445*4bdc9457SAndroid Build Coastguard Worker const int32x2_t vsum1x23 = vpadd_s32(vget_low_s32(vacc1x23), vget_high_s32(vacc1x23));
446*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc1x0123 = vcombine_s32(vsum1x01, vsum1x23);
447*4bdc9457SAndroid Build Coastguard Worker const int32x2_t vsum1x45 = vpadd_s32(vget_low_s32(vacc1x45), vget_high_s32(vacc1x45));
448*4bdc9457SAndroid Build Coastguard Worker const int32x2_t vsum1x67 = vpadd_s32(vget_low_s32(vacc1x67), vget_high_s32(vacc1x67));
449*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc1x4567 = vcombine_s32(vsum1x45, vsum1x67);
450*4bdc9457SAndroid Build Coastguard Worker const int32x2_t vsum1x89 = vpadd_s32(vget_low_s32(vacc1x89), vget_high_s32(vacc1x89));
451*4bdc9457SAndroid Build Coastguard Worker const int32x2_t vsum1xAB = vpadd_s32(vget_low_s32(vacc1xAB), vget_high_s32(vacc1xAB));
452*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc1x89AB = vcombine_s32(vsum1x89, vsum1xAB);
453*4bdc9457SAndroid Build Coastguard Worker const int32x2_t vsum1xCD = vpadd_s32(vget_low_s32(vacc1xCD), vget_high_s32(vacc1xCD));
454*4bdc9457SAndroid Build Coastguard Worker const int32x2_t vsum1xEF = vpadd_s32(vget_low_s32(vacc1xEF), vget_high_s32(vacc1xEF));
455*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc1xCDEF = vcombine_s32(vsum1xCD, vsum1xEF);
456*4bdc9457SAndroid Build Coastguard Worker const int32x2_t vsum2x01 = vpadd_s32(vget_low_s32(vacc2x01), vget_high_s32(vacc2x01));
457*4bdc9457SAndroid Build Coastguard Worker const int32x2_t vsum2x23 = vpadd_s32(vget_low_s32(vacc2x23), vget_high_s32(vacc2x23));
458*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc2x0123 = vcombine_s32(vsum2x01, vsum2x23);
459*4bdc9457SAndroid Build Coastguard Worker const int32x2_t vsum2x45 = vpadd_s32(vget_low_s32(vacc2x45), vget_high_s32(vacc2x45));
460*4bdc9457SAndroid Build Coastguard Worker const int32x2_t vsum2x67 = vpadd_s32(vget_low_s32(vacc2x67), vget_high_s32(vacc2x67));
461*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc2x4567 = vcombine_s32(vsum2x45, vsum2x67);
462*4bdc9457SAndroid Build Coastguard Worker const int32x2_t vsum2x89 = vpadd_s32(vget_low_s32(vacc2x89), vget_high_s32(vacc2x89));
463*4bdc9457SAndroid Build Coastguard Worker const int32x2_t vsum2xAB = vpadd_s32(vget_low_s32(vacc2xAB), vget_high_s32(vacc2xAB));
464*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc2x89AB = vcombine_s32(vsum2x89, vsum2xAB);
465*4bdc9457SAndroid Build Coastguard Worker const int32x2_t vsum2xCD = vpadd_s32(vget_low_s32(vacc2xCD), vget_high_s32(vacc2xCD));
466*4bdc9457SAndroid Build Coastguard Worker const int32x2_t vsum2xEF = vpadd_s32(vget_low_s32(vacc2xEF), vget_high_s32(vacc2xEF));
467*4bdc9457SAndroid Build Coastguard Worker int32x4_t vacc2xCDEF = vcombine_s32(vsum2xCD, vsum2xEF);
468*4bdc9457SAndroid Build Coastguard Worker #endif
469*4bdc9457SAndroid Build Coastguard Worker
470*4bdc9457SAndroid Build Coastguard Worker const int32x4_t vright_pre_shift = vld1q_dup_s32(¶ms->rndnu_neon.right_pre_shift);
471*4bdc9457SAndroid Build Coastguard Worker const int32x4_t vmultiplier = vld1q_dup_s32(¶ms->rndnu_neon.multiplier);
472*4bdc9457SAndroid Build Coastguard Worker const int32x4_t vright_post_shift = vld1q_dup_s32(¶ms->rndnu_neon.right_post_shift);
473*4bdc9457SAndroid Build Coastguard Worker
474*4bdc9457SAndroid Build Coastguard Worker vacc0x0123 = vqshlq_s32(vacc0x0123, vright_pre_shift);
475*4bdc9457SAndroid Build Coastguard Worker vacc0x4567 = vqshlq_s32(vacc0x4567, vright_pre_shift);
476*4bdc9457SAndroid Build Coastguard Worker vacc0x89AB = vqshlq_s32(vacc0x89AB, vright_pre_shift);
477*4bdc9457SAndroid Build Coastguard Worker vacc0xCDEF = vqshlq_s32(vacc0xCDEF, vright_pre_shift);
478*4bdc9457SAndroid Build Coastguard Worker vacc1x0123 = vqshlq_s32(vacc1x0123, vright_pre_shift);
479*4bdc9457SAndroid Build Coastguard Worker vacc1x4567 = vqshlq_s32(vacc1x4567, vright_pre_shift);
480*4bdc9457SAndroid Build Coastguard Worker vacc1x89AB = vqshlq_s32(vacc1x89AB, vright_pre_shift);
481*4bdc9457SAndroid Build Coastguard Worker vacc1xCDEF = vqshlq_s32(vacc1xCDEF, vright_pre_shift);
482*4bdc9457SAndroid Build Coastguard Worker vacc2x0123 = vqshlq_s32(vacc2x0123, vright_pre_shift);
483*4bdc9457SAndroid Build Coastguard Worker vacc2x4567 = vqshlq_s32(vacc2x4567, vright_pre_shift);
484*4bdc9457SAndroid Build Coastguard Worker vacc2x89AB = vqshlq_s32(vacc2x89AB, vright_pre_shift);
485*4bdc9457SAndroid Build Coastguard Worker vacc2xCDEF = vqshlq_s32(vacc2xCDEF, vright_pre_shift);
486*4bdc9457SAndroid Build Coastguard Worker
487*4bdc9457SAndroid Build Coastguard Worker vacc0x0123 = vqdmulhq_s32(vacc0x0123, vmultiplier);
488*4bdc9457SAndroid Build Coastguard Worker vacc0x4567 = vqdmulhq_s32(vacc0x4567, vmultiplier);
489*4bdc9457SAndroid Build Coastguard Worker vacc0x89AB = vqdmulhq_s32(vacc0x89AB, vmultiplier);
490*4bdc9457SAndroid Build Coastguard Worker vacc0xCDEF = vqdmulhq_s32(vacc0xCDEF, vmultiplier);
491*4bdc9457SAndroid Build Coastguard Worker vacc1x0123 = vqdmulhq_s32(vacc1x0123, vmultiplier);
492*4bdc9457SAndroid Build Coastguard Worker vacc1x4567 = vqdmulhq_s32(vacc1x4567, vmultiplier);
493*4bdc9457SAndroid Build Coastguard Worker vacc1x89AB = vqdmulhq_s32(vacc1x89AB, vmultiplier);
494*4bdc9457SAndroid Build Coastguard Worker vacc1xCDEF = vqdmulhq_s32(vacc1xCDEF, vmultiplier);
495*4bdc9457SAndroid Build Coastguard Worker vacc2x0123 = vqdmulhq_s32(vacc2x0123, vmultiplier);
496*4bdc9457SAndroid Build Coastguard Worker vacc2x4567 = vqdmulhq_s32(vacc2x4567, vmultiplier);
497*4bdc9457SAndroid Build Coastguard Worker vacc2x89AB = vqdmulhq_s32(vacc2x89AB, vmultiplier);
498*4bdc9457SAndroid Build Coastguard Worker vacc2xCDEF = vqdmulhq_s32(vacc2xCDEF, vmultiplier);
499*4bdc9457SAndroid Build Coastguard Worker
500*4bdc9457SAndroid Build Coastguard Worker vacc0x0123 = vrshlq_s32(vacc0x0123, vright_post_shift);
501*4bdc9457SAndroid Build Coastguard Worker vacc0x4567 = vrshlq_s32(vacc0x4567, vright_post_shift);
502*4bdc9457SAndroid Build Coastguard Worker vacc0x89AB = vrshlq_s32(vacc0x89AB, vright_post_shift);
503*4bdc9457SAndroid Build Coastguard Worker vacc0xCDEF = vrshlq_s32(vacc0xCDEF, vright_post_shift);
504*4bdc9457SAndroid Build Coastguard Worker vacc1x0123 = vrshlq_s32(vacc1x0123, vright_post_shift);
505*4bdc9457SAndroid Build Coastguard Worker vacc1x4567 = vrshlq_s32(vacc1x4567, vright_post_shift);
506*4bdc9457SAndroid Build Coastguard Worker vacc1x89AB = vrshlq_s32(vacc1x89AB, vright_post_shift);
507*4bdc9457SAndroid Build Coastguard Worker vacc1xCDEF = vrshlq_s32(vacc1xCDEF, vright_post_shift);
508*4bdc9457SAndroid Build Coastguard Worker vacc2x0123 = vrshlq_s32(vacc2x0123, vright_post_shift);
509*4bdc9457SAndroid Build Coastguard Worker vacc2x4567 = vrshlq_s32(vacc2x4567, vright_post_shift);
510*4bdc9457SAndroid Build Coastguard Worker vacc2x89AB = vrshlq_s32(vacc2x89AB, vright_post_shift);
511*4bdc9457SAndroid Build Coastguard Worker vacc2xCDEF = vrshlq_s32(vacc2xCDEF, vright_post_shift);
512*4bdc9457SAndroid Build Coastguard Worker
513*4bdc9457SAndroid Build Coastguard Worker const int16x8_t voutput_zero_point = vld1q_dup_s16(¶ms->rndnu_neon.output_zero_point);
514*4bdc9457SAndroid Build Coastguard Worker #if XNN_ARCH_ARM64
515*4bdc9457SAndroid Build Coastguard Worker int16x8_t vacc0x01234567 = vqmovn_high_s32(vqmovn_s32(vacc0x0123), vacc0x4567);
516*4bdc9457SAndroid Build Coastguard Worker int16x8_t vacc0x89ABCDEF = vqmovn_high_s32(vqmovn_s32(vacc0x89AB), vacc0xCDEF);
517*4bdc9457SAndroid Build Coastguard Worker int16x8_t vacc1x01234567 = vqmovn_high_s32(vqmovn_s32(vacc1x0123), vacc1x4567);
518*4bdc9457SAndroid Build Coastguard Worker int16x8_t vacc1x89ABCDEF = vqmovn_high_s32(vqmovn_s32(vacc1x89AB), vacc1xCDEF);
519*4bdc9457SAndroid Build Coastguard Worker int16x8_t vacc2x01234567 = vqmovn_high_s32(vqmovn_s32(vacc2x0123), vacc2x4567);
520*4bdc9457SAndroid Build Coastguard Worker int16x8_t vacc2x89ABCDEF = vqmovn_high_s32(vqmovn_s32(vacc2x89AB), vacc2xCDEF);
521*4bdc9457SAndroid Build Coastguard Worker
522*4bdc9457SAndroid Build Coastguard Worker vacc0x01234567 = vqaddq_s16(vacc0x01234567, voutput_zero_point);
523*4bdc9457SAndroid Build Coastguard Worker vacc0x89ABCDEF = vqaddq_s16(vacc0x89ABCDEF, voutput_zero_point);
524*4bdc9457SAndroid Build Coastguard Worker vacc1x01234567 = vqaddq_s16(vacc1x01234567, voutput_zero_point);
525*4bdc9457SAndroid Build Coastguard Worker vacc1x89ABCDEF = vqaddq_s16(vacc1x89ABCDEF, voutput_zero_point);
526*4bdc9457SAndroid Build Coastguard Worker vacc2x01234567 = vqaddq_s16(vacc2x01234567, voutput_zero_point);
527*4bdc9457SAndroid Build Coastguard Worker vacc2x89ABCDEF = vqaddq_s16(vacc2x89ABCDEF, voutput_zero_point);
528*4bdc9457SAndroid Build Coastguard Worker
529*4bdc9457SAndroid Build Coastguard Worker int8x16_t vout0x0123456789ABCDEF = vqmovn_high_s16(vqmovn_s16(vacc0x01234567), vacc0x89ABCDEF);
530*4bdc9457SAndroid Build Coastguard Worker int8x16_t vout1x0123456789ABCDEF = vqmovn_high_s16(vqmovn_s16(vacc1x01234567), vacc1x89ABCDEF);
531*4bdc9457SAndroid Build Coastguard Worker int8x16_t vout2x0123456789ABCDEF = vqmovn_high_s16(vqmovn_s16(vacc2x01234567), vacc2x89ABCDEF);
532*4bdc9457SAndroid Build Coastguard Worker #else
533*4bdc9457SAndroid Build Coastguard Worker int16x8_t vacc0x01234567 = vcombine_s16(vqmovn_s32(vacc0x0123), vqmovn_s32(vacc0x4567));
534*4bdc9457SAndroid Build Coastguard Worker int16x8_t vacc0x89ABCDEF = vcombine_s16(vqmovn_s32(vacc0x89AB), vqmovn_s32(vacc0xCDEF));
535*4bdc9457SAndroid Build Coastguard Worker int16x8_t vacc1x01234567 = vcombine_s16(vqmovn_s32(vacc1x0123), vqmovn_s32(vacc1x4567));
536*4bdc9457SAndroid Build Coastguard Worker int16x8_t vacc1x89ABCDEF = vcombine_s16(vqmovn_s32(vacc1x89AB), vqmovn_s32(vacc1xCDEF));
537*4bdc9457SAndroid Build Coastguard Worker int16x8_t vacc2x01234567 = vcombine_s16(vqmovn_s32(vacc2x0123), vqmovn_s32(vacc2x4567));
538*4bdc9457SAndroid Build Coastguard Worker int16x8_t vacc2x89ABCDEF = vcombine_s16(vqmovn_s32(vacc2x89AB), vqmovn_s32(vacc2xCDEF));
539*4bdc9457SAndroid Build Coastguard Worker
540*4bdc9457SAndroid Build Coastguard Worker vacc0x01234567 = vqaddq_s16(vacc0x01234567, voutput_zero_point);
541*4bdc9457SAndroid Build Coastguard Worker vacc0x89ABCDEF = vqaddq_s16(vacc0x89ABCDEF, voutput_zero_point);
542*4bdc9457SAndroid Build Coastguard Worker vacc1x01234567 = vqaddq_s16(vacc1x01234567, voutput_zero_point);
543*4bdc9457SAndroid Build Coastguard Worker vacc1x89ABCDEF = vqaddq_s16(vacc1x89ABCDEF, voutput_zero_point);
544*4bdc9457SAndroid Build Coastguard Worker vacc2x01234567 = vqaddq_s16(vacc2x01234567, voutput_zero_point);
545*4bdc9457SAndroid Build Coastguard Worker vacc2x89ABCDEF = vqaddq_s16(vacc2x89ABCDEF, voutput_zero_point);
546*4bdc9457SAndroid Build Coastguard Worker
547*4bdc9457SAndroid Build Coastguard Worker int8x16_t vout0x0123456789ABCDEF = vcombine_s8(vqmovn_s16(vacc0x01234567), vqmovn_s16(vacc0x89ABCDEF));
548*4bdc9457SAndroid Build Coastguard Worker int8x16_t vout1x0123456789ABCDEF = vcombine_s8(vqmovn_s16(vacc1x01234567), vqmovn_s16(vacc1x89ABCDEF));
549*4bdc9457SAndroid Build Coastguard Worker int8x16_t vout2x0123456789ABCDEF = vcombine_s8(vqmovn_s16(vacc2x01234567), vqmovn_s16(vacc2x89ABCDEF));
550*4bdc9457SAndroid Build Coastguard Worker #endif
551*4bdc9457SAndroid Build Coastguard Worker
552*4bdc9457SAndroid Build Coastguard Worker const int8x16_t voutput_min = vld1q_dup_s8(¶ms->rndnu_neon.output_min);
553*4bdc9457SAndroid Build Coastguard Worker vout0x0123456789ABCDEF = vmaxq_s8(vout0x0123456789ABCDEF, voutput_min);
554*4bdc9457SAndroid Build Coastguard Worker vout1x0123456789ABCDEF = vmaxq_s8(vout1x0123456789ABCDEF, voutput_min);
555*4bdc9457SAndroid Build Coastguard Worker vout2x0123456789ABCDEF = vmaxq_s8(vout2x0123456789ABCDEF, voutput_min);
556*4bdc9457SAndroid Build Coastguard Worker
557*4bdc9457SAndroid Build Coastguard Worker const int8x16_t voutput_max = vld1q_dup_s8(¶ms->rndnu_neon.output_max);
558*4bdc9457SAndroid Build Coastguard Worker vout0x0123456789ABCDEF = vminq_s8(vout0x0123456789ABCDEF, voutput_max);
559*4bdc9457SAndroid Build Coastguard Worker vout1x0123456789ABCDEF = vminq_s8(vout1x0123456789ABCDEF, voutput_max);
560*4bdc9457SAndroid Build Coastguard Worker vout2x0123456789ABCDEF = vminq_s8(vout2x0123456789ABCDEF, voutput_max);
561*4bdc9457SAndroid Build Coastguard Worker
562*4bdc9457SAndroid Build Coastguard Worker if (nc >= 16) {
563*4bdc9457SAndroid Build Coastguard Worker vst1q_s8(c2 + 0, vout2x0123456789ABCDEF);
564*4bdc9457SAndroid Build Coastguard Worker vst1q_s8(c1 + 0, vout1x0123456789ABCDEF);
565*4bdc9457SAndroid Build Coastguard Worker vst1q_s8(c0 + 0, vout0x0123456789ABCDEF);
566*4bdc9457SAndroid Build Coastguard Worker
567*4bdc9457SAndroid Build Coastguard Worker c2 = (int8_t*) ((uintptr_t) c2 + cn_stride);
568*4bdc9457SAndroid Build Coastguard Worker c1 = (int8_t*) ((uintptr_t) c1 + cn_stride);
569*4bdc9457SAndroid Build Coastguard Worker c0 = (int8_t*) ((uintptr_t) c0 + cn_stride);
570*4bdc9457SAndroid Build Coastguard Worker
571*4bdc9457SAndroid Build Coastguard Worker a = (const int8_t**restrict) ((uintptr_t) a - ks);
572*4bdc9457SAndroid Build Coastguard Worker
573*4bdc9457SAndroid Build Coastguard Worker nc -= 16;
574*4bdc9457SAndroid Build Coastguard Worker } else {
575*4bdc9457SAndroid Build Coastguard Worker int8x8_t vout2x01234567 = vget_low_s8(vout2x0123456789ABCDEF);
576*4bdc9457SAndroid Build Coastguard Worker int8x16_t vout0x01234567_1x01234567 = vcombine_s8(vget_low_s8(vout0x0123456789ABCDEF), vget_low_s8(vout1x0123456789ABCDEF));
577*4bdc9457SAndroid Build Coastguard Worker if (nc & 8) {
578*4bdc9457SAndroid Build Coastguard Worker vst1_s8(c2, vout2x01234567); c2 += 8;
579*4bdc9457SAndroid Build Coastguard Worker vst1_s8(c1, vget_high_s8(vout0x01234567_1x01234567)); c1 += 8;
580*4bdc9457SAndroid Build Coastguard Worker vst1_s8(c0, vget_low_s8(vout0x01234567_1x01234567)); c0 += 8;
581*4bdc9457SAndroid Build Coastguard Worker vout2x01234567 = vget_high_s8(vout2x0123456789ABCDEF);
582*4bdc9457SAndroid Build Coastguard Worker vout0x01234567_1x01234567 = vcombine_s8(vget_high_s8(vout0x0123456789ABCDEF), vget_high_s8(vout1x0123456789ABCDEF));
583*4bdc9457SAndroid Build Coastguard Worker }
584*4bdc9457SAndroid Build Coastguard Worker if (nc & 4) {
585*4bdc9457SAndroid Build Coastguard Worker vst1_lane_u32((void*) c2, vreinterpret_u32_s8(vout2x01234567), 0); c2 += 4;
586*4bdc9457SAndroid Build Coastguard Worker vst1q_lane_u32((void*) c1, vreinterpretq_u32_s8(vout0x01234567_1x01234567), 2); c1 += 4;
587*4bdc9457SAndroid Build Coastguard Worker vst1q_lane_u32((void*) c0, vreinterpretq_u32_s8(vout0x01234567_1x01234567), 0); c0 += 4;
588*4bdc9457SAndroid Build Coastguard Worker vout2x01234567 = vext_s8(vout2x01234567, vout2x01234567, 4);
589*4bdc9457SAndroid Build Coastguard Worker vout0x01234567_1x01234567 = vextq_s8(vout0x01234567_1x01234567, vout0x01234567_1x01234567, 4);
590*4bdc9457SAndroid Build Coastguard Worker }
591*4bdc9457SAndroid Build Coastguard Worker if (nc & 2) {
592*4bdc9457SAndroid Build Coastguard Worker vst1_lane_u16((void*) c2, vreinterpret_u16_s8(vout2x01234567), 0); c2 += 2;
593*4bdc9457SAndroid Build Coastguard Worker vst1q_lane_u16((void*) c1, vreinterpretq_u16_s8(vout0x01234567_1x01234567), 4); c1 += 2;
594*4bdc9457SAndroid Build Coastguard Worker vst1q_lane_u16((void*) c0, vreinterpretq_u16_s8(vout0x01234567_1x01234567), 0); c0 += 2;
595*4bdc9457SAndroid Build Coastguard Worker vout2x01234567 = vext_s8(vout2x01234567, vout2x01234567, 2);
596*4bdc9457SAndroid Build Coastguard Worker vout0x01234567_1x01234567 = vextq_s8(vout0x01234567_1x01234567, vout0x01234567_1x01234567, 2);
597*4bdc9457SAndroid Build Coastguard Worker }
598*4bdc9457SAndroid Build Coastguard Worker if (nc & 1) {
599*4bdc9457SAndroid Build Coastguard Worker vst1_lane_s8(c2, vout2x01234567, 0);
600*4bdc9457SAndroid Build Coastguard Worker vst1q_lane_s8(c1, vout0x01234567_1x01234567, 8);
601*4bdc9457SAndroid Build Coastguard Worker vst1q_lane_s8(c0, vout0x01234567_1x01234567, 0);
602*4bdc9457SAndroid Build Coastguard Worker }
603*4bdc9457SAndroid Build Coastguard Worker
604*4bdc9457SAndroid Build Coastguard Worker nc = 0;
605*4bdc9457SAndroid Build Coastguard Worker }
606*4bdc9457SAndroid Build Coastguard Worker } while (nc != 0);
607*4bdc9457SAndroid Build Coastguard Worker }
608