Gitiles
Code Review Sign In
review.mlplatform.org / tosa / reference_model / 05c711e8941b05f6c9502fe8ef482a077aca508b / . / verif / conformance / tosa_main_profile_ops_info.json
blob: 0cbe04499451f6335a787727e2139707780c248a [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 },
James Ward52ac9d62022-11-16 12:53:49 +0000[diff] [blame]60 "argmax": {
61 "group": "tensor",
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 "32,64",
74 "--target-rank",
75 "1",
76 "--target-rank",
77 "2",
78 "--target-rank",
79 "3",
80 "--num-rand-permutations",
81 "2"
82 ],
83 [
84 "--target-dtype",
85 "fp32",
86 "--target-dtype",
87 "fp16",
88 "--target-dtype",
89 "bf16",
90 "--fp-values-range",
91 "-2.0,2.0",
92 "--tensor-dim-range",
93 "1,32",
94 "--target-rank",
95 "4",
96 "--num-rand-permutations",
97 "2"
98 ],
99 [
100 "--target-dtype",
101 "fp32",
102 "--fp-values-range",
103 "-2.0,2.0",
104 "--target-shape",
105 "1,3,65535,1",
106 "--target-shape",
107 "1,65537,1,2",
108 "--num-rand-permutations",
109 "2"
110 ]
111 ],
112 "params": {
113 },
114 "permutes": [
115 "shape",
116 "type"
117 ],
118 "profile": [
119 "tosa-mi"
120 ]
121 },
Jeremy Johnsone4b08ff2022-09-15 10:38:17 +0100[diff] [blame]122 "avg_pool2d": {
123 "group": "tensor",
124 "generator_args": [
125 [
126 "--target-dtype",
127 "fp32",
128 "--target-dtype",
129 "fp16",
130 "--target-dtype",
131 "bf16",
132 "--fp-values-range",
133 "-2.0,2.0",
134 "--target-shape",
135 "1,49,42,28",
136 "--target-shape",
137 "3,11,44,3",
138 "--allow-pooling-and-conv-oversizes"
139 ],
140 [
141 "--target-dtype",
142 "fp32",
143 "--fp-values-range",
144 "-2.0,2.0",
145 "--target-shape",
146 "1,65535,5,1",
147 "--target-shape",
148 "1,3,65537,1",
149 "--allow-pooling-and-conv-oversizes"
150 ]
151 ],
152 "params": {
153 },
154 "permutes": [
155 "shape",
156 "type",
157 "accum_type"
158 ],
159 "sparsity": {
160 "pad": 17
161 },
162 "profile": [
163 "tosa-mi"
164 ]
165 },
James Ward71616fe2022-11-23 11:00:47 +0000[diff] [blame]166 "ceil": {
167 "group": "ew_unary",
168 "generator_args": [
169 [
170 "--target-dtype",
171 "fp32",
172 "--target-dtype",
173 "fp16",
174 "--target-dtype",
175 "bf16",
176 "--fp-values-range",
177 "-2.0,2.0",
178 "--tensor-dim-range",
179 "4,64",
180 "--target-rank",
181 "1",
182 "--target-rank",
183 "2",
184 "--target-rank",
185 "3"
186 ],
187 [
188 "--target-dtype",
189 "fp32",
190 "--target-dtype",
191 "fp16",
192 "--target-dtype",
193 "bf16",
194 "--fp-values-range",
195 "-2.0,2.0",
196 "--tensor-dim-range",
197 "1,16",
198 "--target-rank",
199 "4",
200 "--target-rank",
201 "5"
202 ],
203 [
204 "--target-dtype",
205 "fp16",
206 "--fp-values-range",
207 "-2.0,2.0",
208 "--target-shape",
209 "2,1,65530,1",
210 "--target-shape",
211 "1,1,65539,1"
212 ]
213 ],
214 "params": {},
215 "permutes": [
216 "shape",
217 "type"
218 ],
219 "profile": [
220 "tosa-mi"
221 ]
222 },
James Wardf0890992022-11-17 11:15:14 +0000[diff] [blame]223 "concat": {
224 "group": "data_layout",
225 "generator_args": [
226 [
227 "--target-dtype",
228 "fp32",
229 "--target-dtype",
230 "fp16",
231 "--target-dtype",
232 "bf16",
233 "--fp-values-range",
234 "-2.0,2.0",
235 "--tensor-dim-range",
236 "16,64",
237 "--target-rank",
238 "1",
239 "--target-rank",
240 "2",
241 "--target-rank",
242 "3",
243 "--num-const-inputs-concat",
244 "3"
245 ],
246 [
247 "--target-dtype",
248 "fp32",
249 "--target-dtype",
250 "fp16",
251 "--target-dtype",
252 "bf16",
253 "--fp-values-range",
254 "-2.0,2.0",
255 "--tensor-dim-range",
256 "1,16",
257 "--target-rank",
258 "4",
259 "--target-rank",
260 "5",
261 "--num-const-inputs-concat",
262 "1"
263 ],
264 [
265 "--target-dtype",
266 "fp32",
267 "--fp-values-range",
268 "-2.0,2.0",
269 "--target-shape",
270 "1,1,65534,1,1",
271 "--target-shape",
272 "1,65537,1,2",
273 "--num-const-inputs-concat",
274 "2"
275 ]
276 ],
277 "params": {},
278 "permutes": [
279 "shape",
280 "type"
281 ],
282 "profile": [
283 "tosa-mi"
284 ]
285 },
James Ward542dd3b2022-11-17 12:11:55 +0000[diff] [blame]286 "const": {
287 "group": "data_nodes",
288 "no_negative_tests": "true",
289 "generator_args": [
290 [
291 "--target-dtype",
292 "fp32",
293 "--target-dtype",
294 "fp16",
295 "--target-dtype",
296 "bf16",
297 "--fp-values-range",
298 "-2.0,2.0",
299 "--target-rank",
300 "1",
301 "--target-rank",
302 "2",
303 "--target-rank",
304 "3"
305 ],
306 [
307 "--target-dtype",
308 "fp32",
309 "--target-dtype",
310 "fp16",
311 "--target-dtype",
312 "bf16",
313 "--fp-values-range",
314 "-2.0,2.0",
315 "--tensor-dim-range",
316 "1,16",
317 "--target-rank",
318 "4",
319 "--target-rank",
320 "5"
321 ],
322 [
323 "--target-dtype",
324 "fp32",
325 "--fp-values-range",
326 "-2.0,2.0",
327 "--target-shape",
328 "1,65540,1,1,1",
329 "--target-shape",
330 "1,2,1,65529"
331 ]
332 ],
333 "params": {},
334 "permutes": [
335 "shape",
336 "type"
337 ],
338 "profile": [
339 "tosa-mi"
340 ]
341 },
Jeremy Johnsone4b08ff2022-09-15 10:38:17 +0100[diff] [blame]342 "conv2d": {
343 "group": "tensor",
344 "generator_args": [
345 [
346 "--target-dtype",
347 "fp32",
348 "--target-dtype",
349 "fp16",
350 "--target-dtype",
351 "bf16",
352 "--fp-values-range",
353 "-2.0,2.0",
354 "--target-shape",
355 "1,49,42,28",
356 "--target-shape",
357 "1,11,44,13",
358 "--allow-pooling-and-conv-oversizes"
359 ],
360 [
361 "--target-dtype",
362 "fp32",
363 "--fp-values-range",
364 "-2.0,2.0",
365 "--target-shape",
366 "1,65535,4,1",
367 "--target-shape",
368 "1,5,65536,1",
369 "--max-conv-dilation",
370 "1",
371 "--allow-pooling-and-conv-oversizes"
372 ]
373 ],
374 "params": {
375 "shape": [],
376 "type": [],
377 "kernel": [],
378 "stride": [],
379 "pad": [],
380 "dilation": []
381 },
382 "permutes": [
383 "kernel",
384 "shape",
385 "type",
386 "pad"
387 ],
388 "profile": [
389 "tosa-mi"
390 ]
James Ward52ac9d62022-11-16 12:53:49 +0000[diff] [blame]391 },
James Ward71616fe2022-11-23 11:00:47 +0000[diff] [blame]392 "floor": {
393 "group": "ew_unary",
394 "generator_args": [
395 [
396 "--target-dtype",
397 "fp32",
398 "--target-dtype",
399 "fp16",
400 "--target-dtype",
401 "bf16",
402 "--fp-values-range",
403 "-2.0,2.0",
404 "--tensor-dim-range",
405 "4,64",
406 "--target-rank",
407 "1",
408 "--target-rank",
409 "2",
410 "--target-rank",
411 "3"
412 ],
413 [
414 "--target-dtype",
415 "fp32",
416 "--target-dtype",
417 "fp16",
418 "--target-dtype",
419 "bf16",
420 "--fp-values-range",
421 "-2.0,2.0",
422 "--tensor-dim-range",
423 "1,16",
424 "--target-rank",
425 "4",
426 "--target-rank",
427 "5"
428 ],
429 [
430 "--target-dtype",
431 "fp32",
432 "--fp-values-range",
433 "-2.0,2.0",
434 "--target-shape",
435 "1,1,65533,2",
436 "--target-shape",
437 "1,65537,1,1"
438 ]
439 ],
440 "params": {},
441 "permutes": [
442 "shape",
443 "type"
444 ],
445 "profile": [
446 "tosa-mi"
447 ]
448 },
James Ward990cc4b2022-11-16 15:36:37 +0000[diff] [blame]449 "gather": {
450 "group": "scatter_gather",
451 "generator_args": [
452 [
453 "--target-dtype",
454 "fp32",
455 "--target-dtype",
456 "fp16",
457 "--target-dtype",
458 "bf16",
459 "--fp-values-range",
460 "-2.0,2.0",
461 "--tensor-dim-range",
462 "5,72"
463 ],
464 [
465 "--target-dtype",
466 "fp32",
467 "--fp-values-range",
468 "-2.0,2.0",
469 "--target-shape",
470 "2,65536,1",
471 "--target-shape",
472 "1,2,65535",
473 "--target-shape",
474 "5000,1,1"
475 ]
476 ],
477 "params": {},
478 "permutes": [
479 "shape",
480 "type"
481 ],
482 "profile": [
483 "tosa-mi"
484 ]
485 },
James Ward542dd3b2022-11-17 12:11:55 +0000[diff] [blame]486 "identity": {
487 "group": "data_nodes",
488 "no_negative_tests": "true",
489 "generator_args": [
490 [
491 "--target-dtype",
492 "fp32",
493 "--target-dtype",
494 "fp16",
495 "--target-dtype",
496 "bf16",
497 "--fp-values-range",
498 "-2.0,2.0",
499 "--tensor-dim-range",
500 "1,67",
501 "--target-rank",
502 "1",
503 "--target-rank",
504 "2",
505 "--target-rank",
506 "3"
507 ],
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 "--tensor-dim-range",
518 "1,19",
519 "--target-rank",
520 "4",
521 "--target-rank",
522 "5"
523 ],
524 [
525 "--target-dtype",
526 "fp32",
527 "--fp-values-range",
528 "-2.0,2.0",
529 "--target-shape",
530 "1,1,65539,1,1",
531 "--target-shape",
532 "1,1,2,65529"
533 ]
534 ],
535 "params": {},
536 "permutes": [
537 "shape",
538 "type"
539 ],
540 "profile": [
541 "tosa-mi"
542 ]
543 },
James Ward52ac9d62022-11-16 12:53:49 +0000[diff] [blame]544 "max_pool2d": {
545 "group": "tensor",
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 "--target-shape",
557 "1,47,37,25",
558 "--target-shape",
559 "1,15,13,38",
560 "--allow-pooling-and-conv-oversizes"
561 ],
562 [
563 "--target-dtype",
564 "fp32",
565 "--fp-values-range",
566 "-2.0,2.0",
567 "--target-shape",
568 "1,65534,4,1",
569 "--target-shape",
570 "1,6,65536,1",
571 "--max-conv-dilation",
572 "1",
573 "--allow-pooling-and-conv-oversizes"
574 ]
575 ],
576 "params": {
577 "shape": [],
578 "type": [
579 "f32",
580 "f16",
581 "bf16"
582 ],
583 "stride": [],
584 "pad": []
585 },
586 "permutes": [
587 "shape",
588 "type",
589 "pad"
590 ],
591 "sparsity": {
592 "pad": 11
593 },
594 "profile": [
595 "tosa-mi"
596 ]
James Ward990cc4b2022-11-16 15:36:37 +0000[diff] [blame]597 },
James Ward71616fe2022-11-23 11:00:47 +0000[diff] [blame]598 "negate": {
599 "group": "ew_unary",
600 "generator_args": [
601 [
602 "--target-dtype",
603 "fp32",
604 "--target-dtype",
605 "fp16",
606 "--target-dtype",
607 "bf16",
608 "--fp-values-range",
609 "-2.0,2.0",
610 "--tensor-dim-range",
611 "17,64",
612 "--target-rank",
613 "1",
614 "--target-rank",
615 "2",
616 "--target-rank",
617 "3"
618 ],
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 "1,22",
630 "--target-rank",
631 "4",
632 "--target-rank",
633 "5"
634 ],
635 [
636 "--target-dtype",
637 "fp32",
638 "--fp-values-range",
639 "-2.0,2.0",
640 "--target-shape",
641 "1,65541,3,1",
642 "--target-shape",
643 "1,2,1,65532,1"
644 ]
645 ],
646 "params": {},
647 "permutes": [
648 "shape",
649 "type"
650 ],
651 "profile": [
652 "tosa-mi"
653 ]
654 },
James Wardf0890992022-11-17 11:15:14 +0000[diff] [blame]655 "pad": {
656 "group": "data_layout",
657 "generator_args": [
658 [
659 "--target-dtype",
660 "fp32",
661 "--target-dtype",
662 "fp16",
663 "--target-dtype",
664 "bf16",
665 "--fp-values-range",
666 "-2.0,2.0",
667 "--tensor-dim-range",
668 "1,65",
669 "--target-rank",
670 "1",
671 "--target-rank",
672 "2",
673 "--target-rank",
674 "3"
675 ],
676 [
677 "--target-dtype",
678 "fp16",
679 "--fp-values-range",
680 "-2.0,2.0",
681 "--tensor-dim-range",
682 "1,17",
683 "--target-rank",
684 "4"
685 ],
686 [
687 "--target-dtype",
688 "bf16",
689 "--fp-values-range",
690 "-2.0,2.0",
691 "--tensor-dim-range",
692 "1,16",
693 "--target-rank",
694 "5"
695 ],
696 [
697 "--target-dtype",
698 "fp32",
699 "--fp-values-range",
700 "-2.0,2.0",
701 "--target-shape",
702 "1,1,65539,1"
703 ]
704 ],
705 "params": {},
706 "permutes": [
707 "shape",
708 "type"
709 ],
710 "sparsity": {
711 "pad": 21
712 },
713 "preselected": [
714 {
715 "shape": "50",
716 "type": "bf16",
717 "pad": "pad11"
718 },
719 {
720 "shape": "63x46",
721 "type": "bf16",
722 "pad": "pad1010"
723 },
724 {
725 "shape": "6",
726 "type": "f16",
727 "pad": "pad01"
728 }
729 ],
730 "profile": [
731 "tosa-mi"
732 ]
733 },
734 "reshape": {
735 "group": "data_layout",
736 "generator_args": [
737 [
738 "--target-dtype",
739 "fp32",
740 "--target-dtype",
741 "fp16",
742 "--target-dtype",
743 "bf16",
744 "--fp-values-range",
745 "-2.0,2.0",
746 "--num-rand-permutations",
747 "2"
748 ],
749 [
750 "--target-dtype",
751 "fp32",
752 "--fp-values-range",
753 "-2.0,2.0",
754 "--target-shape",
755 "1,1,65536,1",
756 "--target-shape",
757 "1,65535,1,2",
758 "--num-rand-permutations",
759 "2"
760 ]
761 ],
762 "params": {},
763 "permutes": [
764 "shape",
765 "type",
766 "rank",
767 "perm"
768 ],
769 "profile": [
770 "tosa-mi"
771 ]
772 },
773 "reverse": {
774 "group": "data_layout",
775 "generator_args": [
776 [
777 "--target-dtype",
778 "fp32",
779 "--target-dtype",
780 "fp16",
781 "--target-dtype",
782 "bf16",
783 "--fp-values-range",
784 "-2.0,2.0",
785 "--tensor-dim-range",
786 "1,48"
787 ],
788 [
789 "--target-dtype",
790 "fp32",
791 "--fp-values-range",
792 "-2.0,2.0",
793 "--target-shape",
794 "1,1,65537",
795 "--target-shape",
796 "1,65535,1,2"
797 ]
798 ],
799 "params": {},
800 "permutes": [
801 "shape",
802 "type"
803 ],
804 "profile": [
805 "tosa-mi"
806 ]
807 },
James Ward990cc4b2022-11-16 15:36:37 +0000[diff] [blame]808 "scatter": {
809 "group": "scatter_gather",
810 "generator_args": [
811 [
812 "--target-dtype",
813 "fp32",
814 "--target-dtype",
815 "fp16",
816 "--target-dtype",
817 "bf16",
818 "--fp-values-range",
819 "-2.0,2.0",
820 "--tensor-dim-range",
821 "5,56"
822 ],
823 [
824 "--target-dtype",
825 "fp32",
826 "--fp-values-range",
827 "-2.0,2.0",
828 "--target-shape",
829 "1,65541,1",
830 "--target-shape",
831 "1,1,65531",
832 "--target-shape",
833 "2989,6,1"
834 ]
835 ],
836 "params": {},
837 "permutes": [
838 "shape",
839 "type"
840 ],
841 "profile": [
842 "tosa-mi"
843 ]
James Wardf0890992022-11-17 11:15:14 +0000[diff] [blame]844 },
845 "tile": {
846 "group": "data_layout",
847 "generator_args": [
848 [
849 "--target-dtype",
850 "fp32",
851 "--target-dtype",
852 "fp16",
853 "--target-dtype",
854 "bf16",
855 "--fp-values-range",
856 "-2.0,2.0",
857 "--tensor-dim-range",
858 "4,32"
859 ],
860 [
861 "--target-dtype",
862 "fp32",
863 "--fp-values-range",
864 "-2.0,2.0",
865 "--target-shape",
866 "65535",
867 "--target-shape",
868 "1,65543,2,1"
869 ]
870 ],
871 "params": {},
872 "permutes": [
873 "shape",
874 "type"
875 ],
876 "profile": [
877 "tosa-mi"
878 ]
879 },
880 "transpose": {
881 "group": "data_layout",
882 "generator_args": [
883 [
884 "--filter",
885 "^transpose$",
886 "--target-dtype",
887 "fp32",
888 "--target-dtype",
889 "fp16",
890 "--fp-values-range",
891 "-2.0,2.0",
892 "--target-rank",
893 "1",
894 "--target-rank",
895 "2",
896 "--num-rand-permutations",
897 "2"
898 ],
899 [
900 "--filter",
901 "^transpose$",
902 "--target-dtype",
903 "fp32",
904 "--target-dtype",
905 "fp16",
906 "--target-dtype",
907 "bf16",
908 "--fp-values-range",
909 "-2.0,2.0",
910 "--target-rank",
911 "3",
912 "--target-rank",
913 "4",
914 "--num-rand-permutations",
915 "2"
916 ],
917 [
918 "--filter",
919 "^transpose$",
920 "--target-dtype",
921 "fp32",
922 "--fp-values-range",
923 "-2.0,2.0",
924 "--target-shape",
925 "1,1,65539,2",
926 "--target-shape",
927 "1,1,1,65535",
928 "--num-rand-permutations",
929 "1"
930 ]
931 ],
932 "params": {},
933 "permutes": [
934 "shape",
935 "type",
936 "perm"
937 ],
938 "profile": [
939 "tosa-mi"
940 ]
James Wardb9c3a632022-11-22 13:14:50 +0000[diff] [blame]941 },
942 "maximum": {
943 "group": "ew_binary",
944 "generator_args": [
945 [
946 "--target-dtype",
947 "fp32",
948 "--target-dtype",
949 "fp16",
950 "--target-dtype",
951 "bf16",
952 "--fp-values-range",
953 "-2.0,2.0",
954 "--tensor-dim-range",
955 "16,64",
956 "--target-rank",
957 "1",
958 "--target-rank",
959 "2",
960 "--target-rank",
961 "3"
962 ],
963 [
964 "--target-dtype",
965 "fp32",
966 "--target-dtype",
967 "fp16",
968 "--target-dtype",
969 "bf16",
970 "--fp-values-range",
971 "-2.0,2.0",
972 "--tensor-dim-range",
973 "1,16",
974 "--target-rank",
975 "4",
976 "--target-rank",
977 "5"
978 ],
979 [
980 "--target-dtype",
981 "fp32",
982 "--fp-values-range",
983 "-2.0,2.0",
984 "--target-shape",
985 "65531,1,2,1",
986 "--target-shape",
987 "1,1,65541,1"
988 ]
989 ],
990 "params": {},
991 "permutes": [
992 "shape",
993 "type"
994 ],
995 "profile": [
996 "tosa-mi"
997 ]
998 },
999 "minimum": {
1000 "group": "ew_binary",
1001 "generator_args": [
1002 [
1003 "--target-dtype",
1004 "fp32",
1005 "--target-dtype",
1006 "fp16",
1007 "--target-dtype",
1008 "bf16",
1009 "--fp-values-range",
1010 "-2.0,2.0",
1011 "--tensor-dim-range",
1012 "16,64",
1013 "--target-rank",
1014 "1",
1015 "--target-rank",
1016 "2",
1017 "--target-rank",
1018 "3"
1019 ],
1020 [
1021 "--target-dtype",
1022 "fp32",
1023 "--target-dtype",
1024 "fp16",
1025 "--target-dtype",
1026 "bf16",
1027 "--fp-values-range",
1028 "-2.0,2.0",
1029 "--tensor-dim-range",
1030 "1,16",
1031 "--target-rank",
1032 "4",
1033 "--target-rank",
1034 "5"
1035 ],
1036 [
1037 "--target-dtype",
1038 "fp32",
1039 "--fp-values-range",
1040 "-2.0,2.0",
1041 "--target-shape",
1042 "1,65531,2,1",
1043 "--target-shape",
1044 "1,1,1,65538"
1045 ]
1046 ],
1047 "params": {},
1048 "permutes": [
1049 "shape",
1050 "type"
1051 ],
1052 "profile": [
1053 "tosa-mi"
1054 ]
James Ward635bc992022-11-23 11:55:32 +0000[diff] [blame]1055 },
1056 "reduce_max": {
1057 "group": "reduction",
1058 "generator_args": [
1059 [
1060 "--target-dtype",
1061 "fp32",
1062 "--target-dtype",
1063 "fp16",
1064 "--target-dtype",
1065 "bf16",
1066 "--fp-values-range",
1067 "-2.0,2.0",
1068 "--tensor-dim-range",
1069 "1,32"
1070 ],
1071 [
1072 "--target-dtype",
1073 "fp32",
1074 "--fp-values-range",
1075 "-2.0,2.0",
1076 "--target-shape",
1077 "1,1,1,65531",
1078 "--target-shape",
1079 "1,1,65539,2"
1080 ]
1081 ],
1082 "params": {},
1083 "permutes": [
1084 "shape",
1085 "type",
1086 "axis"
1087 ],
1088 "profile": [
1089 "tosa-mi"
1090 ]
1091 },
1092 "reduce_min": {
1093 "group": "reduction",
1094 "generator_args": [
1095 [
1096 "--target-dtype",
1097 "fp32",
1098 "--target-dtype",
1099 "fp16",
1100 "--target-dtype",
1101 "bf16",
1102 "--fp-values-range",
1103 "-2.0,2.0",
1104 "--tensor-dim-range",
1105 "1,32"
1106 ],
1107 [
1108 "--target-dtype",
1109 "fp32",
1110 "--fp-values-range",
1111 "-2.0,2.0",
1112 "--target-shape",
1113 "1,1,65531,1",
1114 "--target-shape",
1115 "1,65539,2,1"
1116 ]
1117 ],
1118 "params": {},
1119 "permutes": [
1120 "shape",
1121 "type",
1122 "axis"
1123 ],
1124 "profile": [
1125 "tosa-mi"
1126 ]
James Ward2a27c8f2022-11-23 11:29:43 +0000[diff] [blame]1127 },
James Ward7f1ea8e2022-12-12 18:39:11 +0000[diff] [blame]1128 "resize": {
1129 "group": "image",
1130 "generator_args": [
1131 [
1132 "--target-dtype",
1133 "fp32",
1134 "--target-dtype",
1135 "fp16",
1136 "--target-dtype",
1137 "bf16",
1138 "--fp-values-range",
1139 "-2.0,2.0"
1140 ],
1141 [
1142 "--target-dtype",
1143 "fp32",
1144 "--fp-values-range",
1145 "-2.0,2.0",
1146 "--target-shape",
1147 "1,1103,1,2",
1148 "--max-resize-output-dim",
1149 "9100"
1150 ]
1151 ],
1152 "params": {
1153 "offset": [],
1154 "shape": [],
1155 "type": [],
1156 "mode": [],
1157 "output_type": [],
1158 "scale": ["sc2x1x3x1"],
1159 "border": ["bor0x1"]
1160 },
1161 "permutes": [
1162 "shape",
1163 "type"
1164 ],
1165 "profile": [
1166 "tosa-mi"
1167 ]
1168 },
James Ward2a27c8f2022-11-23 11:29:43 +0000[diff] [blame]1169 "select": {
1170 "group": "ew_ternary",
1171 "generator_args": [
1172 [
1173 "--target-dtype",
1174 "fp32",
1175 "--target-dtype",
1176 "fp16",
1177 "--target-dtype",
1178 "bf16",
1179 "--fp-values-range",
1180 "-2.0,2.0",
1181 "--tensor-dim-range",
1182 "16,64",
1183 "--target-rank",
1184 "1",
1185 "--target-rank",
1186 "2",
1187 "--target-rank",
1188 "3"
1189 ],
1190 [
1191 "--target-dtype",
1192 "fp32",
1193 "--target-dtype",
1194 "fp16",
1195 "--target-dtype",
1196 "bf16",
1197 "--fp-values-range",
1198 "-2.0,2.0",
1199 "--tensor-dim-range",
1200 "1,16",
1201 "--target-rank",
1202 "4",
1203 "--target-rank",
1204 "5"
1205 ],
1206 [
1207 "--target-dtype",
1208 "fp32",
1209 "--fp-values-range",
1210 "-2.0,2.0",
1211 "--target-shape",
1212 "1,2,65534,2,1",
1213 "--target-shape",
1214 "1,1,65539,2",
1215 "--num-rand-permutations",
1216 "2"
1217 ]
1218 ],
1219 "params": {},
1220 "permutes": [
1221 "shape",
1222 "type"
1223 ],
1224 "profile": [
1225 "tosa-mi"
1226 ]
James Wardb45db9a2022-12-12 13:02:44 +0000[diff] [blame]1227 },
1228 "sigmoid": {
1229 "group": "activation",
1230 "generator_args": [
1231 [
1232 "--target-dtype",
1233 "fp32",
1234 "--target-dtype",
1235 "fp16",
1236 "--target-dtype",
1237 "bf16",
1238 "--fp-values-range",
1239 "-2.0,2.0",
1240 "--tensor-dim-range",
1241 "16,64",
1242 "--target-rank",
1243 "1",
1244 "--target-rank",
1245 "2",
1246 "--target-rank",
1247 "3"
1248 ],
1249 [
1250 "--target-dtype",
1251 "fp32",
1252 "--target-dtype",
1253 "fp16",
1254 "--target-dtype",
1255 "bf16",
1256 "--fp-values-range",
1257 "-2.0,2.0",
1258 "--tensor-dim-range",
1259 "1,19",
1260 "--target-rank",
1261 "4",
1262 "--target-rank",
1263 "5"
1264 ],
1265 [
1266 "--target-dtype",
1267 "fp32",
1268 "--fp-values-range",
1269 "-2.0,2.0",
1270 "--target-shape",
1271 "1,2,65538,2,1",
1272 "--target-shape",
1273 "1,3,1,65533"
1274 ]
1275 ],
1276 "params": {},
1277 "permutes": [
1278 "shape",
1279 "type"
1280 ],
1281 "profile": [
1282 "tosa-mi"
1283 ]
1284 },
1285 "tanh": {
1286 "group": "activation",
1287 "generator_args": [
1288 [
1289 "--target-dtype",
1290 "fp32",
1291 "--target-dtype",
1292 "fp16",
1293 "--target-dtype",
1294 "bf16",
1295 "--fp-values-range",
1296 "-2.0,2.0",
1297 "--tensor-dim-range",
1298 "18,60",
1299 "--target-rank",
1300 "1",
1301 "--target-rank",
1302 "2",
1303 "--target-rank",
1304 "3"
1305 ],
1306 [
1307 "--target-dtype",
1308 "fp32",
1309 "--target-dtype",
1310 "fp16",
1311 "--target-dtype",
1312 "bf16",
1313 "--fp-values-range",
1314 "-2.0,2.0",
1315 "--tensor-dim-range",
1316 "1,24",
1317 "--target-rank",
1318 "4",
1319 "--target-rank",
1320 "5"
1321 ],
1322 [
1323 "--target-dtype",
1324 "fp32",
1325 "--fp-values-range",
1326 "-2.0,2.0",
1327 "--target-shape",
1328 "1,65535,2,1,1",
1329 "--target-shape",
1330 "1,65540,1,2"
1331 ]
1332 ],
1333 "params": {},
1334 "permutes": [
1335 "shape",
1336 "type"
1337 ],
1338 "profile": [
1339 "tosa-mi"
1340 ]
Jeremy Johnsone4b08ff2022-09-15 10:38:17 +0100[diff] [blame]1341 }
1342}