Gitiles
Code Review Sign In
review.mlplatform.org / tosa / reference_model / 635bc99c8fd012e4e7b3dcc1dc72a9a0db6661fb / . / verif / conformance / tosa_main_profile_ops_info.json
blob: ce9b71fcffd8558be23f967c4ec0875515fde7fc [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 },
James Ward542dd3b2022-11-17 12:11:55 +0000[diff] [blame]171 "const": {
172 "group": "data_nodes",
173 "no_negative_tests": "true",
174 "generator_args": [
175 [
176 "--target-dtype",
177 "fp32",
178 "--target-dtype",
179 "fp16",
180 "--target-dtype",
181 "bf16",
182 "--fp-values-range",
183 "-2.0,2.0",
184 "--target-rank",
185 "1",
186 "--target-rank",
187 "2",
188 "--target-rank",
189 "3"
190 ],
191 [
192 "--target-dtype",
193 "fp32",
194 "--target-dtype",
195 "fp16",
196 "--target-dtype",
197 "bf16",
198 "--fp-values-range",
199 "-2.0,2.0",
200 "--tensor-dim-range",
201 "1,16",
202 "--target-rank",
203 "4",
204 "--target-rank",
205 "5"
206 ],
207 [
208 "--target-dtype",
209 "fp32",
210 "--fp-values-range",
211 "-2.0,2.0",
212 "--target-shape",
213 "1,65540,1,1,1",
214 "--target-shape",
215 "1,2,1,65529"
216 ]
217 ],
218 "params": {},
219 "permutes": [
220 "shape",
221 "type"
222 ],
223 "profile": [
224 "tosa-mi"
225 ]
226 },
Jeremy Johnsone4b08ff2022-09-15 10:38:17 +0100[diff] [blame]227 "conv2d": {
228 "group": "tensor",
229 "generator_args": [
230 [
231 "--target-dtype",
232 "fp32",
233 "--target-dtype",
234 "fp16",
235 "--target-dtype",
236 "bf16",
237 "--fp-values-range",
238 "-2.0,2.0",
239 "--target-shape",
240 "1,49,42,28",
241 "--target-shape",
242 "1,11,44,13",
243 "--allow-pooling-and-conv-oversizes"
244 ],
245 [
246 "--target-dtype",
247 "fp32",
248 "--fp-values-range",
249 "-2.0,2.0",
250 "--target-shape",
251 "1,65535,4,1",
252 "--target-shape",
253 "1,5,65536,1",
254 "--max-conv-dilation",
255 "1",
256 "--allow-pooling-and-conv-oversizes"
257 ]
258 ],
259 "params": {
260 "shape": [],
261 "type": [],
262 "kernel": [],
263 "stride": [],
264 "pad": [],
265 "dilation": []
266 },
267 "permutes": [
268 "kernel",
269 "shape",
270 "type",
271 "pad"
272 ],
273 "profile": [
274 "tosa-mi"
275 ]
James Ward52ac9d62022-11-16 12:53:49 +0000[diff] [blame]276 },
James Ward990cc4b2022-11-16 15:36:37 +0000[diff] [blame]277 "gather": {
278 "group": "scatter_gather",
279 "generator_args": [
280 [
281 "--target-dtype",
282 "fp32",
283 "--target-dtype",
284 "fp16",
285 "--target-dtype",
286 "bf16",
287 "--fp-values-range",
288 "-2.0,2.0",
289 "--tensor-dim-range",
290 "5,72"
291 ],
292 [
293 "--target-dtype",
294 "fp32",
295 "--fp-values-range",
296 "-2.0,2.0",
297 "--target-shape",
298 "2,65536,1",
299 "--target-shape",
300 "1,2,65535",
301 "--target-shape",
302 "5000,1,1"
303 ]
304 ],
305 "params": {},
306 "permutes": [
307 "shape",
308 "type"
309 ],
310 "profile": [
311 "tosa-mi"
312 ]
313 },
James Ward542dd3b2022-11-17 12:11:55 +0000[diff] [blame]314 "identity": {
315 "group": "data_nodes",
316 "no_negative_tests": "true",
317 "generator_args": [
318 [
319 "--target-dtype",
320 "fp32",
321 "--target-dtype",
322 "fp16",
323 "--target-dtype",
324 "bf16",
325 "--fp-values-range",
326 "-2.0,2.0",
327 "--tensor-dim-range",
328 "1,67",
329 "--target-rank",
330 "1",
331 "--target-rank",
332 "2",
333 "--target-rank",
334 "3"
335 ],
336 [
337 "--target-dtype",
338 "fp32",
339 "--target-dtype",
340 "fp16",
341 "--target-dtype",
342 "bf16",
343 "--fp-values-range",
344 "-2.0,2.0",
345 "--tensor-dim-range",
346 "1,19",
347 "--target-rank",
348 "4",
349 "--target-rank",
350 "5"
351 ],
352 [
353 "--target-dtype",
354 "fp32",
355 "--fp-values-range",
356 "-2.0,2.0",
357 "--target-shape",
358 "1,1,65539,1,1",
359 "--target-shape",
360 "1,1,2,65529"
361 ]
362 ],
363 "params": {},
364 "permutes": [
365 "shape",
366 "type"
367 ],
368 "profile": [
369 "tosa-mi"
370 ]
371 },
James Ward52ac9d62022-11-16 12:53:49 +0000[diff] [blame]372 "max_pool2d": {
373 "group": "tensor",
374 "generator_args": [
375 [
376 "--target-dtype",
377 "fp32",
378 "--target-dtype",
379 "fp16",
380 "--target-dtype",
381 "bf16",
382 "--fp-values-range",
383 "-2.0,2.0",
384 "--target-shape",
385 "1,47,37,25",
386 "--target-shape",
387 "1,15,13,38",
388 "--allow-pooling-and-conv-oversizes"
389 ],
390 [
391 "--target-dtype",
392 "fp32",
393 "--fp-values-range",
394 "-2.0,2.0",
395 "--target-shape",
396 "1,65534,4,1",
397 "--target-shape",
398 "1,6,65536,1",
399 "--max-conv-dilation",
400 "1",
401 "--allow-pooling-and-conv-oversizes"
402 ]
403 ],
404 "params": {
405 "shape": [],
406 "type": [
407 "f32",
408 "f16",
409 "bf16"
410 ],
411 "stride": [],
412 "pad": []
413 },
414 "permutes": [
415 "shape",
416 "type",
417 "pad"
418 ],
419 "sparsity": {
420 "pad": 11
421 },
422 "profile": [
423 "tosa-mi"
424 ]
James Ward990cc4b2022-11-16 15:36:37 +0000[diff] [blame]425 },
James Wardf0890992022-11-17 11:15:14 +0000[diff] [blame]426 "pad": {
427 "group": "data_layout",
428 "generator_args": [
429 [
430 "--target-dtype",
431 "fp32",
432 "--target-dtype",
433 "fp16",
434 "--target-dtype",
435 "bf16",
436 "--fp-values-range",
437 "-2.0,2.0",
438 "--tensor-dim-range",
439 "1,65",
440 "--target-rank",
441 "1",
442 "--target-rank",
443 "2",
444 "--target-rank",
445 "3"
446 ],
447 [
448 "--target-dtype",
449 "fp16",
450 "--fp-values-range",
451 "-2.0,2.0",
452 "--tensor-dim-range",
453 "1,17",
454 "--target-rank",
455 "4"
456 ],
457 [
458 "--target-dtype",
459 "bf16",
460 "--fp-values-range",
461 "-2.0,2.0",
462 "--tensor-dim-range",
463 "1,16",
464 "--target-rank",
465 "5"
466 ],
467 [
468 "--target-dtype",
469 "fp32",
470 "--fp-values-range",
471 "-2.0,2.0",
472 "--target-shape",
473 "1,1,65539,1"
474 ]
475 ],
476 "params": {},
477 "permutes": [
478 "shape",
479 "type"
480 ],
481 "sparsity": {
482 "pad": 21
483 },
484 "preselected": [
485 {
486 "shape": "50",
487 "type": "bf16",
488 "pad": "pad11"
489 },
490 {
491 "shape": "63x46",
492 "type": "bf16",
493 "pad": "pad1010"
494 },
495 {
496 "shape": "6",
497 "type": "f16",
498 "pad": "pad01"
499 }
500 ],
501 "profile": [
502 "tosa-mi"
503 ]
504 },
505 "reshape": {
506 "group": "data_layout",
507 "generator_args": [
508 [
509 "--target-dtype",
510 "fp32",
511 "--target-dtype",
512 "fp16",
513 "--target-dtype",
514 "bf16",
515 "--fp-values-range",
516 "-2.0,2.0",
517 "--num-rand-permutations",
518 "2"
519 ],
520 [
521 "--target-dtype",
522 "fp32",
523 "--fp-values-range",
524 "-2.0,2.0",
525 "--target-shape",
526 "1,1,65536,1",
527 "--target-shape",
528 "1,65535,1,2",
529 "--num-rand-permutations",
530 "2"
531 ]
532 ],
533 "params": {},
534 "permutes": [
535 "shape",
536 "type",
537 "rank",
538 "perm"
539 ],
540 "profile": [
541 "tosa-mi"
542 ]
543 },
544 "reverse": {
545 "group": "data_layout",
546 "generator_args": [
547 [
548 "--target-dtype",
549 "fp32",
550 "--target-dtype",
551 "fp16",
552 "--target-dtype",
553 "bf16",
554 "--fp-values-range",
555 "-2.0,2.0",
556 "--tensor-dim-range",
557 "1,48"
558 ],
559 [
560 "--target-dtype",
561 "fp32",
562 "--fp-values-range",
563 "-2.0,2.0",
564 "--target-shape",
565 "1,1,65537",
566 "--target-shape",
567 "1,65535,1,2"
568 ]
569 ],
570 "params": {},
571 "permutes": [
572 "shape",
573 "type"
574 ],
575 "profile": [
576 "tosa-mi"
577 ]
578 },
James Ward990cc4b2022-11-16 15:36:37 +0000[diff] [blame]579 "scatter": {
580 "group": "scatter_gather",
581 "generator_args": [
582 [
583 "--target-dtype",
584 "fp32",
585 "--target-dtype",
586 "fp16",
587 "--target-dtype",
588 "bf16",
589 "--fp-values-range",
590 "-2.0,2.0",
591 "--tensor-dim-range",
592 "5,56"
593 ],
594 [
595 "--target-dtype",
596 "fp32",
597 "--fp-values-range",
598 "-2.0,2.0",
599 "--target-shape",
600 "1,65541,1",
601 "--target-shape",
602 "1,1,65531",
603 "--target-shape",
604 "2989,6,1"
605 ]
606 ],
607 "params": {},
608 "permutes": [
609 "shape",
610 "type"
611 ],
612 "profile": [
613 "tosa-mi"
614 ]
James Wardf0890992022-11-17 11:15:14 +0000[diff] [blame]615 },
616 "tile": {
617 "group": "data_layout",
618 "generator_args": [
619 [
620 "--target-dtype",
621 "fp32",
622 "--target-dtype",
623 "fp16",
624 "--target-dtype",
625 "bf16",
626 "--fp-values-range",
627 "-2.0,2.0",
628 "--tensor-dim-range",
629 "4,32"
630 ],
631 [
632 "--target-dtype",
633 "fp32",
634 "--fp-values-range",
635 "-2.0,2.0",
636 "--target-shape",
637 "65535",
638 "--target-shape",
639 "1,65543,2,1"
640 ]
641 ],
642 "params": {},
643 "permutes": [
644 "shape",
645 "type"
646 ],
647 "profile": [
648 "tosa-mi"
649 ]
650 },
651 "transpose": {
652 "group": "data_layout",
653 "generator_args": [
654 [
655 "--filter",
656 "^transpose$",
657 "--target-dtype",
658 "fp32",
659 "--target-dtype",
660 "fp16",
661 "--fp-values-range",
662 "-2.0,2.0",
663 "--target-rank",
664 "1",
665 "--target-rank",
666 "2",
667 "--num-rand-permutations",
668 "2"
669 ],
670 [
671 "--filter",
672 "^transpose$",
673 "--target-dtype",
674 "fp32",
675 "--target-dtype",
676 "fp16",
677 "--target-dtype",
678 "bf16",
679 "--fp-values-range",
680 "-2.0,2.0",
681 "--target-rank",
682 "3",
683 "--target-rank",
684 "4",
685 "--num-rand-permutations",
686 "2"
687 ],
688 [
689 "--filter",
690 "^transpose$",
691 "--target-dtype",
692 "fp32",
693 "--fp-values-range",
694 "-2.0,2.0",
695 "--target-shape",
696 "1,1,65539,2",
697 "--target-shape",
698 "1,1,1,65535",
699 "--num-rand-permutations",
700 "1"
701 ]
702 ],
703 "params": {},
704 "permutes": [
705 "shape",
706 "type",
707 "perm"
708 ],
709 "profile": [
710 "tosa-mi"
711 ]
James Wardb9c3a632022-11-22 13:14:50 +0000[diff] [blame]712 },
713 "maximum": {
714 "group": "ew_binary",
715 "generator_args": [
716 [
717 "--target-dtype",
718 "fp32",
719 "--target-dtype",
720 "fp16",
721 "--target-dtype",
722 "bf16",
723 "--fp-values-range",
724 "-2.0,2.0",
725 "--tensor-dim-range",
726 "16,64",
727 "--target-rank",
728 "1",
729 "--target-rank",
730 "2",
731 "--target-rank",
732 "3"
733 ],
734 [
735 "--target-dtype",
736 "fp32",
737 "--target-dtype",
738 "fp16",
739 "--target-dtype",
740 "bf16",
741 "--fp-values-range",
742 "-2.0,2.0",
743 "--tensor-dim-range",
744 "1,16",
745 "--target-rank",
746 "4",
747 "--target-rank",
748 "5"
749 ],
750 [
751 "--target-dtype",
752 "fp32",
753 "--fp-values-range",
754 "-2.0,2.0",
755 "--target-shape",
756 "65531,1,2,1",
757 "--target-shape",
758 "1,1,65541,1"
759 ]
760 ],
761 "params": {},
762 "permutes": [
763 "shape",
764 "type"
765 ],
766 "profile": [
767 "tosa-mi"
768 ]
769 },
770 "minimum": {
771 "group": "ew_binary",
772 "generator_args": [
773 [
774 "--target-dtype",
775 "fp32",
776 "--target-dtype",
777 "fp16",
778 "--target-dtype",
779 "bf16",
780 "--fp-values-range",
781 "-2.0,2.0",
782 "--tensor-dim-range",
783 "16,64",
784 "--target-rank",
785 "1",
786 "--target-rank",
787 "2",
788 "--target-rank",
789 "3"
790 ],
791 [
792 "--target-dtype",
793 "fp32",
794 "--target-dtype",
795 "fp16",
796 "--target-dtype",
797 "bf16",
798 "--fp-values-range",
799 "-2.0,2.0",
800 "--tensor-dim-range",
801 "1,16",
802 "--target-rank",
803 "4",
804 "--target-rank",
805 "5"
806 ],
807 [
808 "--target-dtype",
809 "fp32",
810 "--fp-values-range",
811 "-2.0,2.0",
812 "--target-shape",
813 "1,65531,2,1",
814 "--target-shape",
815 "1,1,1,65538"
816 ]
817 ],
818 "params": {},
819 "permutes": [
820 "shape",
821 "type"
822 ],
823 "profile": [
824 "tosa-mi"
825 ]
James Ward635bc992022-11-23 11:55:32 +0000[diff] [blame^]826 },
827 "reduce_max": {
828 "group": "reduction",
829 "generator_args": [
830 [
831 "--target-dtype",
832 "fp32",
833 "--target-dtype",
834 "fp16",
835 "--target-dtype",
836 "bf16",
837 "--fp-values-range",
838 "-2.0,2.0",
839 "--tensor-dim-range",
840 "1,32"
841 ],
842 [
843 "--target-dtype",
844 "fp32",
845 "--fp-values-range",
846 "-2.0,2.0",
847 "--target-shape",
848 "1,1,1,65531",
849 "--target-shape",
850 "1,1,65539,2"
851 ]
852 ],
853 "params": {},
854 "permutes": [
855 "shape",
856 "type",
857 "axis"
858 ],
859 "profile": [
860 "tosa-mi"
861 ]
862 },
863 "reduce_min": {
864 "group": "reduction",
865 "generator_args": [
866 [
867 "--target-dtype",
868 "fp32",
869 "--target-dtype",
870 "fp16",
871 "--target-dtype",
872 "bf16",
873 "--fp-values-range",
874 "-2.0,2.0",
875 "--tensor-dim-range",
876 "1,32"
877 ],
878 [
879 "--target-dtype",
880 "fp32",
881 "--fp-values-range",
882 "-2.0,2.0",
883 "--target-shape",
884 "1,1,65531,1",
885 "--target-shape",
886 "1,65539,2,1"
887 ]
888 ],
889 "params": {},
890 "permutes": [
891 "shape",
892 "type",
893 "axis"
894 ],
895 "profile": [
896 "tosa-mi"
897 ]
Jeremy Johnsone4b08ff2022-09-15 10:38:17 +0100[diff] [blame]898 }
899}