24.04
generic.cpp
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2021, 2023 Arm Limited.
3
*
4
* SPDX-License-Identifier: MIT
5
*
6
* Permission is hereby granted, free of charge, to any person obtaining a copy
7
* of this software and associated documentation files (the "Software"), to
8
* deal in the Software without restriction, including without limitation the
9
* rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10
* sell copies of the Software, and to permit persons to whom the Software is
11
* furnished to do so, subject to the following conditions:
12
*
13
* The above copyright notice and this permission notice shall be included in all
14
* copies or substantial portions of the Software.
15
*
16
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22
* SOFTWARE.
23
*/
24
25
#include <cstddef>
26
#include <cstdint>
27
28
#if defined(__aarch64__)
29
30
namespace
arm_conv
{
31
namespace
depthwise
{
32
33
void
a64_fp32_packed_to_nhwc_3x3_s2_with_multiplier_output3x3_mla_depthfirst_impl(
34
const
float
*
const
*
const
inptrs,
35
float
*
const
*
const
outptrs,
36
const
void
*params,
37
const
unsigned
int
n_output_channels,
38
const
float
activation_min
,
39
const
float
activation_max
40
)
41
{
42
const
float
minmax_vals[2] = {
activation_min
,
activation_max
};
43
44
__asm__ __volatile__(
45
"ld1r { v27.4s }, [%x[clamps]]\n"
46
"ldr x21, [%x[inptrs], #0x0]\n"
47
"lsr x22, %x[channel_multiplier], #0x2\n"
48
"add x20, %x[clamps], #0x4\n"
49
"ldr q0, [x21, #0x0]\n"
50
"ldr q1, [x21, #0x10]\n"
51
"mov x21, #0x0\n"
52
"mov x14, #0x0\n"
53
"ld1r { v26.4s }, [x20]\n"
54
"ldr x20, [%x[inptrs], #0x8]\n"
55
"ldr q2, [x20, #0x0]\n"
56
"ldr q3, [x20, #0x10]\n"
57
"ldr x20, [%x[inptrs], #0x10]\n"
58
"ldr q4, [x20, #0x0]\n"
59
"ldr q5, [x20, #0x10]\n"
60
"ldr x20, [%x[inptrs], #0x18]\n"
61
"ldr q6, [x20, #0x0]\n"
62
"ldr q7, [x20, #0x10]\n"
63
"ldr x20, [%x[inptrs], #0x20]\n"
64
"ldr q8, [x20, #0x0]\n"
65
"ldr q9, [x20, #0x10]\n"
66
"ldr x20, [%x[inptrs], #0x28]\n"
67
"ldr q10, [x20, #0x0]\n"
68
"ldr q11, [x20, #0x10]\n"
69
"ldr x20, [%x[inptrs], #0x30]\n"
70
"ldr q12, [x20, #0x0]\n"
71
"ldr q13, [x20, #0x10]\n"
72
"ldp x13, x12, [%x[outptrs], #0x0]\n"
73
"ldp x11, x10, [%x[outptrs], #0x10]\n"
74
"ldp x9, x28, [%x[outptrs], #0x20]\n"
75
"ldp x27, x26, [%x[outptrs], #0x30]\n"
76
"ldr x25, [%x[outptrs], #0x40]\n"
77
"cbz x22, 3f\n"
78
"ldr q14, [%x[params], #0x0]\n"
79
"ldr q31, [%x[params], #0x10]\n"
80
"subs x22, x22, #0x1\n"
81
"mov v15.16b, v14.16b\n"
82
"ldr q30, [%x[params], #0x20]\n"
83
"ldr q29, [%x[params], #0x30]\n"
84
"mov v16.16b, v14.16b\n"
85
"mov v17.16b, v14.16b\n"
86
"mov v18.16b, v14.16b\n"
87
"mov v19.16b, v14.16b\n"
88
"add %x[params], %x[params], #0x40\n"
89
"mov v20.16b, v14.16b\n"
90
"mov v21.16b, v14.16b\n"
91
"mov v22.16b, v14.16b\n"
92
"beq 2f\n"
93
"1:"
// Output channel complete vector loop
94
"fmla v14.4s, v31.4s, v0.s[0]\n"
95
"fmla v15.4s, v31.4s, v0.s[2]\n"
96
"subs x22, x22, #0x1\n"
97
"add x21, x21, #0x4\n"
98
"fmla v16.4s, v31.4s, v1.s[0]\n"
99
"fmla v17.4s, v31.4s, v4.s[0]\n"
100
"fmla v18.4s, v31.4s, v4.s[2]\n"
101
"fmla v19.4s, v31.4s, v5.s[0]\n"
102
"fmla v20.4s, v31.4s, v8.s[0]\n"
103
"fmla v21.4s, v31.4s, v8.s[2]\n"
104
"fmla v22.4s, v31.4s, v9.s[0]\n"
105
"ldr q25, [%x[params], #0x0]\n"
106
"fmla v14.4s, v30.4s, v0.s[1]\n"
107
"fmla v15.4s, v30.4s, v0.s[3]\n"
108
"fmla v16.4s, v30.4s, v1.s[1]\n"
109
"fmla v17.4s, v30.4s, v4.s[1]\n"
110
"fmla v18.4s, v30.4s, v4.s[3]\n"
111
"fmla v19.4s, v30.4s, v5.s[1]\n"
112
"fmla v20.4s, v30.4s, v8.s[1]\n"
113
"fmla v21.4s, v30.4s, v8.s[3]\n"
114
"fmla v22.4s, v30.4s, v9.s[1]\n"
115
"ldr q24, [%x[params], #0x10]\n"
116
"fmla v14.4s, v29.4s, v0.s[2]\n"
117
"fmla v15.4s, v29.4s, v1.s[0]\n"
118
"fmla v16.4s, v29.4s, v1.s[2]\n"
119
"fmla v17.4s, v29.4s, v4.s[2]\n"
120
"fmla v18.4s, v29.4s, v5.s[0]\n"
121
"fmla v19.4s, v29.4s, v5.s[2]\n"
122
"fmla v20.4s, v29.4s, v8.s[2]\n"
123
"fmla v21.4s, v29.4s, v9.s[0]\n"
124
"fmla v22.4s, v29.4s, v9.s[2]\n"
125
"ldr q23, [%x[params], #0x20]\n"
126
"fmla v14.4s, v25.4s, v2.s[0]\n"
127
"fmla v15.4s, v25.4s, v2.s[2]\n"
128
"fmla v16.4s, v25.4s, v3.s[0]\n"
129
"fmla v17.4s, v25.4s, v6.s[0]\n"
130
"fmla v18.4s, v25.4s, v6.s[2]\n"
131
"fmla v19.4s, v25.4s, v7.s[0]\n"
132
"fmla v20.4s, v25.4s, v10.s[0]\n"
133
"fmla v21.4s, v25.4s, v10.s[2]\n"
134
"fmla v22.4s, v25.4s, v11.s[0]\n"
135
"ldr q25, [%x[params], #0x30]\n"
136
"fmla v14.4s, v24.4s, v2.s[1]\n"
137
"fmla v15.4s, v24.4s, v2.s[3]\n"
138
"fmla v16.4s, v24.4s, v3.s[1]\n"
139
"fmla v17.4s, v24.4s, v6.s[1]\n"
140
"fmla v18.4s, v24.4s, v6.s[3]\n"
141
"fmla v19.4s, v24.4s, v7.s[1]\n"
142
"fmla v20.4s, v24.4s, v10.s[1]\n"
143
"fmla v21.4s, v24.4s, v10.s[3]\n"
144
"fmla v22.4s, v24.4s, v11.s[1]\n"
145
"ldr q24, [%x[params], #0x40]\n"
146
"fmla v14.4s, v23.4s, v2.s[2]\n"
147
"fmla v15.4s, v23.4s, v3.s[0]\n"
148
"fmla v16.4s, v23.4s, v3.s[2]\n"
149
"fmla v17.4s, v23.4s, v6.s[2]\n"
150
"fmla v18.4s, v23.4s, v7.s[0]\n"
151
"fmla v19.4s, v23.4s, v7.s[2]\n"
152
"fmla v20.4s, v23.4s, v10.s[2]\n"
153
"fmla v21.4s, v23.4s, v11.s[0]\n"
154
"fmla v22.4s, v23.4s, v11.s[2]\n"
155
"ldr q23, [%x[params], #0x50]\n"
156
"fmla v14.4s, v25.4s, v4.s[0]\n"
157
"fmla v15.4s, v25.4s, v4.s[2]\n"
158
"fmla v16.4s, v25.4s, v5.s[0]\n"
159
"fmla v17.4s, v25.4s, v8.s[0]\n"
160
"fmla v18.4s, v25.4s, v8.s[2]\n"
161
"fmla v19.4s, v25.4s, v9.s[0]\n"
162
"fmla v20.4s, v25.4s, v12.s[0]\n"
163
"fmla v21.4s, v25.4s, v12.s[2]\n"
164
"fmla v22.4s, v25.4s, v13.s[0]\n"
165
"ldr q31, [%x[params], #0x70]\n"
166
"fmla v14.4s, v24.4s, v4.s[1]\n"
167
"fmla v15.4s, v24.4s, v4.s[3]\n"
168
"fmla v16.4s, v24.4s, v5.s[1]\n"
169
"fmla v17.4s, v24.4s, v8.s[1]\n"
170
"fmla v18.4s, v24.4s, v8.s[3]\n"
171
"fmla v19.4s, v24.4s, v9.s[1]\n"
172
"fmla v20.4s, v24.4s, v12.s[1]\n"
173
"fmla v21.4s, v24.4s, v12.s[3]\n"
174
"fmla v22.4s, v24.4s, v13.s[1]\n"
175
"ldr q30, [%x[params], #0x80]\n"
176
"fmla v14.4s, v23.4s, v4.s[2]\n"
177
"fmla v15.4s, v23.4s, v5.s[0]\n"
178
"fmin v14.4s, v14.4s, v26.4s\n"
179
"fmla v16.4s, v23.4s, v5.s[2]\n"
180
"fmla v17.4s, v23.4s, v8.s[2]\n"
181
"fmax v14.4s, v14.4s, v27.4s\n"
182
"str q14, [x13, x14]\n"
183
"ldr q14, [%x[params], #0x60]\n"
184
"fmla v18.4s, v23.4s, v9.s[0]\n"
185
"fmla v19.4s, v23.4s, v9.s[2]\n"
186
"fmin v15.4s, v15.4s, v26.4s\n"
187
"fmla v20.4s, v23.4s, v12.s[2]\n"
188
"fmla v21.4s, v23.4s, v13.s[0]\n"
189
"fmin v16.4s, v16.4s, v26.4s\n"
190
"fmla v22.4s, v23.4s, v13.s[2]\n"
191
"ldr q29, [%x[params], #0x90]\n"
192
"fmin v17.4s, v17.4s, v26.4s\n"
193
"add %x[params], %x[params], #0xa0\n"
194
"fmin v18.4s, v18.4s, v26.4s\n"
195
"fmin v19.4s, v19.4s, v26.4s\n"
196
"fmin v20.4s, v20.4s, v26.4s\n"
197
"fmin v21.4s, v21.4s, v26.4s\n"
198
"fmin v22.4s, v22.4s, v26.4s\n"
199
"fmax v15.4s, v15.4s, v27.4s\n"
200
"str q15, [x12, x14]\n"
201
"fmax v16.4s, v16.4s, v27.4s\n"
202
"fmax v17.4s, v17.4s, v27.4s\n"
203
"str q16, [x11, x14]\n"
204
"fmax v18.4s, v18.4s, v27.4s\n"
205
"fmax v19.4s, v19.4s, v27.4s\n"
206
"str q17, [x10, x14]\n"
207
"fmax v20.4s, v20.4s, v27.4s\n"
208
"fmax v21.4s, v21.4s, v27.4s\n"
209
"str q18, [x9, x14]\n"
210
"fmax v22.4s, v22.4s, v27.4s\n"
211
"str q19, [x28, x14]\n"
212
"mov v15.16b, v14.16b\n"
213
"str q20, [x27, x14]\n"
214
"mov v16.16b, v14.16b\n"
215
"mov v17.16b, v14.16b\n"
216
"str q21, [x26, x14]\n"
217
"mov v18.16b, v14.16b\n"
218
"mov v19.16b, v14.16b\n"
219
"str q22, [x25, x14]\n"
220
"mov v20.16b, v14.16b\n"
221
"mov v21.16b, v14.16b\n"
222
"add x14, x14, #0x10\n"
223
"mov v22.16b, v14.16b\n"
224
"bgt 1b\n"
225
"2:"
// Output channel complete vector tail
226
"fmla v14.4s, v31.4s, v0.s[0]\n"
227
"fmla v15.4s, v31.4s, v0.s[2]\n"
228
"fmla v16.4s, v31.4s, v1.s[0]\n"
229
"fmla v17.4s, v31.4s, v4.s[0]\n"
230
"fmla v18.4s, v31.4s, v4.s[2]\n"
231
"fmla v19.4s, v31.4s, v5.s[0]\n"
232
"fmla v20.4s, v31.4s, v8.s[0]\n"
233
"fmla v21.4s, v31.4s, v8.s[2]\n"
234
"fmla v22.4s, v31.4s, v9.s[0]\n"
235
"ldr q25, [%x[params], #0x0]\n"
236
"fmla v14.4s, v30.4s, v0.s[1]\n"
237
"fmla v15.4s, v30.4s, v0.s[3]\n"
238
"fmla v16.4s, v30.4s, v1.s[1]\n"
239
"fmla v17.4s, v30.4s, v4.s[1]\n"
240
"fmla v18.4s, v30.4s, v4.s[3]\n"
241
"fmla v19.4s, v30.4s, v5.s[1]\n"
242
"fmla v20.4s, v30.4s, v8.s[1]\n"
243
"fmla v21.4s, v30.4s, v8.s[3]\n"
244
"fmla v22.4s, v30.4s, v9.s[1]\n"
245
"ldr q24, [%x[params], #0x10]\n"
246
"fmla v14.4s, v29.4s, v0.s[2]\n"
247
"fmla v15.4s, v29.4s, v1.s[0]\n"
248
"fmla v16.4s, v29.4s, v1.s[2]\n"
249
"fmla v17.4s, v29.4s, v4.s[2]\n"
250
"fmla v18.4s, v29.4s, v5.s[0]\n"
251
"fmla v19.4s, v29.4s, v5.s[2]\n"
252
"fmla v20.4s, v29.4s, v8.s[2]\n"
253
"fmla v21.4s, v29.4s, v9.s[0]\n"
254
"fmla v22.4s, v29.4s, v9.s[2]\n"
255
"ldr q23, [%x[params], #0x20]\n"
256
"fmla v14.4s, v25.4s, v2.s[0]\n"
257
"fmla v15.4s, v25.4s, v2.s[2]\n"
258
"fmla v16.4s, v25.4s, v3.s[0]\n"
259
"fmla v17.4s, v25.4s, v6.s[0]\n"
260
"fmla v18.4s, v25.4s, v6.s[2]\n"
261
"fmla v19.4s, v25.4s, v7.s[0]\n"
262
"fmla v20.4s, v25.4s, v10.s[0]\n"
263
"fmla v21.4s, v25.4s, v10.s[2]\n"
264
"fmla v22.4s, v25.4s, v11.s[0]\n"
265
"ldr q25, [%x[params], #0x30]\n"
266
"fmla v14.4s, v24.4s, v2.s[1]\n"
267
"fmla v15.4s, v24.4s, v2.s[3]\n"
268
"fmla v16.4s, v24.4s, v3.s[1]\n"
269
"fmla v17.4s, v24.4s, v6.s[1]\n"
270
"fmla v18.4s, v24.4s, v6.s[3]\n"
271
"fmla v19.4s, v24.4s, v7.s[1]\n"
272
"fmla v20.4s, v24.4s, v10.s[1]\n"
273
"fmla v21.4s, v24.4s, v10.s[3]\n"
274
"fmla v22.4s, v24.4s, v11.s[1]\n"
275
"ldr q24, [%x[params], #0x40]\n"
276
"fmla v14.4s, v23.4s, v2.s[2]\n"
277
"fmla v15.4s, v23.4s, v3.s[0]\n"
278
"fmla v16.4s, v23.4s, v3.s[2]\n"
279
"fmla v17.4s, v23.4s, v6.s[2]\n"
280
"fmla v18.4s, v23.4s, v7.s[0]\n"
281
"fmla v19.4s, v23.4s, v7.s[2]\n"
282
"fmla v20.4s, v23.4s, v10.s[2]\n"
283
"fmla v21.4s, v23.4s, v11.s[0]\n"
284
"fmla v22.4s, v23.4s, v11.s[2]\n"
285
"ldr q23, [%x[params], #0x50]\n"
286
"add %x[params], %x[params], #0x60\n"
287
"fmla v14.4s, v25.4s, v4.s[0]\n"
288
"fmla v15.4s, v25.4s, v4.s[2]\n"
289
"fmla v16.4s, v25.4s, v5.s[0]\n"
290
"fmla v17.4s, v25.4s, v8.s[0]\n"
291
"fmla v18.4s, v25.4s, v8.s[2]\n"
292
"fmla v19.4s, v25.4s, v9.s[0]\n"
293
"fmla v20.4s, v25.4s, v12.s[0]\n"
294
"fmla v21.4s, v25.4s, v12.s[2]\n"
295
"fmla v22.4s, v25.4s, v13.s[0]\n"
296
"fmla v14.4s, v24.4s, v4.s[1]\n"
297
"fmla v15.4s, v24.4s, v4.s[3]\n"
298
"fmla v16.4s, v24.4s, v5.s[1]\n"
299
"fmla v17.4s, v24.4s, v8.s[1]\n"
300
"fmla v18.4s, v24.4s, v8.s[3]\n"
301
"fmla v19.4s, v24.4s, v9.s[1]\n"
302
"fmla v20.4s, v24.4s, v12.s[1]\n"
303
"fmla v21.4s, v24.4s, v12.s[3]\n"
304
"fmla v22.4s, v24.4s, v13.s[1]\n"
305
"fmla v14.4s, v23.4s, v4.s[2]\n"
306
"fmla v15.4s, v23.4s, v5.s[0]\n"
307
"fmin v14.4s, v14.4s, v26.4s\n"
308
"fmla v16.4s, v23.4s, v5.s[2]\n"
309
"fmla v17.4s, v23.4s, v8.s[2]\n"
310
"fmin v15.4s, v15.4s, v26.4s\n"
311
"fmla v18.4s, v23.4s, v9.s[0]\n"
312
"fmla v19.4s, v23.4s, v9.s[2]\n"
313
"fmin v16.4s, v16.4s, v26.4s\n"
314
"fmla v20.4s, v23.4s, v12.s[2]\n"
315
"fmla v21.4s, v23.4s, v13.s[0]\n"
316
"fmin v17.4s, v17.4s, v26.4s\n"
317
"fmla v22.4s, v23.4s, v13.s[2]\n"
318
"fmin v18.4s, v18.4s, v26.4s\n"
319
"fmin v19.4s, v19.4s, v26.4s\n"
320
"fmin v20.4s, v20.4s, v26.4s\n"
321
"fmin v21.4s, v21.4s, v26.4s\n"
322
"fmin v22.4s, v22.4s, v26.4s\n"
323
"fmax v14.4s, v14.4s, v27.4s\n"
324
"fmax v15.4s, v15.4s, v27.4s\n"
325
"str q14, [x13, x14]\n"
326
"fmax v16.4s, v16.4s, v27.4s\n"
327
"fmax v17.4s, v17.4s, v27.4s\n"
328
"str q15, [x12, x14]\n"
329
"fmax v18.4s, v18.4s, v27.4s\n"
330
"fmax v19.4s, v19.4s, v27.4s\n"
331
"str q16, [x11, x14]\n"
332
"fmax v20.4s, v20.4s, v27.4s\n"
333
"fmax v21.4s, v21.4s, v27.4s\n"
334
"str q17, [x10, x14]\n"
335
"fmax v22.4s, v22.4s, v27.4s\n"
336
"str q18, [x9, x14]\n"
337
"str q19, [x28, x14]\n"
338
"str q20, [x27, x14]\n"
339
"str q21, [x26, x14]\n"
340
"str q22, [x25, x14]\n"
341
"add x14, x14, #0x10\n"
342
"3:"
// Output channel oddments
343
"tst %x[channel_multiplier], #0x3\n"
344
"beq 6f\n"
345
"ldr q14, [%x[params], #0x0]\n"
346
"ldr q25, [%x[params], #0x10]\n"
347
"mov v15.16b, v14.16b\n"
348
"mov v16.16b, v14.16b\n"
349
"ldr q24, [%x[params], #0x20]\n"
350
"ldr q23, [%x[params], #0x30]\n"
351
"mov v17.16b, v14.16b\n"
352
"mov v18.16b, v14.16b\n"
353
"mov v19.16b, v14.16b\n"
354
"mov v20.16b, v14.16b\n"
355
"fmla v15.4s, v25.4s, v0.s[2]\n"
356
"mov v21.16b, v14.16b\n"
357
"mov v22.16b, v14.16b\n"
358
"fmla v14.4s, v25.4s, v0.s[0]\n"
359
"fmla v16.4s, v25.4s, v1.s[0]\n"
360
"fmla v17.4s, v25.4s, v4.s[0]\n"
361
"fmla v18.4s, v25.4s, v4.s[2]\n"
362
"fmla v19.4s, v25.4s, v5.s[0]\n"
363
"fmla v20.4s, v25.4s, v8.s[0]\n"
364
"fmla v21.4s, v25.4s, v8.s[2]\n"
365
"fmla v22.4s, v25.4s, v9.s[0]\n"
366
"ldr q25, [%x[params], #0x40]\n"
367
"fmla v14.4s, v24.4s, v0.s[1]\n"
368
"fmla v15.4s, v24.4s, v0.s[3]\n"
369
"fmla v16.4s, v24.4s, v1.s[1]\n"
370
"fmla v17.4s, v24.4s, v4.s[1]\n"
371
"fmla v18.4s, v24.4s, v4.s[3]\n"
372
"fmla v19.4s, v24.4s, v5.s[1]\n"
373
"fmla v20.4s, v24.4s, v8.s[1]\n"
374
"fmla v21.4s, v24.4s, v8.s[3]\n"
375
"fmla v22.4s, v24.4s, v9.s[1]\n"
376
"ldr q24, [%x[params], #0x50]\n"
377
"fmla v14.4s, v23.4s, v0.s[2]\n"
378
"fmla v15.4s, v23.4s, v1.s[0]\n"
379
"fmla v16.4s, v23.4s, v1.s[2]\n"
380
"fmla v17.4s, v23.4s, v4.s[2]\n"
381
"fmla v18.4s, v23.4s, v5.s[0]\n"
382
"fmla v19.4s, v23.4s, v5.s[2]\n"
383
"fmla v20.4s, v23.4s, v8.s[2]\n"
384
"fmla v21.4s, v23.4s, v9.s[0]\n"
385
"fmla v22.4s, v23.4s, v9.s[2]\n"
386
"ldr q23, [%x[params], #0x60]\n"
387
"fmla v14.4s, v25.4s, v2.s[0]\n"
388
"fmla v15.4s, v25.4s, v2.s[2]\n"
389
"fmla v16.4s, v25.4s, v3.s[0]\n"
390
"fmla v17.4s, v25.4s, v6.s[0]\n"
391
"fmla v18.4s, v25.4s, v6.s[2]\n"
392
"fmla v19.4s, v25.4s, v7.s[0]\n"
393
"fmla v20.4s, v25.4s, v10.s[0]\n"
394
"fmla v21.4s, v25.4s, v10.s[2]\n"
395
"fmla v22.4s, v25.4s, v11.s[0]\n"
396
"ldr q25, [%x[params], #0x70]\n"
397
"fmla v14.4s, v24.4s, v2.s[1]\n"
398
"fmla v15.4s, v24.4s, v2.s[3]\n"
399
"fmla v16.4s, v24.4s, v3.s[1]\n"
400
"fmla v17.4s, v24.4s, v6.s[1]\n"
401
"fmla v18.4s, v24.4s, v6.s[3]\n"
402
"fmla v19.4s, v24.4s, v7.s[1]\n"
403
"fmla v20.4s, v24.4s, v10.s[1]\n"
404
"fmla v21.4s, v24.4s, v10.s[3]\n"
405
"fmla v22.4s, v24.4s, v11.s[1]\n"
406
"ldr q24, [%x[params], #0x80]\n"
407
"fmla v14.4s, v23.4s, v2.s[2]\n"
408
"fmla v15.4s, v23.4s, v3.s[0]\n"
409
"fmla v16.4s, v23.4s, v3.s[2]\n"
410
"fmla v17.4s, v23.4s, v6.s[2]\n"
411
"fmla v18.4s, v23.4s, v7.s[0]\n"
412
"fmla v19.4s, v23.4s, v7.s[2]\n"
413
"fmla v20.4s, v23.4s, v10.s[2]\n"
414
"fmla v21.4s, v23.4s, v11.s[0]\n"
415
"fmla v22.4s, v23.4s, v11.s[2]\n"
416
"ldr q23, [%x[params], #0x90]\n"
417
"add %x[params], %x[params], #0xa0\n"
418
"fmla v14.4s, v25.4s, v4.s[0]\n"
419
"fmla v15.4s, v25.4s, v4.s[2]\n"
420
"fmla v16.4s, v25.4s, v5.s[0]\n"
421
"fmla v17.4s, v25.4s, v8.s[0]\n"
422
"fmla v18.4s, v25.4s, v8.s[2]\n"
423
"fmla v19.4s, v25.4s, v9.s[0]\n"
424
"fmla v20.4s, v25.4s, v12.s[0]\n"
425
"fmla v21.4s, v25.4s, v12.s[2]\n"
426
"fmla v22.4s, v25.4s, v13.s[0]\n"
427
"fmla v14.4s, v24.4s, v4.s[1]\n"
428
"fmla v15.4s, v24.4s, v4.s[3]\n"
429
"fmla v16.4s, v24.4s, v5.s[1]\n"
430
"fmla v17.4s, v24.4s, v8.s[1]\n"
431
"fmla v18.4s, v24.4s, v8.s[3]\n"
432
"fmla v19.4s, v24.4s, v9.s[1]\n"
433
"fmla v20.4s, v24.4s, v12.s[1]\n"
434
"fmla v21.4s, v24.4s, v12.s[3]\n"
435
"fmla v22.4s, v24.4s, v13.s[1]\n"
436
"fmla v14.4s, v23.4s, v4.s[2]\n"
437
"fmla v15.4s, v23.4s, v5.s[0]\n"
438
"fmin v14.4s, v14.4s, v26.4s\n"
439
"fmla v16.4s, v23.4s, v5.s[2]\n"
440
"fmla v17.4s, v23.4s, v8.s[2]\n"
441
"fmin v15.4s, v15.4s, v26.4s\n"
442
"fmla v18.4s, v23.4s, v9.s[0]\n"
443
"fmla v19.4s, v23.4s, v9.s[2]\n"
444
"fmin v16.4s, v16.4s, v26.4s\n"
445
"fmla v20.4s, v23.4s, v12.s[2]\n"
446
"fmla v21.4s, v23.4s, v13.s[0]\n"
447
"fmin v17.4s, v17.4s, v26.4s\n"
448
"fmla v22.4s, v23.4s, v13.s[2]\n"
449
"fmin v18.4s, v18.4s, v26.4s\n"
450
"fmin v19.4s, v19.4s, v26.4s\n"
451
"fmin v20.4s, v20.4s, v26.4s\n"
452
"fmin v21.4s, v21.4s, v26.4s\n"
453
"fmin v22.4s, v22.4s, v26.4s\n"
454
"fmax v14.4s, v14.4s, v27.4s\n"
455
"fmax v15.4s, v15.4s, v27.4s\n"
456
"fmax v16.4s, v16.4s, v27.4s\n"
457
"fmax v17.4s, v17.4s, v27.4s\n"
458
"fmax v18.4s, v18.4s, v27.4s\n"
459
"fmax v19.4s, v19.4s, v27.4s\n"
460
"fmax v20.4s, v20.4s, v27.4s\n"
461
"fmax v21.4s, v21.4s, v27.4s\n"
462
"fmax v22.4s, v22.4s, v27.4s\n"
463
"tbz %x[channel_multiplier], #1, 4f\n"
464
"add x20, x13, x14\n"
465
"add x22, x12, x14\n"
466
"st1 { v14.d }[0], [x20]\n"
467
"add x21, x11, x14\n"
468
"add x20, x10, x14\n"
469
"st1 { v15.d }[0], [x22]\n"
470
"add x24, x9, x14\n"
471
"add x23, x28, x14\n"
472
"st1 { v16.d }[0], [x21]\n"
473
"add x22, x27, x14\n"
474
"add x21, x26, x14\n"
475
"st1 { v17.d }[0], [x20]\n"
476
"add x20, x25, x14\n"
477
"st1 { v18.d }[0], [x24]\n"
478
"add x14, x14, #0x8\n"
479
"st1 { v19.d }[0], [x23]\n"
480
"st1 { v20.d }[0], [x22]\n"
481
"st1 { v21.d }[0], [x21]\n"
482
"st1 { v22.d }[0], [x20]\n"
483
"tbz %x[channel_multiplier], #0, 5f\n"
484
"add x20, x13, x14\n"
485
"add x22, x12, x14\n"
486
"st1 { v14.s }[2], [x20]\n"
487
"add x21, x11, x14\n"
488
"add x20, x10, x14\n"
489
"st1 { v15.s }[2], [x22]\n"
490
"add x24, x9, x14\n"
491
"add x23, x28, x14\n"
492
"st1 { v16.s }[2], [x21]\n"
493
"add x22, x27, x14\n"
494
"add x21, x26, x14\n"
495
"st1 { v17.s }[2], [x20]\n"
496
"add x20, x25, x14\n"
497
"st1 { v18.s }[2], [x24]\n"
498
"st1 { v19.s }[2], [x23]\n"
499
"st1 { v20.s }[2], [x22]\n"
500
"st1 { v21.s }[2], [x21]\n"
501
"st1 { v22.s }[2], [x20]\n"
502
"b 5f\n"
503
"4:"
// Output channel oddments: Store: Bit 1: Unset
504
"add x20, x13, x14\n"
505
"add x22, x12, x14\n"
506
"st1 { v14.s }[0], [x20]\n"
507
"add x21, x11, x14\n"
508
"add x20, x10, x14\n"
509
"st1 { v15.s }[0], [x22]\n"
510
"add x24, x9, x14\n"
511
"add x23, x28, x14\n"
512
"st1 { v16.s }[0], [x21]\n"
513
"add x22, x27, x14\n"
514
"add x21, x26, x14\n"
515
"st1 { v17.s }[0], [x20]\n"
516
"add x20, x25, x14\n"
517
"st1 { v18.s }[0], [x24]\n"
518
"st1 { v19.s }[0], [x23]\n"
519
"st1 { v20.s }[0], [x22]\n"
520
"st1 { v21.s }[0], [x21]\n"
521
"st1 { v22.s }[0], [x20]\n"
522
"5:"
// Output channel oddments: Store: Bit 1: End
523
"6:"
// End
524
: [params]
"+&r"
(params)
525
: [channel_multiplier]
"r"
(n_output_channels), [clamps]
"r"
(minmax_vals), [inptrs]
"r"
(inptrs), [outptrs]
"r"
(outptrs)
526
:
"cc"
,
"memory"
,
"v0"
,
"v1"
,
"v2"
,
"v3"
,
"v4"
,
"v5"
,
"v6"
,
"v7"
,
"v8"
,
"v9"
,
"v10"
,
"v11"
,
"v12"
,
"v13"
,
"v14"
,
"v15"
,
"v16"
,
"v17"
,
"v18"
,
"v19"
,
"v20"
,
"v21"
,
"v22"
,
"v23"
,
"v24"
,
"v25"
,
"v26"
,
"v27"
,
"v29"
,
"v30"
,
"v31"
,
"x9"
,
"x10"
,
"x11"
,
"x12"
,
"x13"
,
"x14"
,
"x20"
,
"x21"
,
"x22"
,
"x23"
,
"x24"
,
"x25"
,
"x26"
,
"x27"
,
"x28"
527
);
528
}
529
530
}
// namespace depthwise
531
}
// namespace arm_conv
532
533
#endif // defined(__aarch64__)
arm_conv::depthwise::depthwise
template UniqueDepthwiseCommon< float > depthwise(const DepthwiseArgs &, const Nothing &)
activation_min
T activation_min
Definition:
working_space.hpp:145
activation_max
T activation_max
Definition:
working_space.hpp:145
arm_conv
Definition:
addressing.cpp:30
src
core
NEON
kernels
arm_conv
depthwise
kernels
a64_fp32_packed_to_nhwc_3x3_s2_with_multiplier_output3x3_mla_depthfirst
generic.cpp
Generated on Mon Apr 29 2024 10:53:54 for Compute Library by
1.8.17