1*4bdc9457SAndroid Build Coastguard Worker // Auto-generated file. Do not edit!
2*4bdc9457SAndroid Build Coastguard Worker // Template: src/qs8-gavgpool/multipass-wasmsimd.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 <wasm_simd128.h>
13*4bdc9457SAndroid Build Coastguard Worker
14*4bdc9457SAndroid Build Coastguard Worker #include <xnnpack/gavgpool.h>
15*4bdc9457SAndroid Build Coastguard Worker #include <xnnpack/math.h>
16*4bdc9457SAndroid Build Coastguard Worker
17*4bdc9457SAndroid Build Coastguard Worker
xnn_qs8_gavgpool_minmax_fp32_ukernel_7p7x__wasmsimd_c24(size_t rows,size_t channels,const int8_t * input,size_t input_stride,const int8_t * zero,int32_t * buffer,int8_t * output,const union xnn_qs8_avgpool_minmax_params params[restrict XNN_MIN_ELEMENTS (1)])18*4bdc9457SAndroid Build Coastguard Worker void xnn_qs8_gavgpool_minmax_fp32_ukernel_7p7x__wasmsimd_c24(
19*4bdc9457SAndroid Build Coastguard Worker size_t rows,
20*4bdc9457SAndroid Build Coastguard Worker size_t channels,
21*4bdc9457SAndroid Build Coastguard Worker const int8_t* input,
22*4bdc9457SAndroid Build Coastguard Worker size_t input_stride,
23*4bdc9457SAndroid Build Coastguard Worker const int8_t* zero,
24*4bdc9457SAndroid Build Coastguard Worker int32_t* buffer,
25*4bdc9457SAndroid Build Coastguard Worker int8_t* output,
26*4bdc9457SAndroid Build Coastguard Worker const union xnn_qs8_avgpool_minmax_params params[restrict XNN_MIN_ELEMENTS(1)]) XNN_OOB_READS
27*4bdc9457SAndroid Build Coastguard Worker {
28*4bdc9457SAndroid Build Coastguard Worker assert(rows > 7);
29*4bdc9457SAndroid Build Coastguard Worker assert(channels != 0);
30*4bdc9457SAndroid Build Coastguard Worker
31*4bdc9457SAndroid Build Coastguard Worker const int8_t* i0 = input;
32*4bdc9457SAndroid Build Coastguard Worker const int8_t* i1 = (const int8_t*) ((uintptr_t) i0 + input_stride);
33*4bdc9457SAndroid Build Coastguard Worker const int8_t* i2 = (const int8_t*) ((uintptr_t) i1 + input_stride);
34*4bdc9457SAndroid Build Coastguard Worker const int8_t* i3 = (const int8_t*) ((uintptr_t) i2 + input_stride);
35*4bdc9457SAndroid Build Coastguard Worker const int8_t* i4 = (const int8_t*) ((uintptr_t) i3 + input_stride);
36*4bdc9457SAndroid Build Coastguard Worker const int8_t* i5 = (const int8_t*) ((uintptr_t) i4 + input_stride);
37*4bdc9457SAndroid Build Coastguard Worker const int8_t* i6 = (const int8_t*) ((uintptr_t) i5 + input_stride);
38*4bdc9457SAndroid Build Coastguard Worker const size_t input_increment = 7 * input_stride - round_up_po2(channels, 8) * sizeof(int8_t);
39*4bdc9457SAndroid Build Coastguard Worker
40*4bdc9457SAndroid Build Coastguard Worker const v128_t vinit_bias = wasm_v128_load64_splat(params->fp32_wasmsimd.init_bias);
41*4bdc9457SAndroid Build Coastguard Worker int32_t* b = buffer;
42*4bdc9457SAndroid Build Coastguard Worker size_t c = channels;
43*4bdc9457SAndroid Build Coastguard Worker for (; c >= 24; c -= 24) {
44*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi0x01234567 = wasm_i16x8_load8x8(i0);
45*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi0x89ABCDEF = wasm_i16x8_load8x8(i0 + 8);
46*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi0xGHIJKLMN = wasm_i16x8_load8x8(i0 + 16);
47*4bdc9457SAndroid Build Coastguard Worker i0 += 24;
48*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi1x01234567 = wasm_i16x8_load8x8(i1);
49*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi1x89ABCDEF = wasm_i16x8_load8x8(i1 + 8);
50*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi1xGHIJKLMN = wasm_i16x8_load8x8(i1 + 16);
51*4bdc9457SAndroid Build Coastguard Worker i1 += 24;
52*4bdc9457SAndroid Build Coastguard Worker
53*4bdc9457SAndroid Build Coastguard Worker v128_t vacc01234567 = wasm_i16x8_add(vxi0x01234567, vxi1x01234567);
54*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi2x01234567 = wasm_i16x8_load8x8(i2);
55*4bdc9457SAndroid Build Coastguard Worker v128_t vacc89ABCDEF = wasm_i16x8_add(vxi0x89ABCDEF, vxi1x89ABCDEF);
56*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi2x89ABCDEF = wasm_i16x8_load8x8(i2 + 8);
57*4bdc9457SAndroid Build Coastguard Worker v128_t vaccGHIJKLMN = wasm_i16x8_add(vxi0xGHIJKLMN, vxi1xGHIJKLMN);
58*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi2xGHIJKLMN = wasm_i16x8_load8x8(i2 + 16);
59*4bdc9457SAndroid Build Coastguard Worker i2 += 24;
60*4bdc9457SAndroid Build Coastguard Worker
61*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi2x01234567);
62*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi3x01234567 = wasm_i16x8_load8x8(i3);
63*4bdc9457SAndroid Build Coastguard Worker vacc89ABCDEF = wasm_i16x8_add(vacc89ABCDEF, vxi2x89ABCDEF);
64*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi3x89ABCDEF = wasm_i16x8_load8x8(i3 + 8);
65*4bdc9457SAndroid Build Coastguard Worker vaccGHIJKLMN = wasm_i16x8_add(vaccGHIJKLMN, vxi2xGHIJKLMN);
66*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi3xGHIJKLMN = wasm_i16x8_load8x8(i3 + 16);
67*4bdc9457SAndroid Build Coastguard Worker i3 += 24;
68*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi3x01234567);
69*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi4x01234567 = wasm_i16x8_load8x8(i4);
70*4bdc9457SAndroid Build Coastguard Worker vacc89ABCDEF = wasm_i16x8_add(vacc89ABCDEF, vxi3x89ABCDEF);
71*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi4x89ABCDEF = wasm_i16x8_load8x8(i4 + 8);
72*4bdc9457SAndroid Build Coastguard Worker vaccGHIJKLMN = wasm_i16x8_add(vaccGHIJKLMN, vxi3xGHIJKLMN);
73*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi4xGHIJKLMN = wasm_i16x8_load8x8(i4 + 16);
74*4bdc9457SAndroid Build Coastguard Worker i4 += 24;
75*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi4x01234567);
76*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi5x01234567 = wasm_i16x8_load8x8(i5);
77*4bdc9457SAndroid Build Coastguard Worker vacc89ABCDEF = wasm_i16x8_add(vacc89ABCDEF, vxi4x89ABCDEF);
78*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi5x89ABCDEF = wasm_i16x8_load8x8(i5 + 8);
79*4bdc9457SAndroid Build Coastguard Worker vaccGHIJKLMN = wasm_i16x8_add(vaccGHIJKLMN, vxi4xGHIJKLMN);
80*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi5xGHIJKLMN = wasm_i16x8_load8x8(i5 + 16);
81*4bdc9457SAndroid Build Coastguard Worker i5 += 24;
82*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi5x01234567);
83*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi6x01234567 = wasm_i16x8_load8x8(i6);
84*4bdc9457SAndroid Build Coastguard Worker vacc89ABCDEF = wasm_i16x8_add(vacc89ABCDEF, vxi5x89ABCDEF);
85*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi6x89ABCDEF = wasm_i16x8_load8x8(i6 + 8);
86*4bdc9457SAndroid Build Coastguard Worker vaccGHIJKLMN = wasm_i16x8_add(vaccGHIJKLMN, vxi5xGHIJKLMN);
87*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi6xGHIJKLMN = wasm_i16x8_load8x8(i6 + 16);
88*4bdc9457SAndroid Build Coastguard Worker i6 += 24;
89*4bdc9457SAndroid Build Coastguard Worker
90*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi6x01234567);
91*4bdc9457SAndroid Build Coastguard Worker vacc89ABCDEF = wasm_i16x8_add(vacc89ABCDEF, vxi6x89ABCDEF);
92*4bdc9457SAndroid Build Coastguard Worker vaccGHIJKLMN = wasm_i16x8_add(vaccGHIJKLMN, vxi6xGHIJKLMN);
93*4bdc9457SAndroid Build Coastguard Worker
94*4bdc9457SAndroid Build Coastguard Worker const v128_t vacc0123 = wasm_i32x4_add(vinit_bias, wasm_i32x4_extend_low_i16x8(vacc01234567));
95*4bdc9457SAndroid Build Coastguard Worker const v128_t vacc4567 = wasm_i32x4_add(vinit_bias, wasm_i32x4_extend_high_i16x8(vacc01234567));
96*4bdc9457SAndroid Build Coastguard Worker const v128_t vacc89AB = wasm_i32x4_add(vinit_bias, wasm_i32x4_extend_low_i16x8(vacc89ABCDEF));
97*4bdc9457SAndroid Build Coastguard Worker const v128_t vaccCDEF = wasm_i32x4_add(vinit_bias, wasm_i32x4_extend_high_i16x8(vacc89ABCDEF));
98*4bdc9457SAndroid Build Coastguard Worker const v128_t vaccGHIJ = wasm_i32x4_add(vinit_bias, wasm_i32x4_extend_low_i16x8(vaccGHIJKLMN));
99*4bdc9457SAndroid Build Coastguard Worker const v128_t vaccKLMN = wasm_i32x4_add(vinit_bias, wasm_i32x4_extend_high_i16x8(vaccGHIJKLMN));
100*4bdc9457SAndroid Build Coastguard Worker
101*4bdc9457SAndroid Build Coastguard Worker wasm_v128_store(b, vacc0123);
102*4bdc9457SAndroid Build Coastguard Worker wasm_v128_store(b + 4, vacc4567);
103*4bdc9457SAndroid Build Coastguard Worker wasm_v128_store(b + 8, vacc89AB);
104*4bdc9457SAndroid Build Coastguard Worker wasm_v128_store(b + 12, vaccCDEF);
105*4bdc9457SAndroid Build Coastguard Worker wasm_v128_store(b + 16, vaccGHIJ);
106*4bdc9457SAndroid Build Coastguard Worker wasm_v128_store(b + 20, vaccKLMN);
107*4bdc9457SAndroid Build Coastguard Worker b += 24;
108*4bdc9457SAndroid Build Coastguard Worker }
109*4bdc9457SAndroid Build Coastguard Worker if XNN_UNLIKELY(c != 0) {
110*4bdc9457SAndroid Build Coastguard Worker do {
111*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi0x01234567 = wasm_i16x8_load8x8(i0);
112*4bdc9457SAndroid Build Coastguard Worker i0 += 8;
113*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi1x01234567 = wasm_i16x8_load8x8(i1);
114*4bdc9457SAndroid Build Coastguard Worker i1 += 8;
115*4bdc9457SAndroid Build Coastguard Worker
116*4bdc9457SAndroid Build Coastguard Worker v128_t vacc01234567 = wasm_i16x8_add(vxi0x01234567, vxi1x01234567);
117*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi2x01234567 = wasm_i16x8_load8x8(i2);
118*4bdc9457SAndroid Build Coastguard Worker i2 += 8;
119*4bdc9457SAndroid Build Coastguard Worker
120*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi2x01234567);
121*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi3x01234567 = wasm_i16x8_load8x8(i3);
122*4bdc9457SAndroid Build Coastguard Worker i3 += 8;
123*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi3x01234567);
124*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi4x01234567 = wasm_i16x8_load8x8(i4);
125*4bdc9457SAndroid Build Coastguard Worker i4 += 8;
126*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi4x01234567);
127*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi5x01234567 = wasm_i16x8_load8x8(i5);
128*4bdc9457SAndroid Build Coastguard Worker i5 += 8;
129*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi5x01234567);
130*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi6x01234567 = wasm_i16x8_load8x8(i6);
131*4bdc9457SAndroid Build Coastguard Worker i6 += 8;
132*4bdc9457SAndroid Build Coastguard Worker
133*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi6x01234567);
134*4bdc9457SAndroid Build Coastguard Worker
135*4bdc9457SAndroid Build Coastguard Worker const v128_t vacc0123 = wasm_i32x4_add(vinit_bias, wasm_i32x4_extend_low_i16x8(vacc01234567));
136*4bdc9457SAndroid Build Coastguard Worker const v128_t vacc4567 = wasm_i32x4_add(vinit_bias, wasm_i32x4_extend_high_i16x8(vacc01234567));
137*4bdc9457SAndroid Build Coastguard Worker
138*4bdc9457SAndroid Build Coastguard Worker wasm_v128_store(b, vacc0123);
139*4bdc9457SAndroid Build Coastguard Worker wasm_v128_store(b + 4, vacc4567);
140*4bdc9457SAndroid Build Coastguard Worker b += 8;
141*4bdc9457SAndroid Build Coastguard Worker
142*4bdc9457SAndroid Build Coastguard Worker c = doz(c, 8);
143*4bdc9457SAndroid Build Coastguard Worker } while (c != 0);
144*4bdc9457SAndroid Build Coastguard Worker }
145*4bdc9457SAndroid Build Coastguard Worker
146*4bdc9457SAndroid Build Coastguard Worker for (rows -= 7; rows > 7; rows -= 7) {
147*4bdc9457SAndroid Build Coastguard Worker i0 = (const int8_t*) ((uintptr_t) i0 + input_increment);
148*4bdc9457SAndroid Build Coastguard Worker i1 = (const int8_t*) ((uintptr_t) i1 + input_increment);
149*4bdc9457SAndroid Build Coastguard Worker i2 = (const int8_t*) ((uintptr_t) i2 + input_increment);
150*4bdc9457SAndroid Build Coastguard Worker i3 = (const int8_t*) ((uintptr_t) i3 + input_increment);
151*4bdc9457SAndroid Build Coastguard Worker i4 = (const int8_t*) ((uintptr_t) i4 + input_increment);
152*4bdc9457SAndroid Build Coastguard Worker i5 = (const int8_t*) ((uintptr_t) i5 + input_increment);
153*4bdc9457SAndroid Build Coastguard Worker i6 = (const int8_t*) ((uintptr_t) i6 + input_increment);
154*4bdc9457SAndroid Build Coastguard Worker
155*4bdc9457SAndroid Build Coastguard Worker int32_t* b = buffer;
156*4bdc9457SAndroid Build Coastguard Worker size_t c = channels;
157*4bdc9457SAndroid Build Coastguard Worker for (; c >= 24; c -= 24) {
158*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi0x01234567 = wasm_i16x8_load8x8(i0);
159*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi0x89ABCDEF = wasm_i16x8_load8x8(i0 + 8);
160*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi0xGHIJKLMN = wasm_i16x8_load8x8(i0 + 16);
161*4bdc9457SAndroid Build Coastguard Worker i0 += 24;
162*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi1x01234567 = wasm_i16x8_load8x8(i1);
163*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi1x89ABCDEF = wasm_i16x8_load8x8(i1 + 8);
164*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi1xGHIJKLMN = wasm_i16x8_load8x8(i1 + 16);
165*4bdc9457SAndroid Build Coastguard Worker i1 += 24;
166*4bdc9457SAndroid Build Coastguard Worker
167*4bdc9457SAndroid Build Coastguard Worker v128_t vacc01234567 = wasm_i16x8_add(vxi0x01234567, vxi1x01234567);
168*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi2x01234567 = wasm_i16x8_load8x8(i2);
169*4bdc9457SAndroid Build Coastguard Worker v128_t vacc89ABCDEF = wasm_i16x8_add(vxi0x89ABCDEF, vxi1x89ABCDEF);
170*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi2x89ABCDEF = wasm_i16x8_load8x8(i2 + 8);
171*4bdc9457SAndroid Build Coastguard Worker v128_t vaccGHIJKLMN = wasm_i16x8_add(vxi0xGHIJKLMN, vxi1xGHIJKLMN);
172*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi2xGHIJKLMN = wasm_i16x8_load8x8(i2 + 16);
173*4bdc9457SAndroid Build Coastguard Worker i2 += 24;
174*4bdc9457SAndroid Build Coastguard Worker
175*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi2x01234567);
176*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi3x01234567 = wasm_i16x8_load8x8(i3);
177*4bdc9457SAndroid Build Coastguard Worker vacc89ABCDEF = wasm_i16x8_add(vacc89ABCDEF, vxi2x89ABCDEF);
178*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi3x89ABCDEF = wasm_i16x8_load8x8(i3 + 8);
179*4bdc9457SAndroid Build Coastguard Worker vaccGHIJKLMN = wasm_i16x8_add(vaccGHIJKLMN, vxi2xGHIJKLMN);
180*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi3xGHIJKLMN = wasm_i16x8_load8x8(i3 + 16);
181*4bdc9457SAndroid Build Coastguard Worker i3 += 24;
182*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi3x01234567);
183*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi4x01234567 = wasm_i16x8_load8x8(i4);
184*4bdc9457SAndroid Build Coastguard Worker vacc89ABCDEF = wasm_i16x8_add(vacc89ABCDEF, vxi3x89ABCDEF);
185*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi4x89ABCDEF = wasm_i16x8_load8x8(i4 + 8);
186*4bdc9457SAndroid Build Coastguard Worker vaccGHIJKLMN = wasm_i16x8_add(vaccGHIJKLMN, vxi3xGHIJKLMN);
187*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi4xGHIJKLMN = wasm_i16x8_load8x8(i4 + 16);
188*4bdc9457SAndroid Build Coastguard Worker i4 += 24;
189*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi4x01234567);
190*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi5x01234567 = wasm_i16x8_load8x8(i5);
191*4bdc9457SAndroid Build Coastguard Worker vacc89ABCDEF = wasm_i16x8_add(vacc89ABCDEF, vxi4x89ABCDEF);
192*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi5x89ABCDEF = wasm_i16x8_load8x8(i5 + 8);
193*4bdc9457SAndroid Build Coastguard Worker vaccGHIJKLMN = wasm_i16x8_add(vaccGHIJKLMN, vxi4xGHIJKLMN);
194*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi5xGHIJKLMN = wasm_i16x8_load8x8(i5 + 16);
195*4bdc9457SAndroid Build Coastguard Worker i5 += 24;
196*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi5x01234567);
197*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi6x01234567 = wasm_i16x8_load8x8(i6);
198*4bdc9457SAndroid Build Coastguard Worker vacc89ABCDEF = wasm_i16x8_add(vacc89ABCDEF, vxi5x89ABCDEF);
199*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi6x89ABCDEF = wasm_i16x8_load8x8(i6 + 8);
200*4bdc9457SAndroid Build Coastguard Worker vaccGHIJKLMN = wasm_i16x8_add(vaccGHIJKLMN, vxi5xGHIJKLMN);
201*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi6xGHIJKLMN = wasm_i16x8_load8x8(i6 + 16);
202*4bdc9457SAndroid Build Coastguard Worker i6 += 24;
203*4bdc9457SAndroid Build Coastguard Worker
204*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi6x01234567);
205*4bdc9457SAndroid Build Coastguard Worker vacc89ABCDEF = wasm_i16x8_add(vacc89ABCDEF, vxi6x89ABCDEF);
206*4bdc9457SAndroid Build Coastguard Worker vaccGHIJKLMN = wasm_i16x8_add(vaccGHIJKLMN, vxi6xGHIJKLMN);
207*4bdc9457SAndroid Build Coastguard Worker
208*4bdc9457SAndroid Build Coastguard Worker v128_t vacc0123 = wasm_v128_load(b);
209*4bdc9457SAndroid Build Coastguard Worker v128_t vacc4567 = wasm_v128_load(b + 4);
210*4bdc9457SAndroid Build Coastguard Worker v128_t vacc89AB = wasm_v128_load(b + 8);
211*4bdc9457SAndroid Build Coastguard Worker v128_t vaccCDEF = wasm_v128_load(b + 12);
212*4bdc9457SAndroid Build Coastguard Worker v128_t vaccGHIJ = wasm_v128_load(b + 16);
213*4bdc9457SAndroid Build Coastguard Worker v128_t vaccKLMN = wasm_v128_load(b + 20);
214*4bdc9457SAndroid Build Coastguard Worker
215*4bdc9457SAndroid Build Coastguard Worker vacc0123 = wasm_i32x4_add(vacc0123, wasm_i32x4_extend_low_i16x8(vacc01234567));
216*4bdc9457SAndroid Build Coastguard Worker vacc4567 = wasm_i32x4_add(vacc4567, wasm_i32x4_extend_high_i16x8(vacc01234567));
217*4bdc9457SAndroid Build Coastguard Worker vacc89AB = wasm_i32x4_add(vacc89AB, wasm_i32x4_extend_low_i16x8(vacc89ABCDEF));
218*4bdc9457SAndroid Build Coastguard Worker vaccCDEF = wasm_i32x4_add(vaccCDEF, wasm_i32x4_extend_high_i16x8(vacc89ABCDEF));
219*4bdc9457SAndroid Build Coastguard Worker vaccGHIJ = wasm_i32x4_add(vaccGHIJ, wasm_i32x4_extend_low_i16x8(vaccGHIJKLMN));
220*4bdc9457SAndroid Build Coastguard Worker vaccKLMN = wasm_i32x4_add(vaccKLMN, wasm_i32x4_extend_high_i16x8(vaccGHIJKLMN));
221*4bdc9457SAndroid Build Coastguard Worker
222*4bdc9457SAndroid Build Coastguard Worker wasm_v128_store(b, vacc0123);
223*4bdc9457SAndroid Build Coastguard Worker wasm_v128_store(b + 4, vacc4567);
224*4bdc9457SAndroid Build Coastguard Worker wasm_v128_store(b + 8, vacc89AB);
225*4bdc9457SAndroid Build Coastguard Worker wasm_v128_store(b + 12, vaccCDEF);
226*4bdc9457SAndroid Build Coastguard Worker wasm_v128_store(b + 16, vaccGHIJ);
227*4bdc9457SAndroid Build Coastguard Worker wasm_v128_store(b + 20, vaccKLMN);
228*4bdc9457SAndroid Build Coastguard Worker b += 24;
229*4bdc9457SAndroid Build Coastguard Worker }
230*4bdc9457SAndroid Build Coastguard Worker if XNN_UNLIKELY(c != 0) {
231*4bdc9457SAndroid Build Coastguard Worker do {
232*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi0x01234567 = wasm_i16x8_load8x8(i0);
233*4bdc9457SAndroid Build Coastguard Worker i0 += 8;
234*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi1x01234567 = wasm_i16x8_load8x8(i1);
235*4bdc9457SAndroid Build Coastguard Worker i1 += 8;
236*4bdc9457SAndroid Build Coastguard Worker
237*4bdc9457SAndroid Build Coastguard Worker v128_t vacc01234567 = wasm_i16x8_add(vxi0x01234567, vxi1x01234567);
238*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi2x01234567 = wasm_i16x8_load8x8(i2);
239*4bdc9457SAndroid Build Coastguard Worker i2 += 8;
240*4bdc9457SAndroid Build Coastguard Worker
241*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi2x01234567);
242*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi3x01234567 = wasm_i16x8_load8x8(i3);
243*4bdc9457SAndroid Build Coastguard Worker i3 += 8;
244*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi3x01234567);
245*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi4x01234567 = wasm_i16x8_load8x8(i4);
246*4bdc9457SAndroid Build Coastguard Worker i4 += 8;
247*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi4x01234567);
248*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi5x01234567 = wasm_i16x8_load8x8(i5);
249*4bdc9457SAndroid Build Coastguard Worker i5 += 8;
250*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi5x01234567);
251*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi6x01234567 = wasm_i16x8_load8x8(i6);
252*4bdc9457SAndroid Build Coastguard Worker i6 += 8;
253*4bdc9457SAndroid Build Coastguard Worker
254*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi6x01234567);
255*4bdc9457SAndroid Build Coastguard Worker
256*4bdc9457SAndroid Build Coastguard Worker v128_t vacc0123 = wasm_v128_load(b);
257*4bdc9457SAndroid Build Coastguard Worker v128_t vacc4567 = wasm_v128_load(b + 4);
258*4bdc9457SAndroid Build Coastguard Worker
259*4bdc9457SAndroid Build Coastguard Worker vacc0123 = wasm_i32x4_add(vacc0123, wasm_i32x4_extend_low_i16x8(vacc01234567));
260*4bdc9457SAndroid Build Coastguard Worker vacc4567 = wasm_i32x4_add(vacc4567, wasm_i32x4_extend_high_i16x8(vacc01234567));
261*4bdc9457SAndroid Build Coastguard Worker
262*4bdc9457SAndroid Build Coastguard Worker wasm_v128_store(b, vacc0123);
263*4bdc9457SAndroid Build Coastguard Worker wasm_v128_store(b + 4, vacc4567);
264*4bdc9457SAndroid Build Coastguard Worker b += 8;
265*4bdc9457SAndroid Build Coastguard Worker
266*4bdc9457SAndroid Build Coastguard Worker c = doz(c, 8);
267*4bdc9457SAndroid Build Coastguard Worker } while (c != 0);
268*4bdc9457SAndroid Build Coastguard Worker }
269*4bdc9457SAndroid Build Coastguard Worker }
270*4bdc9457SAndroid Build Coastguard Worker
271*4bdc9457SAndroid Build Coastguard Worker i0 = (const int8_t*) ((uintptr_t) i0 + input_increment);
272*4bdc9457SAndroid Build Coastguard Worker i1 = (const int8_t*) ((uintptr_t) i1 + input_increment);
273*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(rows < 2) {
274*4bdc9457SAndroid Build Coastguard Worker i1 = zero;
275*4bdc9457SAndroid Build Coastguard Worker }
276*4bdc9457SAndroid Build Coastguard Worker i2 = (const int8_t*) ((uintptr_t) i2 + input_increment);
277*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(rows <= 2) {
278*4bdc9457SAndroid Build Coastguard Worker i2 = zero;
279*4bdc9457SAndroid Build Coastguard Worker }
280*4bdc9457SAndroid Build Coastguard Worker i3 = (const int8_t*) ((uintptr_t) i3 + input_increment);
281*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(rows < 4) {
282*4bdc9457SAndroid Build Coastguard Worker i3 = zero;
283*4bdc9457SAndroid Build Coastguard Worker }
284*4bdc9457SAndroid Build Coastguard Worker i4 = (const int8_t*) ((uintptr_t) i4 + input_increment);
285*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(rows <= 4) {
286*4bdc9457SAndroid Build Coastguard Worker i4 = zero;
287*4bdc9457SAndroid Build Coastguard Worker }
288*4bdc9457SAndroid Build Coastguard Worker i5 = (const int8_t*) ((uintptr_t) i5 + input_increment);
289*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(rows < 6) {
290*4bdc9457SAndroid Build Coastguard Worker i5 = zero;
291*4bdc9457SAndroid Build Coastguard Worker }
292*4bdc9457SAndroid Build Coastguard Worker i6 = (const int8_t*) ((uintptr_t) i6 + input_increment);
293*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(rows <= 6) {
294*4bdc9457SAndroid Build Coastguard Worker i6 = zero;
295*4bdc9457SAndroid Build Coastguard Worker }
296*4bdc9457SAndroid Build Coastguard Worker
297*4bdc9457SAndroid Build Coastguard Worker const v128_t vscale = wasm_v128_load64_splat(params->fp32_wasmsimd.scale);
298*4bdc9457SAndroid Build Coastguard Worker const v128_t vmagic_bias = wasm_v128_load64_splat(params->fp32_wasmsimd.magic_bias);
299*4bdc9457SAndroid Build Coastguard Worker const v128_t vmagic_min = wasm_v128_load64_splat(params->fp32_wasmsimd.magic_min);
300*4bdc9457SAndroid Build Coastguard Worker const v128_t vmagic_bias_less_output_zero_point = wasm_v128_load64_splat(params->fp32_wasmsimd.magic_bias_less_output_zero_point);
301*4bdc9457SAndroid Build Coastguard Worker const v128_t voutput_max = wasm_v128_load64_splat(params->fp32_wasmsimd.output_max);
302*4bdc9457SAndroid Build Coastguard Worker for (; channels >= 24; channels -= 24) {
303*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi0x01234567 = wasm_i16x8_load8x8(i0);
304*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi0x89ABCDEF = wasm_i16x8_load8x8(i0 + 8);
305*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi0xGHIJKLMN = wasm_i16x8_load8x8(i0 + 16);
306*4bdc9457SAndroid Build Coastguard Worker i0 += 24;
307*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi1x01234567 = wasm_i16x8_load8x8(i1);
308*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi1x89ABCDEF = wasm_i16x8_load8x8(i1 + 8);
309*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi1xGHIJKLMN = wasm_i16x8_load8x8(i1 + 16);
310*4bdc9457SAndroid Build Coastguard Worker i1 += 24;
311*4bdc9457SAndroid Build Coastguard Worker
312*4bdc9457SAndroid Build Coastguard Worker v128_t vacc01234567 = wasm_i16x8_add(vxi0x01234567, vxi1x01234567);
313*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi2x01234567 = wasm_i16x8_load8x8(i2);
314*4bdc9457SAndroid Build Coastguard Worker v128_t vacc89ABCDEF = wasm_i16x8_add(vxi0x89ABCDEF, vxi1x89ABCDEF);
315*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi2x89ABCDEF = wasm_i16x8_load8x8(i2 + 8);
316*4bdc9457SAndroid Build Coastguard Worker v128_t vaccGHIJKLMN = wasm_i16x8_add(vxi0xGHIJKLMN, vxi1xGHIJKLMN);
317*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi2xGHIJKLMN = wasm_i16x8_load8x8(i2 + 16);
318*4bdc9457SAndroid Build Coastguard Worker i2 += 24;
319*4bdc9457SAndroid Build Coastguard Worker
320*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi2x01234567);
321*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi3x01234567 = wasm_i16x8_load8x8(i3);
322*4bdc9457SAndroid Build Coastguard Worker vacc89ABCDEF = wasm_i16x8_add(vacc89ABCDEF, vxi2x89ABCDEF);
323*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi3x89ABCDEF = wasm_i16x8_load8x8(i3 + 8);
324*4bdc9457SAndroid Build Coastguard Worker vaccGHIJKLMN = wasm_i16x8_add(vaccGHIJKLMN, vxi2xGHIJKLMN);
325*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi3xGHIJKLMN = wasm_i16x8_load8x8(i3 + 16);
326*4bdc9457SAndroid Build Coastguard Worker i3 += 24;
327*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi3x01234567);
328*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi4x01234567 = wasm_i16x8_load8x8(i4);
329*4bdc9457SAndroid Build Coastguard Worker vacc89ABCDEF = wasm_i16x8_add(vacc89ABCDEF, vxi3x89ABCDEF);
330*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi4x89ABCDEF = wasm_i16x8_load8x8(i4 + 8);
331*4bdc9457SAndroid Build Coastguard Worker vaccGHIJKLMN = wasm_i16x8_add(vaccGHIJKLMN, vxi3xGHIJKLMN);
332*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi4xGHIJKLMN = wasm_i16x8_load8x8(i4 + 16);
333*4bdc9457SAndroid Build Coastguard Worker i4 += 24;
334*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi4x01234567);
335*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi5x01234567 = wasm_i16x8_load8x8(i5);
336*4bdc9457SAndroid Build Coastguard Worker vacc89ABCDEF = wasm_i16x8_add(vacc89ABCDEF, vxi4x89ABCDEF);
337*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi5x89ABCDEF = wasm_i16x8_load8x8(i5 + 8);
338*4bdc9457SAndroid Build Coastguard Worker vaccGHIJKLMN = wasm_i16x8_add(vaccGHIJKLMN, vxi4xGHIJKLMN);
339*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi5xGHIJKLMN = wasm_i16x8_load8x8(i5 + 16);
340*4bdc9457SAndroid Build Coastguard Worker i5 += 24;
341*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi5x01234567);
342*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi6x01234567 = wasm_i16x8_load8x8(i6);
343*4bdc9457SAndroid Build Coastguard Worker vacc89ABCDEF = wasm_i16x8_add(vacc89ABCDEF, vxi5x89ABCDEF);
344*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi6x89ABCDEF = wasm_i16x8_load8x8(i6 + 8);
345*4bdc9457SAndroid Build Coastguard Worker vaccGHIJKLMN = wasm_i16x8_add(vaccGHIJKLMN, vxi5xGHIJKLMN);
346*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi6xGHIJKLMN = wasm_i16x8_load8x8(i6 + 16);
347*4bdc9457SAndroid Build Coastguard Worker i6 += 24;
348*4bdc9457SAndroid Build Coastguard Worker
349*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi6x01234567);
350*4bdc9457SAndroid Build Coastguard Worker vacc89ABCDEF = wasm_i16x8_add(vacc89ABCDEF, vxi6x89ABCDEF);
351*4bdc9457SAndroid Build Coastguard Worker vaccGHIJKLMN = wasm_i16x8_add(vaccGHIJKLMN, vxi6xGHIJKLMN);
352*4bdc9457SAndroid Build Coastguard Worker
353*4bdc9457SAndroid Build Coastguard Worker v128_t vacc0123 = wasm_v128_load(buffer);
354*4bdc9457SAndroid Build Coastguard Worker v128_t vacc4567 = wasm_v128_load(buffer + 4);
355*4bdc9457SAndroid Build Coastguard Worker v128_t vacc89AB = wasm_v128_load(buffer + 8);
356*4bdc9457SAndroid Build Coastguard Worker v128_t vaccCDEF = wasm_v128_load(buffer + 12);
357*4bdc9457SAndroid Build Coastguard Worker v128_t vaccGHIJ = wasm_v128_load(buffer + 16);
358*4bdc9457SAndroid Build Coastguard Worker v128_t vaccKLMN = wasm_v128_load(buffer + 20);
359*4bdc9457SAndroid Build Coastguard Worker buffer += 24;
360*4bdc9457SAndroid Build Coastguard Worker
361*4bdc9457SAndroid Build Coastguard Worker vacc0123 = wasm_i32x4_add(vacc0123, wasm_i32x4_extend_low_i16x8(vacc01234567));
362*4bdc9457SAndroid Build Coastguard Worker vacc4567 = wasm_i32x4_add(vacc4567, wasm_i32x4_extend_high_i16x8(vacc01234567));
363*4bdc9457SAndroid Build Coastguard Worker vacc89AB = wasm_i32x4_add(vacc89AB, wasm_i32x4_extend_low_i16x8(vacc89ABCDEF));
364*4bdc9457SAndroid Build Coastguard Worker vaccCDEF = wasm_i32x4_add(vaccCDEF, wasm_i32x4_extend_high_i16x8(vacc89ABCDEF));
365*4bdc9457SAndroid Build Coastguard Worker vaccGHIJ = wasm_i32x4_add(vaccGHIJ, wasm_i32x4_extend_low_i16x8(vaccGHIJKLMN));
366*4bdc9457SAndroid Build Coastguard Worker vaccKLMN = wasm_i32x4_add(vaccKLMN, wasm_i32x4_extend_high_i16x8(vaccGHIJKLMN));
367*4bdc9457SAndroid Build Coastguard Worker
368*4bdc9457SAndroid Build Coastguard Worker vacc0123 = wasm_f32x4_convert_i32x4(vacc0123);
369*4bdc9457SAndroid Build Coastguard Worker vacc4567 = wasm_f32x4_convert_i32x4(vacc4567);
370*4bdc9457SAndroid Build Coastguard Worker vacc89AB = wasm_f32x4_convert_i32x4(vacc89AB);
371*4bdc9457SAndroid Build Coastguard Worker vaccCDEF = wasm_f32x4_convert_i32x4(vaccCDEF);
372*4bdc9457SAndroid Build Coastguard Worker vaccGHIJ = wasm_f32x4_convert_i32x4(vaccGHIJ);
373*4bdc9457SAndroid Build Coastguard Worker vaccKLMN = wasm_f32x4_convert_i32x4(vaccKLMN);
374*4bdc9457SAndroid Build Coastguard Worker
375*4bdc9457SAndroid Build Coastguard Worker vacc0123 = wasm_f32x4_mul(vacc0123, vscale);
376*4bdc9457SAndroid Build Coastguard Worker vacc4567 = wasm_f32x4_mul(vacc4567, vscale);
377*4bdc9457SAndroid Build Coastguard Worker vacc89AB = wasm_f32x4_mul(vacc89AB, vscale);
378*4bdc9457SAndroid Build Coastguard Worker vaccCDEF = wasm_f32x4_mul(vaccCDEF, vscale);
379*4bdc9457SAndroid Build Coastguard Worker vaccGHIJ = wasm_f32x4_mul(vaccGHIJ, vscale);
380*4bdc9457SAndroid Build Coastguard Worker vaccKLMN = wasm_f32x4_mul(vaccKLMN, vscale);
381*4bdc9457SAndroid Build Coastguard Worker
382*4bdc9457SAndroid Build Coastguard Worker vacc0123 = wasm_f32x4_add(vacc0123, vmagic_bias);
383*4bdc9457SAndroid Build Coastguard Worker vacc4567 = wasm_f32x4_add(vacc4567, vmagic_bias);
384*4bdc9457SAndroid Build Coastguard Worker vacc89AB = wasm_f32x4_add(vacc89AB, vmagic_bias);
385*4bdc9457SAndroid Build Coastguard Worker vaccCDEF = wasm_f32x4_add(vaccCDEF, vmagic_bias);
386*4bdc9457SAndroid Build Coastguard Worker vaccGHIJ = wasm_f32x4_add(vaccGHIJ, vmagic_bias);
387*4bdc9457SAndroid Build Coastguard Worker vaccKLMN = wasm_f32x4_add(vaccKLMN, vmagic_bias);
388*4bdc9457SAndroid Build Coastguard Worker
389*4bdc9457SAndroid Build Coastguard Worker vacc0123 = wasm_i32x4_max(vacc0123, vmagic_min);
390*4bdc9457SAndroid Build Coastguard Worker vacc4567 = wasm_i32x4_max(vacc4567, vmagic_min);
391*4bdc9457SAndroid Build Coastguard Worker vacc89AB = wasm_i32x4_max(vacc89AB, vmagic_min);
392*4bdc9457SAndroid Build Coastguard Worker vaccCDEF = wasm_i32x4_max(vaccCDEF, vmagic_min);
393*4bdc9457SAndroid Build Coastguard Worker vaccGHIJ = wasm_i32x4_max(vaccGHIJ, vmagic_min);
394*4bdc9457SAndroid Build Coastguard Worker vaccKLMN = wasm_i32x4_max(vaccKLMN, vmagic_min);
395*4bdc9457SAndroid Build Coastguard Worker
396*4bdc9457SAndroid Build Coastguard Worker vacc0123 = wasm_i32x4_sub(vacc0123, vmagic_bias_less_output_zero_point);
397*4bdc9457SAndroid Build Coastguard Worker vacc4567 = wasm_i32x4_sub(vacc4567, vmagic_bias_less_output_zero_point);
398*4bdc9457SAndroid Build Coastguard Worker vacc89AB = wasm_i32x4_sub(vacc89AB, vmagic_bias_less_output_zero_point);
399*4bdc9457SAndroid Build Coastguard Worker vaccCDEF = wasm_i32x4_sub(vaccCDEF, vmagic_bias_less_output_zero_point);
400*4bdc9457SAndroid Build Coastguard Worker vaccGHIJ = wasm_i32x4_sub(vaccGHIJ, vmagic_bias_less_output_zero_point);
401*4bdc9457SAndroid Build Coastguard Worker vaccKLMN = wasm_i32x4_sub(vaccKLMN, vmagic_bias_less_output_zero_point);
402*4bdc9457SAndroid Build Coastguard Worker
403*4bdc9457SAndroid Build Coastguard Worker v128_t vout01234567 = wasm_i16x8_narrow_i32x4(vacc0123, vacc4567);
404*4bdc9457SAndroid Build Coastguard Worker v128_t vout89ABCDEF = wasm_i16x8_narrow_i32x4(vacc89AB, vaccCDEF);
405*4bdc9457SAndroid Build Coastguard Worker v128_t voutGHIJKLMN = wasm_i16x8_narrow_i32x4(vaccGHIJ, vaccKLMN);
406*4bdc9457SAndroid Build Coastguard Worker
407*4bdc9457SAndroid Build Coastguard Worker v128_t vout0123456789ABCDEF = wasm_i8x16_narrow_i16x8(vout01234567, vout89ABCDEF);
408*4bdc9457SAndroid Build Coastguard Worker v128_t voutGHIJKLMNGHIJKLMN = wasm_i8x16_narrow_i16x8(voutGHIJKLMN, voutGHIJKLMN);
409*4bdc9457SAndroid Build Coastguard Worker
410*4bdc9457SAndroid Build Coastguard Worker vout0123456789ABCDEF = wasm_i8x16_min(vout0123456789ABCDEF, voutput_max);
411*4bdc9457SAndroid Build Coastguard Worker voutGHIJKLMNGHIJKLMN = wasm_i8x16_min(voutGHIJKLMNGHIJKLMN, voutput_max);
412*4bdc9457SAndroid Build Coastguard Worker
413*4bdc9457SAndroid Build Coastguard Worker wasm_v128_store(output, vout0123456789ABCDEF);
414*4bdc9457SAndroid Build Coastguard Worker *((double*) (output + 16)) = wasm_f64x2_extract_lane(voutGHIJKLMNGHIJKLMN, 0);
415*4bdc9457SAndroid Build Coastguard Worker output += 24;
416*4bdc9457SAndroid Build Coastguard Worker }
417*4bdc9457SAndroid Build Coastguard Worker if XNN_UNLIKELY(channels != 0) {
418*4bdc9457SAndroid Build Coastguard Worker do {
419*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi0x01234567 = wasm_i16x8_load8x8(i0);
420*4bdc9457SAndroid Build Coastguard Worker i0 += 8;
421*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi1x01234567 = wasm_i16x8_load8x8(i1);
422*4bdc9457SAndroid Build Coastguard Worker i1 += 8;
423*4bdc9457SAndroid Build Coastguard Worker
424*4bdc9457SAndroid Build Coastguard Worker v128_t vacc01234567 = wasm_i16x8_add(vxi0x01234567, vxi1x01234567);
425*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi2x01234567 = wasm_i16x8_load8x8(i2);
426*4bdc9457SAndroid Build Coastguard Worker i2 += 8;
427*4bdc9457SAndroid Build Coastguard Worker
428*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi2x01234567);
429*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi3x01234567 = wasm_i16x8_load8x8(i3);
430*4bdc9457SAndroid Build Coastguard Worker i3 += 8;
431*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi3x01234567);
432*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi4x01234567 = wasm_i16x8_load8x8(i4);
433*4bdc9457SAndroid Build Coastguard Worker i4 += 8;
434*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi4x01234567);
435*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi5x01234567 = wasm_i16x8_load8x8(i5);
436*4bdc9457SAndroid Build Coastguard Worker i5 += 8;
437*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi5x01234567);
438*4bdc9457SAndroid Build Coastguard Worker const v128_t vxi6x01234567 = wasm_i16x8_load8x8(i6);
439*4bdc9457SAndroid Build Coastguard Worker i6 += 8;
440*4bdc9457SAndroid Build Coastguard Worker
441*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = wasm_i16x8_add(vacc01234567, vxi6x01234567);
442*4bdc9457SAndroid Build Coastguard Worker
443*4bdc9457SAndroid Build Coastguard Worker v128_t vacc0123 = wasm_v128_load(buffer);
444*4bdc9457SAndroid Build Coastguard Worker v128_t vacc4567 = wasm_v128_load(buffer + 4);
445*4bdc9457SAndroid Build Coastguard Worker buffer += 8;
446*4bdc9457SAndroid Build Coastguard Worker
447*4bdc9457SAndroid Build Coastguard Worker vacc0123 = wasm_i32x4_add(vacc0123, wasm_i32x4_extend_low_i16x8(vacc01234567));
448*4bdc9457SAndroid Build Coastguard Worker vacc4567 = wasm_i32x4_add(vacc4567, wasm_i32x4_extend_high_i16x8(vacc01234567));
449*4bdc9457SAndroid Build Coastguard Worker
450*4bdc9457SAndroid Build Coastguard Worker vacc0123 = wasm_f32x4_convert_i32x4(vacc0123);
451*4bdc9457SAndroid Build Coastguard Worker vacc4567 = wasm_f32x4_convert_i32x4(vacc4567);
452*4bdc9457SAndroid Build Coastguard Worker
453*4bdc9457SAndroid Build Coastguard Worker vacc0123 = wasm_f32x4_mul(vacc0123, vscale);
454*4bdc9457SAndroid Build Coastguard Worker vacc4567 = wasm_f32x4_mul(vacc4567, vscale);
455*4bdc9457SAndroid Build Coastguard Worker
456*4bdc9457SAndroid Build Coastguard Worker vacc0123 = wasm_f32x4_add(vacc0123, vmagic_bias);
457*4bdc9457SAndroid Build Coastguard Worker vacc4567 = wasm_f32x4_add(vacc4567, vmagic_bias);
458*4bdc9457SAndroid Build Coastguard Worker
459*4bdc9457SAndroid Build Coastguard Worker vacc0123 = wasm_i32x4_max(vacc0123, vmagic_min);
460*4bdc9457SAndroid Build Coastguard Worker vacc4567 = wasm_i32x4_max(vacc4567, vmagic_min);
461*4bdc9457SAndroid Build Coastguard Worker
462*4bdc9457SAndroid Build Coastguard Worker vacc0123 = wasm_i32x4_sub(vacc0123, vmagic_bias_less_output_zero_point);
463*4bdc9457SAndroid Build Coastguard Worker vacc4567 = wasm_i32x4_sub(vacc4567, vmagic_bias_less_output_zero_point);
464*4bdc9457SAndroid Build Coastguard Worker
465*4bdc9457SAndroid Build Coastguard Worker const v128_t vout01234567 = wasm_i16x8_narrow_i32x4(vacc0123, vacc4567);
466*4bdc9457SAndroid Build Coastguard Worker v128_t vout0123456701234567 = wasm_i8x16_narrow_i16x8(vout01234567, vout01234567);
467*4bdc9457SAndroid Build Coastguard Worker vout0123456701234567 = wasm_i8x16_min(vout0123456701234567, voutput_max);
468*4bdc9457SAndroid Build Coastguard Worker
469*4bdc9457SAndroid Build Coastguard Worker if XNN_LIKELY(channels >= 8) {
470*4bdc9457SAndroid Build Coastguard Worker *((double*) output) = wasm_f64x2_extract_lane(vout0123456701234567, 0);
471*4bdc9457SAndroid Build Coastguard Worker output += 8;
472*4bdc9457SAndroid Build Coastguard Worker channels -= 8;
473*4bdc9457SAndroid Build Coastguard Worker } else {
474*4bdc9457SAndroid Build Coastguard Worker if (channels & 4) {
475*4bdc9457SAndroid Build Coastguard Worker *((float*) output) = wasm_f32x4_extract_lane(vout0123456701234567, 0);
476*4bdc9457SAndroid Build Coastguard Worker vout0123456701234567 = wasm_u64x2_shr(vout0123456701234567, 32);
477*4bdc9457SAndroid Build Coastguard Worker output += 4;
478*4bdc9457SAndroid Build Coastguard Worker }
479*4bdc9457SAndroid Build Coastguard Worker uint32_t vout0123 = wasm_i32x4_extract_lane(vout0123456701234567, 0);
480*4bdc9457SAndroid Build Coastguard Worker if (channels & 2) {
481*4bdc9457SAndroid Build Coastguard Worker *((uint16_t*) output) = (uint16_t) vout0123;
482*4bdc9457SAndroid Build Coastguard Worker vout0123 >>= 16;
483*4bdc9457SAndroid Build Coastguard Worker output += 2;
484*4bdc9457SAndroid Build Coastguard Worker }
485*4bdc9457SAndroid Build Coastguard Worker if (channels & 1) {
486*4bdc9457SAndroid Build Coastguard Worker *output = (int8_t) vout0123;
487*4bdc9457SAndroid Build Coastguard Worker output += 1;
488*4bdc9457SAndroid Build Coastguard Worker }
489*4bdc9457SAndroid Build Coastguard Worker channels = 0;
490*4bdc9457SAndroid Build Coastguard Worker }
491*4bdc9457SAndroid Build Coastguard Worker } while (channels != 0);
492*4bdc9457SAndroid Build Coastguard Worker }
493*4bdc9457SAndroid Build Coastguard Worker }
494