Gitiles
Code Review Sign In
review.mlplatform.org / tosa / reference_model / 35396f233cefaa515e216fb58bc0f6d410eb142f / . / verif / conformance / tosa_main_profile_ops_info.json
blob: 47668db0564d0db2a3ea9318447e996a2cb8ccfa [file] [log] [blame]
James Ward71616fe2022-11-23 11:00:47 +0000[diff] [blame]1
Jeremy Johnsone4b08ff2022-09-15 10:38:17 +0100[diff] [blame]2{
James Ward71616fe2022-11-23 11:00:47 +0000[diff] [blame]3 "abs": {
4 "group": "ew_unary",
5 "generator_args": [
6 [
7 "--target-dtype",
8 "fp32",
9 "--target-dtype",
10 "fp16",
11 "--target-dtype",
12 "bf16",
13 "--fp-values-range",
14 "-2.0,2.0",
15 "--tensor-dim-range",
16 "16,64",
17 "--target-rank",
18 "1",
19 "--target-rank",
20 "2",
21 "--target-rank",
22 "3"
23 ],
24 [
25 "--target-dtype",
26 "fp32",
27 "--target-dtype",
28 "fp16",
29 "--target-dtype",
30 "bf16",
31 "--fp-values-range",
32 "-2.0,2.0",
33 "--tensor-dim-range",
34 "1,16",
35 "--target-rank",
36 "4",
37 "--target-rank",
38 "5"
39 ],
40 [
41 "--target-dtype",
42 "bf16",
43 "--fp-values-range",
44 "-2.0,2.0",
45 "--target-shape",
46 "1,65531,2,1",
47 "--target-shape",
48 "65538,1,1,1"
49 ]
50 ],
51 "params": {},
52 "permutes": [
53 "shape",
54 "type"
55 ],
56 "profile": [
57 "tosa-mi"
58 ]
59 },
Jeremy Johnsondd8d9c22022-12-12 14:18:10 +0000[diff] [blame]60 "add": {
61 "group": "ew_binary",
62 "generator_args": [
63 [
64 "--target-dtype",
65 "fp32",
66 "--target-dtype",
67 "fp16",
68 "--target-dtype",
69 "bf16",
70 "--fp-values-range",
71 "-2.0,2.0",
72 "--tensor-dim-range",
73 "16,64",
74 "--target-rank",
75 "1",
76 "--target-rank",
77 "2",
78 "--target-rank",
79 "3"
80 ],
81 [
82 "--target-dtype",
83 "fp32",
84 "--fp-values-range",
85 "-2.0,2.0",
86 "--tensor-dim-range",
87 "1,16",
88 "--target-rank",
89 "4",
90 "--target-rank",
91 "5"
92 ],
93 [
94 "--target-dtype",
95 "fp16",
96 "--fp-values-range",
97 "-2.0,2.0",
98 "--target-shape",
99 "2,1,1,65532",
100 "--target-shape",
101 "65542,2,1,1,1"
102 ]
103 ],
104 "params": {},
105 "permutes": [
106 "shape",
107 "type"
108 ],
109 "profile": [
110 "tosa-mi"
111 ]
112 },
James Ward52ac9d62022-11-16 12:53:49 +0000[diff] [blame]113 "argmax": {
114 "group": "tensor",
115 "generator_args": [
116 [
117 "--target-dtype",
118 "fp32",
119 "--target-dtype",
120 "fp16",
121 "--target-dtype",
122 "bf16",
123 "--fp-values-range",
124 "-2.0,2.0",
125 "--tensor-dim-range",
126 "32,64",
127 "--target-rank",
128 "1",
129 "--target-rank",
130 "2",
131 "--target-rank",
132 "3",
133 "--num-rand-permutations",
134 "2"
135 ],
136 [
137 "--target-dtype",
138 "fp32",
139 "--target-dtype",
140 "fp16",
141 "--target-dtype",
142 "bf16",
143 "--fp-values-range",
144 "-2.0,2.0",
145 "--tensor-dim-range",
146 "1,32",
147 "--target-rank",
148 "4",
149 "--num-rand-permutations",
150 "2"
151 ],
152 [
153 "--target-dtype",
154 "fp32",
155 "--fp-values-range",
156 "-2.0,2.0",
157 "--target-shape",
158 "1,3,65535,1",
159 "--target-shape",
160 "1,65537,1,2",
161 "--num-rand-permutations",
162 "2"
163 ]
164 ],
165 "params": {
166 },
167 "permutes": [
168 "shape",
169 "type"
170 ],
171 "profile": [
172 "tosa-mi"
173 ]
174 },
Jeremy Johnsone4b08ff2022-09-15 10:38:17 +0100[diff] [blame]175 "avg_pool2d": {
176 "group": "tensor",
177 "generator_args": [
178 [
179 "--target-dtype",
180 "fp32",
181 "--target-dtype",
182 "fp16",
183 "--target-dtype",
184 "bf16",
185 "--fp-values-range",
186 "-2.0,2.0",
187 "--target-shape",
188 "1,49,42,28",
189 "--target-shape",
190 "3,11,44,3",
191 "--allow-pooling-and-conv-oversizes"
192 ],
193 [
194 "--target-dtype",
195 "fp32",
196 "--fp-values-range",
197 "-2.0,2.0",
198 "--target-shape",
199 "1,65535,5,1",
200 "--target-shape",
201 "1,3,65537,1",
202 "--allow-pooling-and-conv-oversizes"
203 ]
204 ],
205 "params": {
206 },
207 "permutes": [
208 "shape",
209 "type",
210 "accum_type"
211 ],
212 "sparsity": {
213 "pad": 17
214 },
215 "profile": [
216 "tosa-mi"
217 ]
218 },
James Ward71616fe2022-11-23 11:00:47 +0000[diff] [blame]219 "ceil": {
220 "group": "ew_unary",
221 "generator_args": [
222 [
223 "--target-dtype",
224 "fp32",
225 "--target-dtype",
226 "fp16",
227 "--target-dtype",
228 "bf16",
229 "--fp-values-range",
230 "-2.0,2.0",
231 "--tensor-dim-range",
232 "4,64",
233 "--target-rank",
234 "1",
235 "--target-rank",
236 "2",
237 "--target-rank",
238 "3"
239 ],
240 [
241 "--target-dtype",
242 "fp32",
243 "--target-dtype",
244 "fp16",
245 "--target-dtype",
246 "bf16",
247 "--fp-values-range",
248 "-2.0,2.0",
249 "--tensor-dim-range",
250 "1,16",
251 "--target-rank",
252 "4",
253 "--target-rank",
254 "5"
255 ],
256 [
257 "--target-dtype",
258 "fp16",
259 "--fp-values-range",
260 "-2.0,2.0",
261 "--target-shape",
262 "2,1,65530,1",
263 "--target-shape",
264 "1,1,65539,1"
265 ]
266 ],
267 "params": {},
268 "permutes": [
269 "shape",
270 "type"
271 ],
272 "profile": [
273 "tosa-mi"
274 ]
275 },
James Ward34071252022-12-07 15:48:47 +0000[diff] [blame]276 "clamp": {
277 "group": "activation",
278 "generator_args": [
279 [
280 "--target-dtype",
281 "fp32",
282 "--target-dtype",
283 "fp16",
284 "--target-dtype",
285 "bf16",
286 "--fp-values-range",
287 "-2.0,2.0",
288 "--tensor-dim-range",
289 "20,64",
290 "--target-rank",
291 "1",
292 "--target-rank",
293 "2",
294 "--target-rank",
295 "3"
296 ],
297 [
298 "--target-dtype",
299 "fp32",
300 "--target-dtype",
301 "fp16",
302 "--target-dtype",
303 "bf16",
304 "--fp-values-range",
305 "-2.0,2.0",
306 "--tensor-dim-range",
307 "1,20",
308 "--target-rank",
309 "4",
310 "--target-rank",
311 "5"
312 ],
313 [
314 "--target-dtype",
315 "fp32",
316 "--fp-values-range",
317 "-2.0,2.0",
318 "--target-shape",
319 "1,2,1,65536",
320 "--target-shape",
321 "1,3,65537,1,1"
322 ]
323 ],
324 "params": {},
325 "permutes": [
326 "shape",
327 "type"
328 ],
329 "profile": [
330 "tosa-mi"
331 ]
332 },
James Wardf0890992022-11-17 11:15:14 +0000[diff] [blame]333 "concat": {
334 "group": "data_layout",
335 "generator_args": [
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 "16,64",
347 "--target-rank",
348 "1",
349 "--target-rank",
350 "2",
351 "--target-rank",
352 "3",
353 "--num-const-inputs-concat",
354 "3"
355 ],
356 [
357 "--target-dtype",
358 "fp32",
359 "--target-dtype",
360 "fp16",
361 "--target-dtype",
362 "bf16",
363 "--fp-values-range",
364 "-2.0,2.0",
365 "--tensor-dim-range",
366 "1,16",
367 "--target-rank",
368 "4",
369 "--target-rank",
370 "5",
371 "--num-const-inputs-concat",
372 "1"
373 ],
374 [
375 "--target-dtype",
376 "fp32",
377 "--fp-values-range",
378 "-2.0,2.0",
379 "--target-shape",
380 "1,1,65534,1,1",
381 "--target-shape",
382 "1,65537,1,2",
383 "--num-const-inputs-concat",
384 "2"
385 ]
386 ],
387 "params": {},
388 "permutes": [
389 "shape",
390 "type"
391 ],
392 "profile": [
393 "tosa-mi"
394 ]
395 },
James Ward542dd3b2022-11-17 12:11:55 +0000[diff] [blame]396 "const": {
397 "group": "data_nodes",
398 "no_negative_tests": "true",
399 "generator_args": [
400 [
401 "--target-dtype",
402 "fp32",
403 "--target-dtype",
404 "fp16",
405 "--target-dtype",
406 "bf16",
407 "--fp-values-range",
408 "-2.0,2.0",
409 "--target-rank",
410 "1",
411 "--target-rank",
412 "2",
413 "--target-rank",
414 "3"
415 ],
416 [
417 "--target-dtype",
418 "fp32",
419 "--target-dtype",
420 "fp16",
421 "--target-dtype",
422 "bf16",
423 "--fp-values-range",
424 "-2.0,2.0",
425 "--tensor-dim-range",
426 "1,16",
427 "--target-rank",
428 "4",
429 "--target-rank",
430 "5"
431 ],
432 [
433 "--target-dtype",
434 "fp32",
435 "--fp-values-range",
436 "-2.0,2.0",
437 "--target-shape",
438 "1,65540,1,1,1",
439 "--target-shape",
440 "1,2,1,65529"
441 ]
442 ],
443 "params": {},
444 "permutes": [
445 "shape",
446 "type"
447 ],
448 "profile": [
449 "tosa-mi"
450 ]
451 },
Jeremy Johnsone4b08ff2022-09-15 10:38:17 +0100[diff] [blame]452 "conv2d": {
453 "group": "tensor",
454 "generator_args": [
455 [
456 "--target-dtype",
457 "fp32",
458 "--target-dtype",
459 "fp16",
460 "--target-dtype",
461 "bf16",
462 "--fp-values-range",
463 "-2.0,2.0",
464 "--target-shape",
465 "1,49,42,28",
466 "--target-shape",
467 "1,11,44,13",
468 "--allow-pooling-and-conv-oversizes"
469 ],
470 [
471 "--target-dtype",
472 "fp32",
473 "--fp-values-range",
474 "-2.0,2.0",
475 "--target-shape",
476 "1,65535,4,1",
477 "--target-shape",
478 "1,5,65536,1",
479 "--max-conv-dilation",
480 "1",
481 "--allow-pooling-and-conv-oversizes"
482 ]
483 ],
484 "params": {
485 "shape": [],
486 "type": [],
487 "kernel": [],
488 "stride": [],
489 "pad": [],
490 "dilation": []
491 },
492 "permutes": [
493 "kernel",
494 "shape",
495 "type",
496 "pad"
497 ],
498 "profile": [
499 "tosa-mi"
500 ]
James Ward52ac9d62022-11-16 12:53:49 +0000[diff] [blame]501 },
Jeremy Johnson35396f22023-01-04 17:05:25 +0000[diff] [blame^]502 "exp": {
503 "group": "ew_unary",
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 "15,64",
516 "--target-rank",
517 "1",
518 "--target-rank",
519 "2",
520 "--target-rank",
521 "3"
522 ],
523 [
524 "--target-dtype",
525 "fp16",
526 "--fp-values-range",
527 "-2.0,2.0",
528 "--tensor-dim-range",
529 "1,15",
530 "--target-rank",
531 "4",
532 "--target-rank",
533 "5"
534 ],
535 [
536 "--target-dtype",
537 "fp32",
538 "--fp-values-range",
539 "-2.0,2.0",
540 "--target-shape",
541 "1,1,65535,4",
542 "--target-shape",
543 "2,65536,1,1,1"
544 ]
545 ],
546 "params": {},
547 "permutes": [
548 "shape",
549 "type"
550 ],
551 "profile": [
552 "tosa-mi"
553 ]
554 },
James Ward71616fe2022-11-23 11:00:47 +0000[diff] [blame]555 "floor": {
556 "group": "ew_unary",
557 "generator_args": [
558 [
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 "--tensor-dim-range",
568 "4,64",
569 "--target-rank",
570 "1",
571 "--target-rank",
572 "2",
573 "--target-rank",
574 "3"
575 ],
576 [
577 "--target-dtype",
578 "fp32",
579 "--target-dtype",
580 "fp16",
581 "--target-dtype",
582 "bf16",
583 "--fp-values-range",
584 "-2.0,2.0",
585 "--tensor-dim-range",
586 "1,16",
587 "--target-rank",
588 "4",
589 "--target-rank",
590 "5"
591 ],
592 [
593 "--target-dtype",
594 "fp32",
595 "--fp-values-range",
596 "-2.0,2.0",
597 "--target-shape",
598 "1,1,65533,2",
599 "--target-shape",
600 "1,65537,1,1"
601 ]
602 ],
603 "params": {},
604 "permutes": [
605 "shape",
606 "type"
607 ],
608 "profile": [
609 "tosa-mi"
610 ]
611 },
James Ward990cc4b2022-11-16 15:36:37 +0000[diff] [blame]612 "gather": {
613 "group": "scatter_gather",
614 "generator_args": [
615 [
616 "--target-dtype",
617 "fp32",
618 "--target-dtype",
619 "fp16",
620 "--target-dtype",
621 "bf16",
622 "--fp-values-range",
623 "-2.0,2.0",
624 "--tensor-dim-range",
625 "5,72"
626 ],
627 [
628 "--target-dtype",
629 "fp32",
630 "--fp-values-range",
631 "-2.0,2.0",
632 "--target-shape",
633 "2,65536,1",
634 "--target-shape",
635 "1,2,65535",
636 "--target-shape",
637 "5000,1,1"
638 ]
639 ],
640 "params": {},
641 "permutes": [
642 "shape",
643 "type"
644 ],
645 "profile": [
646 "tosa-mi"
647 ]
648 },
James Ward542dd3b2022-11-17 12:11:55 +0000[diff] [blame]649 "identity": {
650 "group": "data_nodes",
651 "no_negative_tests": "true",
652 "generator_args": [
653 [
654 "--target-dtype",
655 "fp32",
656 "--target-dtype",
657 "fp16",
658 "--target-dtype",
659 "bf16",
660 "--fp-values-range",
661 "-2.0,2.0",
662 "--tensor-dim-range",
663 "1,67",
664 "--target-rank",
665 "1",
666 "--target-rank",
667 "2",
668 "--target-rank",
669 "3"
670 ],
671 [
672 "--target-dtype",
673 "fp32",
674 "--target-dtype",
675 "fp16",
676 "--target-dtype",
677 "bf16",
678 "--fp-values-range",
679 "-2.0,2.0",
680 "--tensor-dim-range",
681 "1,19",
682 "--target-rank",
683 "4",
684 "--target-rank",
685 "5"
686 ],
687 [
688 "--target-dtype",
689 "fp32",
690 "--fp-values-range",
691 "-2.0,2.0",
692 "--target-shape",
693 "1,1,65539,1,1",
694 "--target-shape",
695 "1,1,2,65529"
696 ]
697 ],
698 "params": {},
699 "permutes": [
700 "shape",
701 "type"
702 ],
703 "profile": [
704 "tosa-mi"
705 ]
706 },
Jeremy Johnson35396f22023-01-04 17:05:25 +0000[diff] [blame^]707 "log": {
708 "group": "ew_unary",
709 "generator_args": [
710 [
711 "--filter",
712 "^log$",
713 "--target-dtype",
714 "fp32",
715 "--target-dtype",
716 "fp16",
717 "--target-dtype",
718 "bf16",
719 "--fp-values-range",
720 "-2.0,2.0",
721 "--tensor-dim-range",
722 "15,64",
723 "--target-rank",
724 "1",
725 "--target-rank",
726 "2",
727 "--target-rank",
728 "3"
729 ],
730 [
731 "--filter",
732 "^log$",
733 "--target-dtype",
734 "fp32",
735 "--fp-values-range",
736 "-2.0,2.0",
737 "--tensor-dim-range",
738 "1,15",
739 "--target-rank",
740 "4",
741 "--target-rank",
742 "5"
743 ],
744 [
745 "--filter",
746 "^log$",
747 "--target-dtype",
748 "fp16",
749 "--fp-values-range",
750 "-2.0,2.0",
751 "--target-shape",
752 "1,65530,1,2",
753 "--target-shape",
754 "3,1,65540,1,1"
755 ]
756 ],
757 "params": {},
758 "permutes": [
759 "shape",
760 "type"
761 ],
762 "profile": [
763 "tosa-mi"
764 ]
765 },
James Ward52ac9d62022-11-16 12:53:49 +0000[diff] [blame]766 "max_pool2d": {
767 "group": "tensor",
768 "generator_args": [
769 [
770 "--target-dtype",
771 "fp32",
772 "--target-dtype",
773 "fp16",
774 "--target-dtype",
775 "bf16",
776 "--fp-values-range",
777 "-2.0,2.0",
778 "--target-shape",
779 "1,47,37,25",
780 "--target-shape",
781 "1,15,13,38",
782 "--allow-pooling-and-conv-oversizes"
783 ],
784 [
785 "--target-dtype",
786 "fp32",
787 "--fp-values-range",
788 "-2.0,2.0",
789 "--target-shape",
790 "1,65534,4,1",
791 "--target-shape",
792 "1,6,65536,1",
793 "--max-conv-dilation",
794 "1",
795 "--allow-pooling-and-conv-oversizes"
796 ]
797 ],
798 "params": {
799 "shape": [],
800 "type": [
801 "f32",
802 "f16",
803 "bf16"
804 ],
805 "stride": [],
806 "pad": []
807 },
808 "permutes": [
809 "shape",
810 "type",
811 "pad"
812 ],
813 "sparsity": {
814 "pad": 11
815 },
816 "profile": [
817 "tosa-mi"
818 ]
James Ward990cc4b2022-11-16 15:36:37 +0000[diff] [blame]819 },
James Ward71616fe2022-11-23 11:00:47 +0000[diff] [blame]820 "negate": {
821 "group": "ew_unary",
822 "generator_args": [
823 [
824 "--target-dtype",
825 "fp32",
826 "--target-dtype",
827 "fp16",
828 "--target-dtype",
829 "bf16",
830 "--fp-values-range",
831 "-2.0,2.0",
832 "--tensor-dim-range",
833 "17,64",
834 "--target-rank",
835 "1",
836 "--target-rank",
837 "2",
838 "--target-rank",
839 "3"
840 ],
841 [
842 "--target-dtype",
843 "fp32",
844 "--target-dtype",
845 "fp16",
846 "--target-dtype",
847 "bf16",
848 "--fp-values-range",
849 "-2.0,2.0",
850 "--tensor-dim-range",
851 "1,22",
852 "--target-rank",
853 "4",
854 "--target-rank",
855 "5"
856 ],
857 [
858 "--target-dtype",
859 "fp32",
860 "--fp-values-range",
861 "-2.0,2.0",
862 "--target-shape",
863 "1,65541,3,1",
864 "--target-shape",
865 "1,2,1,65532,1"
866 ]
867 ],
868 "params": {},
869 "permutes": [
870 "shape",
871 "type"
872 ],
873 "profile": [
874 "tosa-mi"
875 ]
876 },
James Wardf0890992022-11-17 11:15:14 +0000[diff] [blame]877 "pad": {
878 "group": "data_layout",
879 "generator_args": [
880 [
881 "--target-dtype",
882 "fp32",
883 "--target-dtype",
884 "fp16",
885 "--target-dtype",
886 "bf16",
887 "--fp-values-range",
888 "-2.0,2.0",
889 "--tensor-dim-range",
890 "1,65",
891 "--target-rank",
892 "1",
893 "--target-rank",
894 "2",
895 "--target-rank",
896 "3"
897 ],
898 [
899 "--target-dtype",
900 "fp16",
901 "--fp-values-range",
902 "-2.0,2.0",
903 "--tensor-dim-range",
904 "1,17",
905 "--target-rank",
906 "4"
907 ],
908 [
909 "--target-dtype",
910 "bf16",
911 "--fp-values-range",
912 "-2.0,2.0",
913 "--tensor-dim-range",
914 "1,16",
915 "--target-rank",
916 "5"
917 ],
918 [
919 "--target-dtype",
920 "fp32",
921 "--fp-values-range",
922 "-2.0,2.0",
923 "--target-shape",
924 "1,1,65539,1"
925 ]
926 ],
927 "params": {},
928 "permutes": [
929 "shape",
930 "type"
931 ],
932 "sparsity": {
933 "pad": 21
934 },
935 "preselected": [
936 {
937 "shape": "50",
938 "type": "bf16",
939 "pad": "pad11"
940 },
941 {
942 "shape": "63x46",
943 "type": "bf16",
944 "pad": "pad1010"
945 },
946 {
947 "shape": "6",
948 "type": "f16",
949 "pad": "pad01"
950 }
951 ],
952 "profile": [
953 "tosa-mi"
954 ]
955 },
Jeremy Johnson6ffb7c82022-12-05 16:59:28 +0000[diff] [blame]956 "pow": {
957 "group": "ew_binary",
958 "generator_args": [
959 [
960 "--target-dtype",
961 "fp32",
962 "--target-dtype",
963 "fp16",
964 "--target-dtype",
965 "bf16",
966 "--fp-values-range",
967 "-2.0,2.0",
968 "--tensor-dim-range",
969 "16,64",
970 "--target-rank",
971 "1",
972 "--target-rank",
973 "2",
974 "--target-rank",
975 "3"
976 ],
977 [
978 "--target-dtype",
979 "fp16",
980 "--fp-values-range",
981 "-2.0,2.0",
982 "--tensor-dim-range",
983 "1,16",
984 "--target-rank",
985 "4",
986 "--target-rank",
987 "5"
988 ],
989 [
990 "--target-dtype",
991 "bf16",
992 "--fp-values-range",
993 "-2.0,2.0",
994 "--target-shape",
995 "65534,3,1,1",
996 "--target-shape",
997 "2,65541,1,1,1"
998 ]
999 ],
1000 "params": {},
1001 "permutes": [
1002 "shape",
1003 "type"
1004 ],
1005 "profile": [
1006 "tosa-mi"
1007 ]
1008 },
Jeremy Johnson35396f22023-01-04 17:05:25 +0000[diff] [blame^]1009 "reciprocal": {
1010 "group": "ew_unary",
1011 "generator_args": [
1012 [
1013 "--target-dtype",
1014 "fp32",
1015 "--target-dtype",
1016 "fp16",
1017 "--target-dtype",
1018 "bf16",
1019 "--fp-values-range",
1020 "-2.0,2.0",
1021 "--tensor-dim-range",
1022 "15,64",
1023 "--target-rank",
1024 "1",
1025 "--target-rank",
1026 "2",
1027 "--target-rank",
1028 "3"
1029 ],
1030 [
1031 "--target-dtype",
1032 "fp16",
1033 "--fp-values-range",
1034 "-2.0,2.0",
1035 "--tensor-dim-range",
1036 "1,15",
1037 "--target-rank",
1038 "4",
1039 "--target-rank",
1040 "5"
1041 ],
1042 [
1043 "--target-dtype",
1044 "bf16",
1045 "--fp-values-range",
1046 "-2.0,2.0",
1047 "--target-shape",
1048 "1,1,65536,4",
1049 "--target-shape",
1050 "2,65535,1,1,1"
1051 ]
1052 ],
1053 "params": {},
1054 "permutes": [
1055 "shape",
1056 "type"
1057 ],
1058 "profile": [
1059 "tosa-mi"
1060 ]
1061 },
James Wardf0890992022-11-17 11:15:14 +0000[diff] [blame]1062 "reshape": {
1063 "group": "data_layout",
1064 "generator_args": [
1065 [
1066 "--target-dtype",
1067 "fp32",
1068 "--target-dtype",
1069 "fp16",
1070 "--target-dtype",
1071 "bf16",
1072 "--fp-values-range",
1073 "-2.0,2.0",
1074 "--num-rand-permutations",
1075 "2"
1076 ],
1077 [
1078 "--target-dtype",
1079 "fp32",
1080 "--fp-values-range",
1081 "-2.0,2.0",
1082 "--target-shape",
1083 "1,1,65536,1",
1084 "--target-shape",
1085 "1,65535,1,2",
1086 "--num-rand-permutations",
1087 "2"
1088 ]
1089 ],
1090 "params": {},
1091 "permutes": [
1092 "shape",
1093 "type",
1094 "rank",
1095 "perm"
1096 ],
1097 "profile": [
1098 "tosa-mi"
1099 ]
1100 },
1101 "reverse": {
1102 "group": "data_layout",
1103 "generator_args": [
1104 [
1105 "--target-dtype",
1106 "fp32",
1107 "--target-dtype",
1108 "fp16",
1109 "--target-dtype",
1110 "bf16",
1111 "--fp-values-range",
1112 "-2.0,2.0",
1113 "--tensor-dim-range",
1114 "1,48"
1115 ],
1116 [
1117 "--target-dtype",
1118 "fp32",
1119 "--fp-values-range",
1120 "-2.0,2.0",
1121 "--target-shape",
1122 "1,1,65537",
1123 "--target-shape",
1124 "1,65535,1,2"
1125 ]
1126 ],
1127 "params": {},
1128 "permutes": [
1129 "shape",
1130 "type"
1131 ],
1132 "profile": [
1133 "tosa-mi"
1134 ]
1135 },
Jeremy Johnson35396f22023-01-04 17:05:25 +0000[diff] [blame^]1136 "rsqrt": {
1137 "group": "ew_unary",
1138 "generator_args": [
1139 [
1140 "--target-dtype",
1141 "fp32",
1142 "--target-dtype",
1143 "fp16",
1144 "--target-dtype",
1145 "bf16",
1146 "--fp-values-range",
1147 "-2.0,2.0",
1148 "--tensor-dim-range",
1149 "15,64",
1150 "--target-rank",
1151 "1",
1152 "--target-rank",
1153 "2",
1154 "--target-rank",
1155 "3"
1156 ],
1157 [
1158 "--target-dtype",
1159 "bf16",
1160 "--fp-values-range",
1161 "-2.0,2.0",
1162 "--tensor-dim-range",
1163 "1,15",
1164 "--target-rank",
1165 "4",
1166 "--target-rank",
1167 "5"
1168 ],
1169 [
1170 "--target-dtype",
1171 "fp32",
1172 "--fp-values-range",
1173 "-2.0,2.0",
1174 "--target-shape",
1175 "1,1,3,65541",
1176 "--target-shape",
1177 "65532,1,3,1,1"
1178 ]
1179 ],
1180 "params": {},
1181 "permutes": [
1182 "shape",
1183 "type"
1184 ],
1185 "profile": [
1186 "tosa-mi"
1187 ]
1188 },
James Ward990cc4b2022-11-16 15:36:37 +0000[diff] [blame]1189 "scatter": {
1190 "group": "scatter_gather",
1191 "generator_args": [
1192 [
1193 "--target-dtype",
1194 "fp32",
1195 "--target-dtype",
1196 "fp16",
1197 "--target-dtype",
1198 "bf16",
1199 "--fp-values-range",
1200 "-2.0,2.0",
1201 "--tensor-dim-range",
1202 "5,56"
1203 ],
1204 [
1205 "--target-dtype",
1206 "fp32",
1207 "--fp-values-range",
1208 "-2.0,2.0",
1209 "--target-shape",
1210 "1,65541,1",
1211 "--target-shape",
1212 "1,1,65531",
1213 "--target-shape",
1214 "2989,6,1"
1215 ]
1216 ],
1217 "params": {},
1218 "permutes": [
1219 "shape",
1220 "type"
1221 ],
1222 "profile": [
1223 "tosa-mi"
1224 ]
James Wardf0890992022-11-17 11:15:14 +0000[diff] [blame]1225 },
1226 "tile": {
1227 "group": "data_layout",
1228 "generator_args": [
1229 [
1230 "--target-dtype",
1231 "fp32",
1232 "--target-dtype",
1233 "fp16",
1234 "--target-dtype",
1235 "bf16",
1236 "--fp-values-range",
1237 "-2.0,2.0",
1238 "--tensor-dim-range",
1239 "4,32"
1240 ],
1241 [
1242 "--target-dtype",
1243 "fp32",
1244 "--fp-values-range",
1245 "-2.0,2.0",
1246 "--target-shape",
1247 "65535",
1248 "--target-shape",
1249 "1,65543,2,1"
1250 ]
1251 ],
1252 "params": {},
1253 "permutes": [
1254 "shape",
1255 "type"
1256 ],
1257 "profile": [
1258 "tosa-mi"
1259 ]
1260 },
1261 "transpose": {
1262 "group": "data_layout",
1263 "generator_args": [
1264 [
1265 "--filter",
1266 "^transpose$",
1267 "--target-dtype",
1268 "fp32",
1269 "--target-dtype",
1270 "fp16",
1271 "--fp-values-range",
1272 "-2.0,2.0",
1273 "--target-rank",
1274 "1",
1275 "--target-rank",
1276 "2",
1277 "--num-rand-permutations",
1278 "2"
1279 ],
1280 [
1281 "--filter",
1282 "^transpose$",
1283 "--target-dtype",
1284 "fp32",
1285 "--target-dtype",
1286 "fp16",
1287 "--target-dtype",
1288 "bf16",
1289 "--fp-values-range",
1290 "-2.0,2.0",
1291 "--target-rank",
1292 "3",
1293 "--target-rank",
1294 "4",
1295 "--num-rand-permutations",
1296 "2"
1297 ],
1298 [
1299 "--filter",
1300 "^transpose$",
1301 "--target-dtype",
1302 "fp32",
1303 "--fp-values-range",
1304 "-2.0,2.0",
1305 "--target-shape",
1306 "1,1,65539,2",
1307 "--target-shape",
1308 "1,1,1,65535",
1309 "--num-rand-permutations",
1310 "1"
1311 ]
1312 ],
1313 "params": {},
1314 "permutes": [
1315 "shape",
1316 "type",
1317 "perm"
1318 ],
1319 "profile": [
1320 "tosa-mi"
1321 ]
James Wardb9c3a632022-11-22 13:14:50 +0000[diff] [blame]1322 },
1323 "maximum": {
1324 "group": "ew_binary",
1325 "generator_args": [
1326 [
1327 "--target-dtype",
1328 "fp32",
1329 "--target-dtype",
1330 "fp16",
1331 "--target-dtype",
1332 "bf16",
1333 "--fp-values-range",
1334 "-2.0,2.0",
1335 "--tensor-dim-range",
1336 "16,64",
1337 "--target-rank",
1338 "1",
1339 "--target-rank",
1340 "2",
1341 "--target-rank",
1342 "3"
1343 ],
1344 [
1345 "--target-dtype",
1346 "fp32",
1347 "--target-dtype",
1348 "fp16",
1349 "--target-dtype",
1350 "bf16",
1351 "--fp-values-range",
1352 "-2.0,2.0",
1353 "--tensor-dim-range",
1354 "1,16",
1355 "--target-rank",
1356 "4",
1357 "--target-rank",
1358 "5"
1359 ],
1360 [
1361 "--target-dtype",
1362 "fp32",
1363 "--fp-values-range",
1364 "-2.0,2.0",
1365 "--target-shape",
1366 "65531,1,2,1",
1367 "--target-shape",
1368 "1,1,65541,1"
1369 ]
1370 ],
1371 "params": {},
1372 "permutes": [
1373 "shape",
1374 "type"
1375 ],
1376 "profile": [
1377 "tosa-mi"
1378 ]
1379 },
1380 "minimum": {
1381 "group": "ew_binary",
1382 "generator_args": [
1383 [
1384 "--target-dtype",
1385 "fp32",
1386 "--target-dtype",
1387 "fp16",
1388 "--target-dtype",
1389 "bf16",
1390 "--fp-values-range",
1391 "-2.0,2.0",
1392 "--tensor-dim-range",
1393 "16,64",
1394 "--target-rank",
1395 "1",
1396 "--target-rank",
1397 "2",
1398 "--target-rank",
1399 "3"
1400 ],
1401 [
1402 "--target-dtype",
1403 "fp32",
1404 "--target-dtype",
1405 "fp16",
1406 "--target-dtype",
1407 "bf16",
1408 "--fp-values-range",
1409 "-2.0,2.0",
1410 "--tensor-dim-range",
1411 "1,16",
1412 "--target-rank",
1413 "4",
1414 "--target-rank",
1415 "5"
1416 ],
1417 [
1418 "--target-dtype",
1419 "fp32",
1420 "--fp-values-range",
1421 "-2.0,2.0",
1422 "--target-shape",
1423 "1,65531,2,1",
1424 "--target-shape",
1425 "1,1,1,65538"
1426 ]
1427 ],
1428 "params": {},
1429 "permutes": [
1430 "shape",
1431 "type"
1432 ],
1433 "profile": [
1434 "tosa-mi"
1435 ]
James Ward635bc992022-11-23 11:55:32 +0000[diff] [blame]1436 },
Jeremy Johnsondd8d9c22022-12-12 14:18:10 +0000[diff] [blame]1437 "mul": {
1438 "group": "ew_binary",
1439 "generator_negative_dim_range": "1,10",
1440 "generator_args": [
1441 [
1442 "--target-dtype",
1443 "fp32",
1444 "--target-dtype",
1445 "fp16",
1446 "--target-dtype",
1447 "bf16",
1448 "--fp-values-range",
1449 "-2.0,2.0",
1450 "--tensor-dim-range",
1451 "16,64",
1452 "--target-rank",
1453 "1",
1454 "--target-rank",
1455 "2",
1456 "--target-rank",
1457 "3"
1458 ],
1459 [
1460 "--target-dtype",
1461 "fp16",
1462 "--fp-values-range",
1463 "-2.0,2.0",
1464 "--tensor-dim-range",
1465 "1,16",
1466 "--target-rank",
1467 "4",
1468 "--target-rank",
1469 "5"
1470 ],
1471 [
1472 "--target-dtype",
1473 "bf16",
1474 "--fp-values-range",
1475 "-2.0,2.0",
1476 "--target-shape",
1477 "1,1,3,65534",
1478 "--target-shape",
1479 "1,2,1,65541,1"
1480 ]
1481 ],
1482 "params": {},
1483 "permutes": [
1484 "shape",
1485 "type"
1486 ],
1487 "profile": [
1488 "tosa-mi"
1489 ]
1490 },
James Ward635bc992022-11-23 11:55:32 +0000[diff] [blame]1491 "reduce_max": {
1492 "group": "reduction",
1493 "generator_args": [
1494 [
1495 "--target-dtype",
1496 "fp32",
1497 "--target-dtype",
1498 "fp16",
1499 "--target-dtype",
1500 "bf16",
1501 "--fp-values-range",
1502 "-2.0,2.0",
1503 "--tensor-dim-range",
1504 "1,32"
1505 ],
1506 [
1507 "--target-dtype",
1508 "fp32",
1509 "--fp-values-range",
1510 "-2.0,2.0",
1511 "--target-shape",
1512 "1,1,1,65531",
1513 "--target-shape",
1514 "1,1,65539,2"
1515 ]
1516 ],
1517 "params": {},
1518 "permutes": [
1519 "shape",
1520 "type",
1521 "axis"
1522 ],
1523 "profile": [
1524 "tosa-mi"
1525 ]
1526 },
1527 "reduce_min": {
1528 "group": "reduction",
1529 "generator_args": [
1530 [
1531 "--target-dtype",
1532 "fp32",
1533 "--target-dtype",
1534 "fp16",
1535 "--target-dtype",
1536 "bf16",
1537 "--fp-values-range",
1538 "-2.0,2.0",
1539 "--tensor-dim-range",
1540 "1,32"
1541 ],
1542 [
1543 "--target-dtype",
1544 "fp32",
1545 "--fp-values-range",
1546 "-2.0,2.0",
1547 "--target-shape",
1548 "1,1,65531,1",
1549 "--target-shape",
1550 "1,65539,2,1"
1551 ]
1552 ],
1553 "params": {},
1554 "permutes": [
1555 "shape",
1556 "type",
1557 "axis"
1558 ],
1559 "profile": [
1560 "tosa-mi"
1561 ]
James Ward2a27c8f2022-11-23 11:29:43 +0000[diff] [blame]1562 },
James Ward7f1ea8e2022-12-12 18:39:11 +0000[diff] [blame]1563 "resize": {
1564 "group": "image",
1565 "generator_args": [
1566 [
1567 "--target-dtype",
1568 "fp32",
1569 "--target-dtype",
1570 "fp16",
1571 "--target-dtype",
1572 "bf16",
1573 "--fp-values-range",
1574 "-2.0,2.0"
1575 ],
1576 [
1577 "--target-dtype",
1578 "fp32",
1579 "--fp-values-range",
1580 "-2.0,2.0",
1581 "--target-shape",
1582 "1,1103,1,2",
1583 "--max-resize-output-dim",
1584 "9100"
1585 ]
1586 ],
1587 "params": {
1588 "offset": [],
1589 "shape": [],
1590 "type": [],
1591 "mode": [],
1592 "output_type": [],
1593 "scale": ["sc2x1x3x1"],
1594 "border": ["bor0x1"]
1595 },
1596 "permutes": [
1597 "shape",
1598 "type"
1599 ],
1600 "profile": [
1601 "tosa-mi"
1602 ]
1603 },
James Ward2a27c8f2022-11-23 11:29:43 +0000[diff] [blame]1604 "select": {
1605 "group": "ew_ternary",
1606 "generator_args": [
1607 [
1608 "--target-dtype",
1609 "fp32",
1610 "--target-dtype",
1611 "fp16",
1612 "--target-dtype",
1613 "bf16",
1614 "--fp-values-range",
1615 "-2.0,2.0",
1616 "--tensor-dim-range",
1617 "16,64",
1618 "--target-rank",
1619 "1",
1620 "--target-rank",
1621 "2",
1622 "--target-rank",
1623 "3"
1624 ],
1625 [
1626 "--target-dtype",
1627 "fp32",
1628 "--target-dtype",
1629 "fp16",
1630 "--target-dtype",
1631 "bf16",
1632 "--fp-values-range",
1633 "-2.0,2.0",
1634 "--tensor-dim-range",
1635 "1,16",
1636 "--target-rank",
1637 "4",
1638 "--target-rank",
1639 "5"
1640 ],
1641 [
1642 "--target-dtype",
1643 "fp32",
1644 "--fp-values-range",
1645 "-2.0,2.0",
1646 "--target-shape",
1647 "1,2,65534,2,1",
1648 "--target-shape",
1649 "1,1,65539,2",
1650 "--num-rand-permutations",
1651 "2"
1652 ]
1653 ],
1654 "params": {},
1655 "permutes": [
1656 "shape",
1657 "type"
1658 ],
1659 "profile": [
1660 "tosa-mi"
1661 ]
James Wardb45db9a2022-12-12 13:02:44 +0000[diff] [blame]1662 },
1663 "sigmoid": {
1664 "group": "activation",
1665 "generator_args": [
1666 [
1667 "--target-dtype",
1668 "fp32",
1669 "--target-dtype",
1670 "fp16",
1671 "--target-dtype",
1672 "bf16",
1673 "--fp-values-range",
1674 "-2.0,2.0",
1675 "--tensor-dim-range",
1676 "16,64",
1677 "--target-rank",
1678 "1",
1679 "--target-rank",
1680 "2",
1681 "--target-rank",
1682 "3"
1683 ],
1684 [
1685 "--target-dtype",
1686 "fp32",
1687 "--target-dtype",
1688 "fp16",
1689 "--target-dtype",
1690 "bf16",
1691 "--fp-values-range",
1692 "-2.0,2.0",
1693 "--tensor-dim-range",
1694 "1,19",
1695 "--target-rank",
1696 "4",
1697 "--target-rank",
1698 "5"
1699 ],
1700 [
1701 "--target-dtype",
1702 "fp32",
1703 "--fp-values-range",
1704 "-2.0,2.0",
1705 "--target-shape",
1706 "1,2,65538,2,1",
1707 "--target-shape",
1708 "1,3,1,65533"
1709 ]
1710 ],
1711 "params": {},
1712 "permutes": [
1713 "shape",
1714 "type"
1715 ],
1716 "profile": [
1717 "tosa-mi"
1718 ]
1719 },
Jeremy Johnsondd8d9c22022-12-12 14:18:10 +0000[diff] [blame]1720 "sub": {
1721 "group": "ew_binary",
1722 "generator_args": [
1723 [
1724 "--target-dtype",
1725 "fp32",
1726 "--target-dtype",
1727 "fp16",
1728 "--target-dtype",
1729 "bf16",
1730 "--fp-values-range",
1731 "-2.0,2.0",
1732 "--tensor-dim-range",
1733 "16,64",
1734 "--target-rank",
1735 "1",
1736 "--target-rank",
1737 "2",
1738 "--target-rank",
1739 "3"
1740 ],
1741 [
1742 "--target-dtype",
1743 "bf16",
1744 "--fp-values-range",
1745 "-2.0,2.0",
1746 "--tensor-dim-range",
1747 "1,16",
1748 "--target-rank",
1749 "4",
1750 "--target-rank",
1751 "5"
1752 ],
1753 [
1754 "--target-dtype",
1755 "fp32",
1756 "--fp-values-range",
1757 "-2.0,2.0",
1758 "--target-shape",
1759 "1,1,65533,3",
1760 "--target-shape",
1761 "1,1,2,1,65545"
1762 ]
1763 ],
1764 "params": {},
1765 "permutes": [
1766 "shape",
1767 "type"
1768 ],
1769 "profile": [
1770 "tosa-mi"
1771 ]
1772 },
James Wardc253e642023-01-17 17:03:12 +0000[diff] [blame]1773 "slice": {
1774 "group": "data_layout",
1775 "generator_args": [
1776 [
1777 "--target-dtype",
1778 "fp32",
1779 "--target-dtype",
1780 "fp16",
1781 "--target-dtype",
1782 "bf16",
1783 "--fp-values-range",
1784 "-2.0,2.0",
1785 "--tensor-dim-range",
1786 "16,64",
1787 "--target-rank",
1788 "1",
1789 "--target-rank",
1790 "2",
1791 "--target-rank",
1792 "3",
1793 "--num-rand-permutations",
1794 "2"
1795 ],
1796 [
1797 "--target-dtype",
1798 "fp32",
1799 "--target-dtype",
1800 "fp16",
1801 "--target-dtype",
1802 "bf16",
1803 "--fp-values-range",
1804 "-2.0,2.0",
1805 "--tensor-dim-range",
1806 "1,16",
1807 "--target-rank",
1808 "4",
1809 "--num-rand-permutations",
1810 "2"
1811 ],
1812 [
1813 "--target-dtype",
1814 "bf16",
1815 "--fp-values-range",
1816 "-2.0,2.0",
1817 "--target-shape",
1818 "1,1,655341,1",
1819 "--target-shape",
1820 "1,65539,1,1",
1821 "--num-rand-permutations",
1822 "2"
1823 ]
1824 ],
1825 "params": {},
1826 "permutes": [
1827 "shape",
1828 "type",
1829 "perm"
1830 ],
1831 "profile": [
1832 "tosa-bi",
1833 "tosa-mi"
1834 ]
1835 },
James Wardb45db9a2022-12-12 13:02:44 +0000[diff] [blame]1836 "tanh": {
1837 "group": "activation",
1838 "generator_args": [
1839 [
1840 "--target-dtype",
1841 "fp32",
1842 "--target-dtype",
1843 "fp16",
1844 "--target-dtype",
1845 "bf16",
1846 "--fp-values-range",
1847 "-2.0,2.0",
1848 "--tensor-dim-range",
1849 "18,60",
1850 "--target-rank",
1851 "1",
1852 "--target-rank",
1853 "2",
1854 "--target-rank",
1855 "3"
1856 ],
1857 [
1858 "--target-dtype",
1859 "fp32",
1860 "--target-dtype",
1861 "fp16",
1862 "--target-dtype",
1863 "bf16",
1864 "--fp-values-range",
1865 "-2.0,2.0",
1866 "--tensor-dim-range",
1867 "1,24",
1868 "--target-rank",
1869 "4",
1870 "--target-rank",
1871 "5"
1872 ],
1873 [
1874 "--target-dtype",
1875 "fp32",
1876 "--fp-values-range",
1877 "-2.0,2.0",
1878 "--target-shape",
1879 "1,65535,2,1,1",
1880 "--target-shape",
1881 "1,65540,1,2"
1882 ]
1883 ],
1884 "params": {},
1885 "permutes": [
1886 "shape",
1887 "type"
1888 ],
1889 "profile": [
1890 "tosa-mi"
1891 ]
Jeremy Johnsone4b08ff2022-09-15 10:38:17 +0100[diff] [blame]1892 }
1893}