Gitiles
Code Review Sign In
review.mlplatform.org / tosa / reference_model / f089099f2d87c7801f72ce1801daace1db83d154 / . / verif / conformance / tosa_main_profile_ops_info.json
blob: 9d2ae09810a47d00191ec41ba3531c655f43a51a [file] [log] [blame]
Jeremy Johnsone4b08ff2022-09-15 10:38:17 +0100[diff] [blame]1{
James Ward52ac9d62022-11-16 12:53:49 +0000[diff] [blame]2 "argmax": {
3 "group": "tensor",
4 "generator_args": [
5 [
6 "--target-dtype",
7 "fp32",
8 "--target-dtype",
9 "fp16",
10 "--target-dtype",
11 "bf16",
12 "--fp-values-range",
13 "-2.0,2.0",
14 "--tensor-dim-range",
15 "32,64",
16 "--target-rank",
17 "1",
18 "--target-rank",
19 "2",
20 "--target-rank",
21 "3",
22 "--num-rand-permutations",
23 "2"
24 ],
25 [
26 "--target-dtype",
27 "fp32",
28 "--target-dtype",
29 "fp16",
30 "--target-dtype",
31 "bf16",
32 "--fp-values-range",
33 "-2.0,2.0",
34 "--tensor-dim-range",
35 "1,32",
36 "--target-rank",
37 "4",
38 "--num-rand-permutations",
39 "2"
40 ],
41 [
42 "--target-dtype",
43 "fp32",
44 "--fp-values-range",
45 "-2.0,2.0",
46 "--target-shape",
47 "1,3,65535,1",
48 "--target-shape",
49 "1,65537,1,2",
50 "--num-rand-permutations",
51 "2"
52 ]
53 ],
54 "params": {
55 },
56 "permutes": [
57 "shape",
58 "type"
59 ],
60 "profile": [
61 "tosa-mi"
62 ]
63 },
Jeremy Johnsone4b08ff2022-09-15 10:38:17 +0100[diff] [blame]64 "avg_pool2d": {
65 "group": "tensor",
66 "generator_args": [
67 [
68 "--target-dtype",
69 "fp32",
70 "--target-dtype",
71 "fp16",
72 "--target-dtype",
73 "bf16",
74 "--fp-values-range",
75 "-2.0,2.0",
76 "--target-shape",
77 "1,49,42,28",
78 "--target-shape",
79 "3,11,44,3",
80 "--allow-pooling-and-conv-oversizes"
81 ],
82 [
83 "--target-dtype",
84 "fp32",
85 "--fp-values-range",
86 "-2.0,2.0",
87 "--target-shape",
88 "1,65535,5,1",
89 "--target-shape",
90 "1,3,65537,1",
91 "--allow-pooling-and-conv-oversizes"
92 ]
93 ],
94 "params": {
95 },
96 "permutes": [
97 "shape",
98 "type",
99 "accum_type"
100 ],
101 "sparsity": {
102 "pad": 17
103 },
104 "profile": [
105 "tosa-mi"
106 ]
107 },
James Wardf0890992022-11-17 11:15:14 +0000[diff] [blame^]108 "concat": {
109 "group": "data_layout",
110 "generator_args": [
111 [
112 "--target-dtype",
113 "fp32",
114 "--target-dtype",
115 "fp16",
116 "--target-dtype",
117 "bf16",
118 "--fp-values-range",
119 "-2.0,2.0",
120 "--tensor-dim-range",
121 "16,64",
122 "--target-rank",
123 "1",
124 "--target-rank",
125 "2",
126 "--target-rank",
127 "3",
128 "--num-const-inputs-concat",
129 "3"
130 ],
131 [
132 "--target-dtype",
133 "fp32",
134 "--target-dtype",
135 "fp16",
136 "--target-dtype",
137 "bf16",
138 "--fp-values-range",
139 "-2.0,2.0",
140 "--tensor-dim-range",
141 "1,16",
142 "--target-rank",
143 "4",
144 "--target-rank",
145 "5",
146 "--num-const-inputs-concat",
147 "1"
148 ],
149 [
150 "--target-dtype",
151 "fp32",
152 "--fp-values-range",
153 "-2.0,2.0",
154 "--target-shape",
155 "1,1,65534,1,1",
156 "--target-shape",
157 "1,65537,1,2",
158 "--num-const-inputs-concat",
159 "2"
160 ]
161 ],
162 "params": {},
163 "permutes": [
164 "shape",
165 "type"
166 ],
167 "profile": [
168 "tosa-mi"
169 ]
170 },
Jeremy Johnsone4b08ff2022-09-15 10:38:17 +0100[diff] [blame]171 "conv2d": {
172 "group": "tensor",
173 "generator_args": [
174 [
175 "--target-dtype",
176 "fp32",
177 "--target-dtype",
178 "fp16",
179 "--target-dtype",
180 "bf16",
181 "--fp-values-range",
182 "-2.0,2.0",
183 "--target-shape",
184 "1,49,42,28",
185 "--target-shape",
186 "1,11,44,13",
187 "--allow-pooling-and-conv-oversizes"
188 ],
189 [
190 "--target-dtype",
191 "fp32",
192 "--fp-values-range",
193 "-2.0,2.0",
194 "--target-shape",
195 "1,65535,4,1",
196 "--target-shape",
197 "1,5,65536,1",
198 "--max-conv-dilation",
199 "1",
200 "--allow-pooling-and-conv-oversizes"
201 ]
202 ],
203 "params": {
204 "shape": [],
205 "type": [],
206 "kernel": [],
207 "stride": [],
208 "pad": [],
209 "dilation": []
210 },
211 "permutes": [
212 "kernel",
213 "shape",
214 "type",
215 "pad"
216 ],
217 "profile": [
218 "tosa-mi"
219 ]
James Ward52ac9d62022-11-16 12:53:49 +0000[diff] [blame]220 },
James Ward990cc4b2022-11-16 15:36:37 +0000[diff] [blame]221 "gather": {
222 "group": "scatter_gather",
223 "generator_args": [
224 [
225 "--target-dtype",
226 "fp32",
227 "--target-dtype",
228 "fp16",
229 "--target-dtype",
230 "bf16",
231 "--fp-values-range",
232 "-2.0,2.0",
233 "--tensor-dim-range",
234 "5,72"
235 ],
236 [
237 "--target-dtype",
238 "fp32",
239 "--fp-values-range",
240 "-2.0,2.0",
241 "--target-shape",
242 "2,65536,1",
243 "--target-shape",
244 "1,2,65535",
245 "--target-shape",
246 "5000,1,1"
247 ]
248 ],
249 "params": {},
250 "permutes": [
251 "shape",
252 "type"
253 ],
254 "profile": [
255 "tosa-mi"
256 ]
257 },
James Ward52ac9d62022-11-16 12:53:49 +0000[diff] [blame]258 "max_pool2d": {
259 "group": "tensor",
260 "generator_args": [
261 [
262 "--target-dtype",
263 "fp32",
264 "--target-dtype",
265 "fp16",
266 "--target-dtype",
267 "bf16",
268 "--fp-values-range",
269 "-2.0,2.0",
270 "--target-shape",
271 "1,47,37,25",
272 "--target-shape",
273 "1,15,13,38",
274 "--allow-pooling-and-conv-oversizes"
275 ],
276 [
277 "--target-dtype",
278 "fp32",
279 "--fp-values-range",
280 "-2.0,2.0",
281 "--target-shape",
282 "1,65534,4,1",
283 "--target-shape",
284 "1,6,65536,1",
285 "--max-conv-dilation",
286 "1",
287 "--allow-pooling-and-conv-oversizes"
288 ]
289 ],
290 "params": {
291 "shape": [],
292 "type": [
293 "f32",
294 "f16",
295 "bf16"
296 ],
297 "stride": [],
298 "pad": []
299 },
300 "permutes": [
301 "shape",
302 "type",
303 "pad"
304 ],
305 "sparsity": {
306 "pad": 11
307 },
308 "profile": [
309 "tosa-mi"
310 ]
James Ward990cc4b2022-11-16 15:36:37 +0000[diff] [blame]311 },
James Wardf0890992022-11-17 11:15:14 +0000[diff] [blame^]312 "pad": {
313 "group": "data_layout",
314 "generator_args": [
315 [
316 "--target-dtype",
317 "fp32",
318 "--target-dtype",
319 "fp16",
320 "--target-dtype",
321 "bf16",
322 "--fp-values-range",
323 "-2.0,2.0",
324 "--tensor-dim-range",
325 "1,65",
326 "--target-rank",
327 "1",
328 "--target-rank",
329 "2",
330 "--target-rank",
331 "3"
332 ],
333 [
334 "--target-dtype",
335 "fp16",
336 "--fp-values-range",
337 "-2.0,2.0",
338 "--tensor-dim-range",
339 "1,17",
340 "--target-rank",
341 "4"
342 ],
343 [
344 "--target-dtype",
345 "bf16",
346 "--fp-values-range",
347 "-2.0,2.0",
348 "--tensor-dim-range",
349 "1,16",
350 "--target-rank",
351 "5"
352 ],
353 [
354 "--target-dtype",
355 "fp32",
356 "--fp-values-range",
357 "-2.0,2.0",
358 "--target-shape",
359 "1,1,65539,1"
360 ]
361 ],
362 "params": {},
363 "permutes": [
364 "shape",
365 "type"
366 ],
367 "sparsity": {
368 "pad": 21
369 },
370 "preselected": [
371 {
372 "shape": "50",
373 "type": "bf16",
374 "pad": "pad11"
375 },
376 {
377 "shape": "63x46",
378 "type": "bf16",
379 "pad": "pad1010"
380 },
381 {
382 "shape": "6",
383 "type": "f16",
384 "pad": "pad01"
385 }
386 ],
387 "profile": [
388 "tosa-mi"
389 ]
390 },
391 "reshape": {
392 "group": "data_layout",
393 "generator_args": [
394 [
395 "--target-dtype",
396 "fp32",
397 "--target-dtype",
398 "fp16",
399 "--target-dtype",
400 "bf16",
401 "--fp-values-range",
402 "-2.0,2.0",
403 "--num-rand-permutations",
404 "2"
405 ],
406 [
407 "--target-dtype",
408 "fp32",
409 "--fp-values-range",
410 "-2.0,2.0",
411 "--target-shape",
412 "1,1,65536,1",
413 "--target-shape",
414 "1,65535,1,2",
415 "--num-rand-permutations",
416 "2"
417 ]
418 ],
419 "params": {},
420 "permutes": [
421 "shape",
422 "type",
423 "rank",
424 "perm"
425 ],
426 "profile": [
427 "tosa-mi"
428 ]
429 },
430 "reverse": {
431 "group": "data_layout",
432 "generator_args": [
433 [
434 "--target-dtype",
435 "fp32",
436 "--target-dtype",
437 "fp16",
438 "--target-dtype",
439 "bf16",
440 "--fp-values-range",
441 "-2.0,2.0",
442 "--tensor-dim-range",
443 "1,48"
444 ],
445 [
446 "--target-dtype",
447 "fp32",
448 "--fp-values-range",
449 "-2.0,2.0",
450 "--target-shape",
451 "1,1,65537",
452 "--target-shape",
453 "1,65535,1,2"
454 ]
455 ],
456 "params": {},
457 "permutes": [
458 "shape",
459 "type"
460 ],
461 "profile": [
462 "tosa-mi"
463 ]
464 },
James Ward990cc4b2022-11-16 15:36:37 +0000[diff] [blame]465 "scatter": {
466 "group": "scatter_gather",
467 "generator_args": [
468 [
469 "--target-dtype",
470 "fp32",
471 "--target-dtype",
472 "fp16",
473 "--target-dtype",
474 "bf16",
475 "--fp-values-range",
476 "-2.0,2.0",
477 "--tensor-dim-range",
478 "5,56"
479 ],
480 [
481 "--target-dtype",
482 "fp32",
483 "--fp-values-range",
484 "-2.0,2.0",
485 "--target-shape",
486 "1,65541,1",
487 "--target-shape",
488 "1,1,65531",
489 "--target-shape",
490 "2989,6,1"
491 ]
492 ],
493 "params": {},
494 "permutes": [
495 "shape",
496 "type"
497 ],
498 "profile": [
499 "tosa-mi"
500 ]
James Wardf0890992022-11-17 11:15:14 +0000[diff] [blame^]501 },
502 "tile": {
503 "group": "data_layout",
504 "generator_args": [
505 [
506 "--target-dtype",
507 "fp32",
508 "--target-dtype",
509 "fp16",
510 "--target-dtype",
511 "bf16",
512 "--fp-values-range",
513 "-2.0,2.0",
514 "--tensor-dim-range",
515 "4,32"
516 ],
517 [
518 "--target-dtype",
519 "fp32",
520 "--fp-values-range",
521 "-2.0,2.0",
522 "--target-shape",
523 "65535",
524 "--target-shape",
525 "1,65543,2,1"
526 ]
527 ],
528 "params": {},
529 "permutes": [
530 "shape",
531 "type"
532 ],
533 "profile": [
534 "tosa-mi"
535 ]
536 },
537 "transpose": {
538 "group": "data_layout",
539 "generator_args": [
540 [
541 "--filter",
542 "^transpose$",
543 "--target-dtype",
544 "fp32",
545 "--target-dtype",
546 "fp16",
547 "--fp-values-range",
548 "-2.0,2.0",
549 "--target-rank",
550 "1",
551 "--target-rank",
552 "2",
553 "--num-rand-permutations",
554 "2"
555 ],
556 [
557 "--filter",
558 "^transpose$",
559 "--target-dtype",
560 "fp32",
561 "--target-dtype",
562 "fp16",
563 "--target-dtype",
564 "bf16",
565 "--fp-values-range",
566 "-2.0,2.0",
567 "--target-rank",
568 "3",
569 "--target-rank",
570 "4",
571 "--num-rand-permutations",
572 "2"
573 ],
574 [
575 "--filter",
576 "^transpose$",
577 "--target-dtype",
578 "fp32",
579 "--fp-values-range",
580 "-2.0,2.0",
581 "--target-shape",
582 "1,1,65539,2",
583 "--target-shape",
584 "1,1,1,65535",
585 "--num-rand-permutations",
586 "1"
587 ]
588 ],
589 "params": {},
590 "permutes": [
591 "shape",
592 "type",
593 "perm"
594 ],
595 "profile": [
596 "tosa-mi"
597 ]
Jeremy Johnsone4b08ff2022-09-15 10:38:17 +0100[diff] [blame]598 }
599}