Gitiles
Code Review Sign In
review.mlplatform.org / tosa / reference_model / dd8d9c251db0fece6453d86116052ad7f3e2d697 / . / verif / conformance / tosa_main_profile_ops_info.json
blob: a175d3e184a116848f6967cdb55eb16c230e82ac [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 Wardf0890992022-11-17 11:15:14 +0000[diff] [blame]276 "concat": {
277 "group": "data_layout",
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 "16,64",
290 "--target-rank",
291 "1",
292 "--target-rank",
293 "2",
294 "--target-rank",
295 "3",
296 "--num-const-inputs-concat",
297 "3"
298 ],
299 [
300 "--target-dtype",
301 "fp32",
302 "--target-dtype",
303 "fp16",
304 "--target-dtype",
305 "bf16",
306 "--fp-values-range",
307 "-2.0,2.0",
308 "--tensor-dim-range",
309 "1,16",
310 "--target-rank",
311 "4",
312 "--target-rank",
313 "5",
314 "--num-const-inputs-concat",
315 "1"
316 ],
317 [
318 "--target-dtype",
319 "fp32",
320 "--fp-values-range",
321 "-2.0,2.0",
322 "--target-shape",
323 "1,1,65534,1,1",
324 "--target-shape",
325 "1,65537,1,2",
326 "--num-const-inputs-concat",
327 "2"
328 ]
329 ],
330 "params": {},
331 "permutes": [
332 "shape",
333 "type"
334 ],
335 "profile": [
336 "tosa-mi"
337 ]
338 },
James Ward542dd3b2022-11-17 12:11:55 +0000[diff] [blame]339 "const": {
340 "group": "data_nodes",
341 "no_negative_tests": "true",
342 "generator_args": [
343 [
344 "--target-dtype",
345 "fp32",
346 "--target-dtype",
347 "fp16",
348 "--target-dtype",
349 "bf16",
350 "--fp-values-range",
351 "-2.0,2.0",
352 "--target-rank",
353 "1",
354 "--target-rank",
355 "2",
356 "--target-rank",
357 "3"
358 ],
359 [
360 "--target-dtype",
361 "fp32",
362 "--target-dtype",
363 "fp16",
364 "--target-dtype",
365 "bf16",
366 "--fp-values-range",
367 "-2.0,2.0",
368 "--tensor-dim-range",
369 "1,16",
370 "--target-rank",
371 "4",
372 "--target-rank",
373 "5"
374 ],
375 [
376 "--target-dtype",
377 "fp32",
378 "--fp-values-range",
379 "-2.0,2.0",
380 "--target-shape",
381 "1,65540,1,1,1",
382 "--target-shape",
383 "1,2,1,65529"
384 ]
385 ],
386 "params": {},
387 "permutes": [
388 "shape",
389 "type"
390 ],
391 "profile": [
392 "tosa-mi"
393 ]
394 },
Jeremy Johnsone4b08ff2022-09-15 10:38:17 +0100[diff] [blame]395 "conv2d": {
396 "group": "tensor",
397 "generator_args": [
398 [
399 "--target-dtype",
400 "fp32",
401 "--target-dtype",
402 "fp16",
403 "--target-dtype",
404 "bf16",
405 "--fp-values-range",
406 "-2.0,2.0",
407 "--target-shape",
408 "1,49,42,28",
409 "--target-shape",
410 "1,11,44,13",
411 "--allow-pooling-and-conv-oversizes"
412 ],
413 [
414 "--target-dtype",
415 "fp32",
416 "--fp-values-range",
417 "-2.0,2.0",
418 "--target-shape",
419 "1,65535,4,1",
420 "--target-shape",
421 "1,5,65536,1",
422 "--max-conv-dilation",
423 "1",
424 "--allow-pooling-and-conv-oversizes"
425 ]
426 ],
427 "params": {
428 "shape": [],
429 "type": [],
430 "kernel": [],
431 "stride": [],
432 "pad": [],
433 "dilation": []
434 },
435 "permutes": [
436 "kernel",
437 "shape",
438 "type",
439 "pad"
440 ],
441 "profile": [
442 "tosa-mi"
443 ]
James Ward52ac9d62022-11-16 12:53:49 +0000[diff] [blame]444 },
James Ward71616fe2022-11-23 11:00:47 +0000[diff] [blame]445 "floor": {
446 "group": "ew_unary",
447 "generator_args": [
448 [
449 "--target-dtype",
450 "fp32",
451 "--target-dtype",
452 "fp16",
453 "--target-dtype",
454 "bf16",
455 "--fp-values-range",
456 "-2.0,2.0",
457 "--tensor-dim-range",
458 "4,64",
459 "--target-rank",
460 "1",
461 "--target-rank",
462 "2",
463 "--target-rank",
464 "3"
465 ],
466 [
467 "--target-dtype",
468 "fp32",
469 "--target-dtype",
470 "fp16",
471 "--target-dtype",
472 "bf16",
473 "--fp-values-range",
474 "-2.0,2.0",
475 "--tensor-dim-range",
476 "1,16",
477 "--target-rank",
478 "4",
479 "--target-rank",
480 "5"
481 ],
482 [
483 "--target-dtype",
484 "fp32",
485 "--fp-values-range",
486 "-2.0,2.0",
487 "--target-shape",
488 "1,1,65533,2",
489 "--target-shape",
490 "1,65537,1,1"
491 ]
492 ],
493 "params": {},
494 "permutes": [
495 "shape",
496 "type"
497 ],
498 "profile": [
499 "tosa-mi"
500 ]
501 },
James Ward990cc4b2022-11-16 15:36:37 +0000[diff] [blame]502 "gather": {
503 "group": "scatter_gather",
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 "5,72"
516 ],
517 [
518 "--target-dtype",
519 "fp32",
520 "--fp-values-range",
521 "-2.0,2.0",
522 "--target-shape",
523 "2,65536,1",
524 "--target-shape",
525 "1,2,65535",
526 "--target-shape",
527 "5000,1,1"
528 ]
529 ],
530 "params": {},
531 "permutes": [
532 "shape",
533 "type"
534 ],
535 "profile": [
536 "tosa-mi"
537 ]
538 },
James Ward542dd3b2022-11-17 12:11:55 +0000[diff] [blame]539 "identity": {
540 "group": "data_nodes",
541 "no_negative_tests": "true",
542 "generator_args": [
543 [
544 "--target-dtype",
545 "fp32",
546 "--target-dtype",
547 "fp16",
548 "--target-dtype",
549 "bf16",
550 "--fp-values-range",
551 "-2.0,2.0",
552 "--tensor-dim-range",
553 "1,67",
554 "--target-rank",
555 "1",
556 "--target-rank",
557 "2",
558 "--target-rank",
559 "3"
560 ],
561 [
562 "--target-dtype",
563 "fp32",
564 "--target-dtype",
565 "fp16",
566 "--target-dtype",
567 "bf16",
568 "--fp-values-range",
569 "-2.0,2.0",
570 "--tensor-dim-range",
571 "1,19",
572 "--target-rank",
573 "4",
574 "--target-rank",
575 "5"
576 ],
577 [
578 "--target-dtype",
579 "fp32",
580 "--fp-values-range",
581 "-2.0,2.0",
582 "--target-shape",
583 "1,1,65539,1,1",
584 "--target-shape",
585 "1,1,2,65529"
586 ]
587 ],
588 "params": {},
589 "permutes": [
590 "shape",
591 "type"
592 ],
593 "profile": [
594 "tosa-mi"
595 ]
596 },
James Ward52ac9d62022-11-16 12:53:49 +0000[diff] [blame]597 "max_pool2d": {
598 "group": "tensor",
599 "generator_args": [
600 [
601 "--target-dtype",
602 "fp32",
603 "--target-dtype",
604 "fp16",
605 "--target-dtype",
606 "bf16",
607 "--fp-values-range",
608 "-2.0,2.0",
609 "--target-shape",
610 "1,47,37,25",
611 "--target-shape",
612 "1,15,13,38",
613 "--allow-pooling-and-conv-oversizes"
614 ],
615 [
616 "--target-dtype",
617 "fp32",
618 "--fp-values-range",
619 "-2.0,2.0",
620 "--target-shape",
621 "1,65534,4,1",
622 "--target-shape",
623 "1,6,65536,1",
624 "--max-conv-dilation",
625 "1",
626 "--allow-pooling-and-conv-oversizes"
627 ]
628 ],
629 "params": {
630 "shape": [],
631 "type": [
632 "f32",
633 "f16",
634 "bf16"
635 ],
636 "stride": [],
637 "pad": []
638 },
639 "permutes": [
640 "shape",
641 "type",
642 "pad"
643 ],
644 "sparsity": {
645 "pad": 11
646 },
647 "profile": [
648 "tosa-mi"
649 ]
James Ward990cc4b2022-11-16 15:36:37 +0000[diff] [blame]650 },
James Ward71616fe2022-11-23 11:00:47 +0000[diff] [blame]651 "negate": {
652 "group": "ew_unary",
653 "generator_args": [
654 [
655 "--target-dtype",
656 "fp32",
657 "--target-dtype",
658 "fp16",
659 "--target-dtype",
660 "bf16",
661 "--fp-values-range",
662 "-2.0,2.0",
663 "--tensor-dim-range",
664 "17,64",
665 "--target-rank",
666 "1",
667 "--target-rank",
668 "2",
669 "--target-rank",
670 "3"
671 ],
672 [
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 "--tensor-dim-range",
682 "1,22",
683 "--target-rank",
684 "4",
685 "--target-rank",
686 "5"
687 ],
688 [
689 "--target-dtype",
690 "fp32",
691 "--fp-values-range",
692 "-2.0,2.0",
693 "--target-shape",
694 "1,65541,3,1",
695 "--target-shape",
696 "1,2,1,65532,1"
697 ]
698 ],
699 "params": {},
700 "permutes": [
701 "shape",
702 "type"
703 ],
704 "profile": [
705 "tosa-mi"
706 ]
707 },
James Wardf0890992022-11-17 11:15:14 +0000[diff] [blame]708 "pad": {
709 "group": "data_layout",
710 "generator_args": [
711 [
712 "--target-dtype",
713 "fp32",
714 "--target-dtype",
715 "fp16",
716 "--target-dtype",
717 "bf16",
718 "--fp-values-range",
719 "-2.0,2.0",
720 "--tensor-dim-range",
721 "1,65",
722 "--target-rank",
723 "1",
724 "--target-rank",
725 "2",
726 "--target-rank",
727 "3"
728 ],
729 [
730 "--target-dtype",
731 "fp16",
732 "--fp-values-range",
733 "-2.0,2.0",
734 "--tensor-dim-range",
735 "1,17",
736 "--target-rank",
737 "4"
738 ],
739 [
740 "--target-dtype",
741 "bf16",
742 "--fp-values-range",
743 "-2.0,2.0",
744 "--tensor-dim-range",
745 "1,16",
746 "--target-rank",
747 "5"
748 ],
749 [
750 "--target-dtype",
751 "fp32",
752 "--fp-values-range",
753 "-2.0,2.0",
754 "--target-shape",
755 "1,1,65539,1"
756 ]
757 ],
758 "params": {},
759 "permutes": [
760 "shape",
761 "type"
762 ],
763 "sparsity": {
764 "pad": 21
765 },
766 "preselected": [
767 {
768 "shape": "50",
769 "type": "bf16",
770 "pad": "pad11"
771 },
772 {
773 "shape": "63x46",
774 "type": "bf16",
775 "pad": "pad1010"
776 },
777 {
778 "shape": "6",
779 "type": "f16",
780 "pad": "pad01"
781 }
782 ],
783 "profile": [
784 "tosa-mi"
785 ]
786 },
Jeremy Johnson6ffb7c82022-12-05 16:59:28 +0000[diff] [blame]787 "pow": {
788 "group": "ew_binary",
789 "generator_args": [
790 [
791 "--target-dtype",
792 "fp32",
793 "--target-dtype",
794 "fp16",
795 "--target-dtype",
796 "bf16",
797 "--fp-values-range",
798 "-2.0,2.0",
799 "--tensor-dim-range",
800 "16,64",
801 "--target-rank",
802 "1",
803 "--target-rank",
804 "2",
805 "--target-rank",
806 "3"
807 ],
808 [
809 "--target-dtype",
810 "fp16",
811 "--fp-values-range",
812 "-2.0,2.0",
813 "--tensor-dim-range",
814 "1,16",
815 "--target-rank",
816 "4",
817 "--target-rank",
818 "5"
819 ],
820 [
821 "--target-dtype",
822 "bf16",
823 "--fp-values-range",
824 "-2.0,2.0",
825 "--target-shape",
826 "65534,3,1,1",
827 "--target-shape",
828 "2,65541,1,1,1"
829 ]
830 ],
831 "params": {},
832 "permutes": [
833 "shape",
834 "type"
835 ],
836 "profile": [
837 "tosa-mi"
838 ]
839 },
James Wardf0890992022-11-17 11:15:14 +0000[diff] [blame]840 "reshape": {
841 "group": "data_layout",
842 "generator_args": [
843 [
844 "--target-dtype",
845 "fp32",
846 "--target-dtype",
847 "fp16",
848 "--target-dtype",
849 "bf16",
850 "--fp-values-range",
851 "-2.0,2.0",
852 "--num-rand-permutations",
853 "2"
854 ],
855 [
856 "--target-dtype",
857 "fp32",
858 "--fp-values-range",
859 "-2.0,2.0",
860 "--target-shape",
861 "1,1,65536,1",
862 "--target-shape",
863 "1,65535,1,2",
864 "--num-rand-permutations",
865 "2"
866 ]
867 ],
868 "params": {},
869 "permutes": [
870 "shape",
871 "type",
872 "rank",
873 "perm"
874 ],
875 "profile": [
876 "tosa-mi"
877 ]
878 },
879 "reverse": {
880 "group": "data_layout",
881 "generator_args": [
882 [
883 "--target-dtype",
884 "fp32",
885 "--target-dtype",
886 "fp16",
887 "--target-dtype",
888 "bf16",
889 "--fp-values-range",
890 "-2.0,2.0",
891 "--tensor-dim-range",
892 "1,48"
893 ],
894 [
895 "--target-dtype",
896 "fp32",
897 "--fp-values-range",
898 "-2.0,2.0",
899 "--target-shape",
900 "1,1,65537",
901 "--target-shape",
902 "1,65535,1,2"
903 ]
904 ],
905 "params": {},
906 "permutes": [
907 "shape",
908 "type"
909 ],
910 "profile": [
911 "tosa-mi"
912 ]
913 },
James Ward990cc4b2022-11-16 15:36:37 +0000[diff] [blame]914 "scatter": {
915 "group": "scatter_gather",
916 "generator_args": [
917 [
918 "--target-dtype",
919 "fp32",
920 "--target-dtype",
921 "fp16",
922 "--target-dtype",
923 "bf16",
924 "--fp-values-range",
925 "-2.0,2.0",
926 "--tensor-dim-range",
927 "5,56"
928 ],
929 [
930 "--target-dtype",
931 "fp32",
932 "--fp-values-range",
933 "-2.0,2.0",
934 "--target-shape",
935 "1,65541,1",
936 "--target-shape",
937 "1,1,65531",
938 "--target-shape",
939 "2989,6,1"
940 ]
941 ],
942 "params": {},
943 "permutes": [
944 "shape",
945 "type"
946 ],
947 "profile": [
948 "tosa-mi"
949 ]
James Wardf0890992022-11-17 11:15:14 +0000[diff] [blame]950 },
951 "tile": {
952 "group": "data_layout",
953 "generator_args": [
954 [
955 "--target-dtype",
956 "fp32",
957 "--target-dtype",
958 "fp16",
959 "--target-dtype",
960 "bf16",
961 "--fp-values-range",
962 "-2.0,2.0",
963 "--tensor-dim-range",
964 "4,32"
965 ],
966 [
967 "--target-dtype",
968 "fp32",
969 "--fp-values-range",
970 "-2.0,2.0",
971 "--target-shape",
972 "65535",
973 "--target-shape",
974 "1,65543,2,1"
975 ]
976 ],
977 "params": {},
978 "permutes": [
979 "shape",
980 "type"
981 ],
982 "profile": [
983 "tosa-mi"
984 ]
985 },
986 "transpose": {
987 "group": "data_layout",
988 "generator_args": [
989 [
990 "--filter",
991 "^transpose$",
992 "--target-dtype",
993 "fp32",
994 "--target-dtype",
995 "fp16",
996 "--fp-values-range",
997 "-2.0,2.0",
998 "--target-rank",
999 "1",
1000 "--target-rank",
1001 "2",
1002 "--num-rand-permutations",
1003 "2"
1004 ],
1005 [
1006 "--filter",
1007 "^transpose$",
1008 "--target-dtype",
1009 "fp32",
1010 "--target-dtype",
1011 "fp16",
1012 "--target-dtype",
1013 "bf16",
1014 "--fp-values-range",
1015 "-2.0,2.0",
1016 "--target-rank",
1017 "3",
1018 "--target-rank",
1019 "4",
1020 "--num-rand-permutations",
1021 "2"
1022 ],
1023 [
1024 "--filter",
1025 "^transpose$",
1026 "--target-dtype",
1027 "fp32",
1028 "--fp-values-range",
1029 "-2.0,2.0",
1030 "--target-shape",
1031 "1,1,65539,2",
1032 "--target-shape",
1033 "1,1,1,65535",
1034 "--num-rand-permutations",
1035 "1"
1036 ]
1037 ],
1038 "params": {},
1039 "permutes": [
1040 "shape",
1041 "type",
1042 "perm"
1043 ],
1044 "profile": [
1045 "tosa-mi"
1046 ]
James Wardb9c3a632022-11-22 13:14:50 +0000[diff] [blame]1047 },
1048 "maximum": {
1049 "group": "ew_binary",
1050 "generator_args": [
1051 [
1052 "--target-dtype",
1053 "fp32",
1054 "--target-dtype",
1055 "fp16",
1056 "--target-dtype",
1057 "bf16",
1058 "--fp-values-range",
1059 "-2.0,2.0",
1060 "--tensor-dim-range",
1061 "16,64",
1062 "--target-rank",
1063 "1",
1064 "--target-rank",
1065 "2",
1066 "--target-rank",
1067 "3"
1068 ],
1069 [
1070 "--target-dtype",
1071 "fp32",
1072 "--target-dtype",
1073 "fp16",
1074 "--target-dtype",
1075 "bf16",
1076 "--fp-values-range",
1077 "-2.0,2.0",
1078 "--tensor-dim-range",
1079 "1,16",
1080 "--target-rank",
1081 "4",
1082 "--target-rank",
1083 "5"
1084 ],
1085 [
1086 "--target-dtype",
1087 "fp32",
1088 "--fp-values-range",
1089 "-2.0,2.0",
1090 "--target-shape",
1091 "65531,1,2,1",
1092 "--target-shape",
1093 "1,1,65541,1"
1094 ]
1095 ],
1096 "params": {},
1097 "permutes": [
1098 "shape",
1099 "type"
1100 ],
1101 "profile": [
1102 "tosa-mi"
1103 ]
1104 },
1105 "minimum": {
1106 "group": "ew_binary",
1107 "generator_args": [
1108 [
1109 "--target-dtype",
1110 "fp32",
1111 "--target-dtype",
1112 "fp16",
1113 "--target-dtype",
1114 "bf16",
1115 "--fp-values-range",
1116 "-2.0,2.0",
1117 "--tensor-dim-range",
1118 "16,64",
1119 "--target-rank",
1120 "1",
1121 "--target-rank",
1122 "2",
1123 "--target-rank",
1124 "3"
1125 ],
1126 [
1127 "--target-dtype",
1128 "fp32",
1129 "--target-dtype",
1130 "fp16",
1131 "--target-dtype",
1132 "bf16",
1133 "--fp-values-range",
1134 "-2.0,2.0",
1135 "--tensor-dim-range",
1136 "1,16",
1137 "--target-rank",
1138 "4",
1139 "--target-rank",
1140 "5"
1141 ],
1142 [
1143 "--target-dtype",
1144 "fp32",
1145 "--fp-values-range",
1146 "-2.0,2.0",
1147 "--target-shape",
1148 "1,65531,2,1",
1149 "--target-shape",
1150 "1,1,1,65538"
1151 ]
1152 ],
1153 "params": {},
1154 "permutes": [
1155 "shape",
1156 "type"
1157 ],
1158 "profile": [
1159 "tosa-mi"
1160 ]
James Ward635bc992022-11-23 11:55:32 +0000[diff] [blame]1161 },
Jeremy Johnsondd8d9c22022-12-12 14:18:10 +0000[diff] [blame^]1162 "mul": {
1163 "group": "ew_binary",
1164 "generator_negative_dim_range": "1,10",
1165 "generator_args": [
1166 [
1167 "--target-dtype",
1168 "fp32",
1169 "--target-dtype",
1170 "fp16",
1171 "--target-dtype",
1172 "bf16",
1173 "--fp-values-range",
1174 "-2.0,2.0",
1175 "--tensor-dim-range",
1176 "16,64",
1177 "--target-rank",
1178 "1",
1179 "--target-rank",
1180 "2",
1181 "--target-rank",
1182 "3"
1183 ],
1184 [
1185 "--target-dtype",
1186 "fp16",
1187 "--fp-values-range",
1188 "-2.0,2.0",
1189 "--tensor-dim-range",
1190 "1,16",
1191 "--target-rank",
1192 "4",
1193 "--target-rank",
1194 "5"
1195 ],
1196 [
1197 "--target-dtype",
1198 "bf16",
1199 "--fp-values-range",
1200 "-2.0,2.0",
1201 "--target-shape",
1202 "1,1,3,65534",
1203 "--target-shape",
1204 "1,2,1,65541,1"
1205 ]
1206 ],
1207 "params": {},
1208 "permutes": [
1209 "shape",
1210 "type"
1211 ],
1212 "profile": [
1213 "tosa-mi"
1214 ]
1215 },
James Ward635bc992022-11-23 11:55:32 +0000[diff] [blame]1216 "reduce_max": {
1217 "group": "reduction",
1218 "generator_args": [
1219 [
1220 "--target-dtype",
1221 "fp32",
1222 "--target-dtype",
1223 "fp16",
1224 "--target-dtype",
1225 "bf16",
1226 "--fp-values-range",
1227 "-2.0,2.0",
1228 "--tensor-dim-range",
1229 "1,32"
1230 ],
1231 [
1232 "--target-dtype",
1233 "fp32",
1234 "--fp-values-range",
1235 "-2.0,2.0",
1236 "--target-shape",
1237 "1,1,1,65531",
1238 "--target-shape",
1239 "1,1,65539,2"
1240 ]
1241 ],
1242 "params": {},
1243 "permutes": [
1244 "shape",
1245 "type",
1246 "axis"
1247 ],
1248 "profile": [
1249 "tosa-mi"
1250 ]
1251 },
1252 "reduce_min": {
1253 "group": "reduction",
1254 "generator_args": [
1255 [
1256 "--target-dtype",
1257 "fp32",
1258 "--target-dtype",
1259 "fp16",
1260 "--target-dtype",
1261 "bf16",
1262 "--fp-values-range",
1263 "-2.0,2.0",
1264 "--tensor-dim-range",
1265 "1,32"
1266 ],
1267 [
1268 "--target-dtype",
1269 "fp32",
1270 "--fp-values-range",
1271 "-2.0,2.0",
1272 "--target-shape",
1273 "1,1,65531,1",
1274 "--target-shape",
1275 "1,65539,2,1"
1276 ]
1277 ],
1278 "params": {},
1279 "permutes": [
1280 "shape",
1281 "type",
1282 "axis"
1283 ],
1284 "profile": [
1285 "tosa-mi"
1286 ]
James Ward2a27c8f2022-11-23 11:29:43 +0000[diff] [blame]1287 },
James Ward7f1ea8e2022-12-12 18:39:11 +0000[diff] [blame]1288 "resize": {
1289 "group": "image",
1290 "generator_args": [
1291 [
1292 "--target-dtype",
1293 "fp32",
1294 "--target-dtype",
1295 "fp16",
1296 "--target-dtype",
1297 "bf16",
1298 "--fp-values-range",
1299 "-2.0,2.0"
1300 ],
1301 [
1302 "--target-dtype",
1303 "fp32",
1304 "--fp-values-range",
1305 "-2.0,2.0",
1306 "--target-shape",
1307 "1,1103,1,2",
1308 "--max-resize-output-dim",
1309 "9100"
1310 ]
1311 ],
1312 "params": {
1313 "offset": [],
1314 "shape": [],
1315 "type": [],
1316 "mode": [],
1317 "output_type": [],
1318 "scale": ["sc2x1x3x1"],
1319 "border": ["bor0x1"]
1320 },
1321 "permutes": [
1322 "shape",
1323 "type"
1324 ],
1325 "profile": [
1326 "tosa-mi"
1327 ]
1328 },
James Ward2a27c8f2022-11-23 11:29:43 +0000[diff] [blame]1329 "select": {
1330 "group": "ew_ternary",
1331 "generator_args": [
1332 [
1333 "--target-dtype",
1334 "fp32",
1335 "--target-dtype",
1336 "fp16",
1337 "--target-dtype",
1338 "bf16",
1339 "--fp-values-range",
1340 "-2.0,2.0",
1341 "--tensor-dim-range",
1342 "16,64",
1343 "--target-rank",
1344 "1",
1345 "--target-rank",
1346 "2",
1347 "--target-rank",
1348 "3"
1349 ],
1350 [
1351 "--target-dtype",
1352 "fp32",
1353 "--target-dtype",
1354 "fp16",
1355 "--target-dtype",
1356 "bf16",
1357 "--fp-values-range",
1358 "-2.0,2.0",
1359 "--tensor-dim-range",
1360 "1,16",
1361 "--target-rank",
1362 "4",
1363 "--target-rank",
1364 "5"
1365 ],
1366 [
1367 "--target-dtype",
1368 "fp32",
1369 "--fp-values-range",
1370 "-2.0,2.0",
1371 "--target-shape",
1372 "1,2,65534,2,1",
1373 "--target-shape",
1374 "1,1,65539,2",
1375 "--num-rand-permutations",
1376 "2"
1377 ]
1378 ],
1379 "params": {},
1380 "permutes": [
1381 "shape",
1382 "type"
1383 ],
1384 "profile": [
1385 "tosa-mi"
1386 ]
James Wardb45db9a2022-12-12 13:02:44 +0000[diff] [blame]1387 },
1388 "sigmoid": {
1389 "group": "activation",
1390 "generator_args": [
1391 [
1392 "--target-dtype",
1393 "fp32",
1394 "--target-dtype",
1395 "fp16",
1396 "--target-dtype",
1397 "bf16",
1398 "--fp-values-range",
1399 "-2.0,2.0",
1400 "--tensor-dim-range",
1401 "16,64",
1402 "--target-rank",
1403 "1",
1404 "--target-rank",
1405 "2",
1406 "--target-rank",
1407 "3"
1408 ],
1409 [
1410 "--target-dtype",
1411 "fp32",
1412 "--target-dtype",
1413 "fp16",
1414 "--target-dtype",
1415 "bf16",
1416 "--fp-values-range",
1417 "-2.0,2.0",
1418 "--tensor-dim-range",
1419 "1,19",
1420 "--target-rank",
1421 "4",
1422 "--target-rank",
1423 "5"
1424 ],
1425 [
1426 "--target-dtype",
1427 "fp32",
1428 "--fp-values-range",
1429 "-2.0,2.0",
1430 "--target-shape",
1431 "1,2,65538,2,1",
1432 "--target-shape",
1433 "1,3,1,65533"
1434 ]
1435 ],
1436 "params": {},
1437 "permutes": [
1438 "shape",
1439 "type"
1440 ],
1441 "profile": [
1442 "tosa-mi"
1443 ]
1444 },
Jeremy Johnsondd8d9c22022-12-12 14:18:10 +0000[diff] [blame^]1445 "sub": {
1446 "group": "ew_binary",
1447 "generator_args": [
1448 [
1449 "--target-dtype",
1450 "fp32",
1451 "--target-dtype",
1452 "fp16",
1453 "--target-dtype",
1454 "bf16",
1455 "--fp-values-range",
1456 "-2.0,2.0",
1457 "--tensor-dim-range",
1458 "16,64",
1459 "--target-rank",
1460 "1",
1461 "--target-rank",
1462 "2",
1463 "--target-rank",
1464 "3"
1465 ],
1466 [
1467 "--target-dtype",
1468 "bf16",
1469 "--fp-values-range",
1470 "-2.0,2.0",
1471 "--tensor-dim-range",
1472 "1,16",
1473 "--target-rank",
1474 "4",
1475 "--target-rank",
1476 "5"
1477 ],
1478 [
1479 "--target-dtype",
1480 "fp32",
1481 "--fp-values-range",
1482 "-2.0,2.0",
1483 "--target-shape",
1484 "1,1,65533,3",
1485 "--target-shape",
1486 "1,1,2,1,65545"
1487 ]
1488 ],
1489 "params": {},
1490 "permutes": [
1491 "shape",
1492 "type"
1493 ],
1494 "profile": [
1495 "tosa-mi"
1496 ]
1497 },
James Wardb45db9a2022-12-12 13:02:44 +0000[diff] [blame]1498 "tanh": {
1499 "group": "activation",
1500 "generator_args": [
1501 [
1502 "--target-dtype",
1503 "fp32",
1504 "--target-dtype",
1505 "fp16",
1506 "--target-dtype",
1507 "bf16",
1508 "--fp-values-range",
1509 "-2.0,2.0",
1510 "--tensor-dim-range",
1511 "18,60",
1512 "--target-rank",
1513 "1",
1514 "--target-rank",
1515 "2",
1516 "--target-rank",
1517 "3"
1518 ],
1519 [
1520 "--target-dtype",
1521 "fp32",
1522 "--target-dtype",
1523 "fp16",
1524 "--target-dtype",
1525 "bf16",
1526 "--fp-values-range",
1527 "-2.0,2.0",
1528 "--tensor-dim-range",
1529 "1,24",
1530 "--target-rank",
1531 "4",
1532 "--target-rank",
1533 "5"
1534 ],
1535 [
1536 "--target-dtype",
1537 "fp32",
1538 "--fp-values-range",
1539 "-2.0,2.0",
1540 "--target-shape",
1541 "1,65535,2,1,1",
1542 "--target-shape",
1543 "1,65540,1,2"
1544 ]
1545 ],
1546 "params": {},
1547 "permutes": [
1548 "shape",
1549 "type"
1550 ],
1551 "profile": [
1552 "tosa-mi"
1553 ]
Jeremy Johnsone4b08ff2022-09-15 10:38:17 +0100[diff] [blame]1554 }
1555}