Gitiles
Code Review Sign In
review.mlplatform.org / tosa / reference_model / 3d6de004bfa6469a2f90eb9c8c5856095f96467d / . / verif / conformance / tosa_base_profile_ops_info.json
blob: 2af4c474de6a8eaf74fd0ca49acc98449f4f7efd [file] [log] [blame]
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1{
2 "abs": {
3 "group": "ew_unary",
4 "generator_args": [
5 [
6 "--target-dtype",
7 "int32",
8 "--tensor-dim-range",
9 "16,64",
10 "--target-rank",
11 "1",
12 "--target-rank",
13 "2",
14 "--target-rank",
15 "3"
16 ],
17 [
18 "--target-dtype",
19 "int32",
20 "--tensor-dim-range",
21 "1,16",
22 "--target-rank",
23 "4",
24 "--target-rank",
25 "5"
26 ],
27 [
28 "--target-shape",
29 "1,2,65533,1",
30 "--target-shape",
31 "1,65539,1,1,1",
32 "--target-dtype",
33 "int32"
34 ]
35 ],
36 "params": {},
37 "permutes": [
38 "shape",
39 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]40 ],
41 "profile": [
42 "tosa-bi",
43 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]44 ]
45 },
46 "add": {
47 "group": "ew_binary",
48 "generator_args": [
49 [
50 "--target-dtype",
51 "int32",
52 "--tensor-dim-range",
53 "16,64",
54 "--target-rank",
55 "1",
56 "--target-rank",
57 "2",
58 "--target-rank",
59 "3"
60 ],
61 [
62 "--target-dtype",
63 "int32",
64 "--tensor-dim-range",
65 "1,16",
66 "--target-rank",
67 "4",
68 "--target-rank",
69 "5"
70 ],
71 [
72 "--target-shape",
73 "1,2,65533,1",
74 "--target-shape",
75 "1,65539,1,1,1",
76 "--target-dtype",
77 "int32"
78 ]
79 ],
80 "params": {},
81 "permutes": [
82 "shape",
83 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]84 ],
85 "profile": [
86 "tosa-bi",
87 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]88 ]
89 },
90 "argmax": {
91 "group": "tensor",
92 "generator_args": [
93 [
94 "--target-dtype",
95 "int8",
96 "--target-dtype",
97 "int16",
98 "--tensor-dim-range",
99 "32,64",
100 "--target-rank",
101 "1",
102 "--target-rank",
103 "2",
104 "--target-rank",
105 "3",
106 "--num-rand-permutations",
107 "2"
108 ],
109 [
110 "--target-dtype",
111 "int8",
112 "--target-dtype",
113 "int16",
114 "--tensor-dim-range",
115 "1,32",
116 "--target-rank",
117 "4",
118 "--num-rand-permutations",
119 "2"
120 ],
121 [
122 "--target-shape",
123 "1,3,1,65535",
124 "--target-shape",
125 "1,65538,1,2",
126 "--target-dtype",
127 "int8",
128 "--num-rand-permutations",
129 "2"
130 ]
131 ],
132 "params": {},
133 "permutes": [
134 "shape",
135 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]136 ],
137 "profile": [
138 "tosa-bi",
139 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]140 ]
141 },
142 "arithmetic_right_shift": {
143 "group": "ew_binary",
144 "generator_args": [
145 [
146 "--tensor-dim-range",
147 "16,64",
148 "--target-rank",
149 "1",
150 "--target-rank",
151 "2",
152 "--target-rank",
153 "3"
154 ],
155 [
156 "--tensor-dim-range",
157 "1,16",
158 "--target-rank",
159 "4",
160 "--target-rank",
161 "5"
162 ],
163 [
164 "--target-shape",
165 "1,2,65533,1",
166 "--target-shape",
167 "1,65539,1,1,1",
168 "--target-dtype",
169 "int8"
170 ]
171 ],
172 "params": {},
173 "permutes": [
174 "shape",
175 "type",
176 "rounding"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]177 ],
178 "profile": [
179 "tosa-bi",
180 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]181 ]
182 },
183 "avg_pool2d": {
184 "group": "tensor",
185 "generator_args": [
186 [
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-shape",
195 "1,65535,5,1",
196 "--target-shape",
197 "1,3,65537,1",
198 "--target-dtype",
199 "int8",
200 "--allow-pooling-and-conv-oversizes"
201 ]
202 ],
203 "params": {
204 "shape": [],
205 "type": [
206 "i8",
207 "i16"
208 ],
209 "kernel": [],
210 "stride": [],
211 "pad": []
212 },
213 "permutes": [
214 "shape",
215 "type",
216 "pad"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]217 ],
218 "profile": [
219 "tosa-bi",
220 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]221 ]
222 },
223 "bitwise_and": {
224 "group": "ew_binary",
225 "generator_args": [
226 [
227 "--tensor-dim-range",
228 "1,64",
229 "--target-rank",
230 "1",
231 "--target-rank",
232 "2",
233 "--target-rank",
234 "3"
235 ],
236 [
237 "--target-dtype",
238 "int8",
239 "--tensor-dim-range",
240 "1,16",
241 "--target-rank",
242 "4",
243 "--target-rank",
244 "5"
245 ],
246 [
247 "--target-shape",
248 "1,2,65536,1",
249 "--target-shape",
250 "1,65539,1,1,1",
251 "--target-dtype",
252 "int8"
253 ]
254 ],
255 "params": {},
256 "permutes": [
257 "shape",
258 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]259 ],
260 "profile": [
261 "tosa-bi",
262 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]263 ]
264 },
265 "bitwise_not": {
266 "group": "ew_unary",
267 "generator_args": [
268 [
269 "--tensor-dim-range",
270 "2,64",
271 "--target-rank",
272 "1",
273 "--target-rank",
274 "2",
275 "--target-rank",
276 "3"
277 ],
278 [
279 "--target-dtype",
280 "int8",
281 "--tensor-dim-range",
282 "1,16",
283 "--target-rank",
284 "4",
285 "--target-rank",
286 "5"
287 ],
288 [
289 "--target-shape",
290 "1,2,65537,1",
291 "--target-shape",
292 "1,65538,1,1,1",
293 "--target-dtype",
294 "int8"
295 ]
296 ],
297 "params": {},
298 "permutes": [
299 "shape",
300 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]301 ],
302 "profile": [
303 "tosa-bi",
304 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]305 ]
306 },
307 "bitwise_or": {
308 "group": "ew_binary",
309 "generator_args": [
310 [
311 "--tensor-dim-range",
312 "3,64",
313 "--target-rank",
314 "1",
315 "--target-rank",
316 "2",
317 "--target-rank",
318 "3"
319 ],
320 [
321 "--target-dtype",
322 "int8",
323 "--tensor-dim-range",
324 "1,16",
325 "--target-rank",
326 "4",
327 "--target-rank",
328 "5"
329 ],
330 [
331 "--target-shape",
332 "1,2,65538,1",
333 "--target-shape",
334 "1,65537,1,1,1",
335 "--target-dtype",
336 "int8"
337 ]
338 ],
339 "params": {},
340 "permutes": [
341 "shape",
342 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]343 ],
344 "profile": [
345 "tosa-bi",
346 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]347 ]
348 },
349 "bitwise_xor": {
350 "group": "ew_binary",
351 "generator_args": [
352 [
353 "--tensor-dim-range",
354 "4,64",
355 "--target-rank",
356 "1",
357 "--target-rank",
358 "2",
359 "--target-rank",
360 "3"
361 ],
362 [
363 "--target-dtype",
364 "int8",
365 "--tensor-dim-range",
366 "1,16",
367 "--target-rank",
368 "4",
369 "--target-rank",
370 "5"
371 ],
372 [
373 "--target-shape",
374 "1,2,65536,1",
375 "--target-shape",
376 "1,65536,1,1,1",
377 "--target-dtype",
378 "int8"
379 ]
380 ],
381 "params": {},
382 "permutes": [
383 "shape",
384 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]385 ],
386 "profile": [
387 "tosa-bi",
388 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]389 ]
390 },
391 "cast": {
392 "group": "type_conversion",
393 "generator_args": [
394 [
395 "--target-dtype",
396 "int8",
397 "--target-dtype",
398 "int16",
399 "--target-dtype",
400 "int32",
401 "--target-dtype",
402 "bool",
403 "--tensor-dim-range",
404 "16,64",
405 "--target-rank",
406 "1",
407 "--target-rank",
408 "2",
409 "--target-rank",
410 "3"
411 ],
412 [
413 "--target-dtype",
414 "int8",
415 "--target-dtype",
416 "int16",
417 "--target-dtype",
418 "int32",
419 "--target-dtype",
420 "bool",
421 "--tensor-dim-range",
422 "1,16",
423 "--target-rank",
424 "4",
425 "--target-rank",
426 "5"
427 ],
428 [
429 "--target-shape",
430 "1,1,65535,1,1",
431 "--target-shape",
432 "1,65538,1,2",
433 "--target-dtype",
434 "int8"
435 ]
436 ],
437 "params": {
438 "shape": [],
439 "type": [],
440 "output_type": [
441 "outINT8",
442 "outINT16",
443 "outINT32",
444 "outBOOL"
445 ]
446 },
447 "permutes": [
448 "shape",
449 "type",
450 "output_type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]451 ],
452 "profile": [
453 "tosa-bi",
454 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]455 ]
456 },
457 "clamp": {
458 "group": "activation",
459 "generator_args": [
460 [
461 "--tensor-dim-range",
462 "20,64",
463 "--target-rank",
464 "1",
465 "--target-rank",
466 "2",
467 "--target-rank",
468 "3"
469 ],
470 [
471 "--tensor-dim-range",
472 "1,20",
473 "--target-rank",
474 "4",
475 "--target-rank",
476 "5"
477 ],
478 [
479 "--target-shape",
480 "1,2,1,65536",
481 "--target-shape",
482 "1,3,65537,1,1",
483 "--target-dtype",
484 "int8"
485 ]
486 ],
487 "params": {},
488 "permutes": [
489 "shape",
490 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]491 ],
492 "profile": [
493 "tosa-bi",
494 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]495 ]
496 },
497 "clz": {
498 "group": "ew_unary",
499 "generator_args": [
500 [
501 "--target-dtype",
502 "int32",
503 "--tensor-dim-range",
504 "16,64",
505 "--target-rank",
506 "1",
507 "--target-rank",
508 "2",
509 "--target-rank",
510 "3"
511 ],
512 [
513 "--target-dtype",
514 "int32",
515 "--tensor-dim-range",
516 "1,16",
517 "--target-rank",
518 "4",
519 "--target-rank",
520 "5"
521 ],
522 [
523 "--target-shape",
524 "1,2,65533,1",
525 "--target-shape",
526 "1,65539,1,1,1",
527 "--target-dtype",
528 "int32"
529 ]
530 ],
531 "params": {},
532 "permutes": [
533 "shape",
534 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]535 ],
536 "profile": [
537 "tosa-bi",
538 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]539 ]
540 },
541 "concat": {
542 "group": "data_layout",
543 "generator_args": [
544 [
545 "--target-dtype",
546 "int8",
547 "--target-dtype",
548 "int16",
549 "--target-dtype",
550 "int32",
551 "--target-dtype",
552 "bool",
553 "--tensor-dim-range",
554 "16,64",
555 "--target-rank",
556 "1",
557 "--target-rank",
558 "2",
559 "--target-rank",
560 "3",
561 "--num-const-inputs-concat",
562 "3"
563 ],
564 [
565 "--target-dtype",
566 "int8",
567 "--target-dtype",
568 "int16",
569 "--target-dtype",
570 "int32",
571 "--target-dtype",
572 "bool",
573 "--tensor-dim-range",
574 "1,16",
575 "--target-rank",
576 "4",
577 "--target-rank",
578 "5",
579 "--num-const-inputs-concat",
580 "1"
581 ],
582 [
583 "--target-shape",
584 "1,1,65535,1,1",
585 "--target-shape",
586 "1,65538,1,2",
587 "--target-dtype",
588 "int8",
589 "--num-const-inputs-concat",
590 "2"
591 ]
592 ],
593 "params": {},
594 "permutes": [
595 "shape",
596 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]597 ],
598 "profile": [
599 "tosa-bi",
600 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]601 ]
602 },
603 "cond_if": {
604 "group": "control_flow",
605 "generator_args": [
606 [
607 "--target-dtype",
608 "bool",
609 "--target-dtype",
610 "int8",
611 "--target-dtype",
612 "int16",
613 "--target-dtype",
614 "int32"
615 ],
616 [
617 "--target-shape",
618 "3,22,17,1,2",
619 "--target-dtype",
620 "int32"
621 ]
622 ],
623 "params": {},
624 "permutes": [
625 "variant",
626 "shape",
627 "cond"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]628 ],
629 "profile": [
630 "tosa-bi",
631 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]632 ]
633 },
634 "const": {
635 "group": "data_nodes",
636 "no_negative_tests": "true",
637 "generator_args": [
638 [
639 "--target-dtype",
640 "int8",
641 "--target-dtype",
642 "int16",
643 "--target-dtype",
644 "int32",
645 "--target-dtype",
646 "bool",
647 "--target-rank",
648 "1",
649 "--target-rank",
650 "2",
651 "--target-rank",
652 "3"
653 ],
654 [
655 "--target-dtype",
656 "int8",
657 "--target-dtype",
658 "int16",
659 "--target-dtype",
660 "int32",
661 "--target-dtype",
662 "bool",
663 "--tensor-dim-range",
664 "1,16",
665 "--target-rank",
666 "4",
667 "--target-rank",
668 "5"
669 ],
670 [
671 "--target-shape",
672 "1,1,65537,1,1",
673 "--target-shape",
674 "1,65530,1,2",
675 "--target-dtype",
676 "int8"
677 ]
678 ],
679 "params": {},
680 "permutes": [
681 "shape",
682 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]683 ],
684 "profile": [
685 "tosa-bi",
686 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]687 ]
688 },
689 "conv2d": {
690 "group": "tensor",
691 "generator_args": [
692 [
693 "--target-shape",
694 "1,49,42,28",
695 "--target-shape",
696 "1,11,44,13",
697 "--allow-pooling-and-conv-oversizes"
698 ],
699 [
700 "--target-shape",
701 "1,65535,4,1",
702 "--target-shape",
703 "1,5,65536,1",
704 "--target-dtype",
705 "int8",
706 "--max-conv-dilation",
707 "1",
708 "--allow-pooling-and-conv-oversizes"
709 ]
710 ],
711 "params": {
712 "shape": [],
713 "type": [
714 "i8xi8",
715 "i16xi8",
716 "i8xi4"
717 ],
718 "kernel": [],
719 "stride": [],
720 "pad": [],
721 "dilation": []
722 },
723 "permutes": [
724 "kernel",
725 "shape",
726 "type",
727 "pad"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]728 ],
729 "profile": [
730 "tosa-bi",
731 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]732 ]
733 },
734 "conv3d": {
735 "group": "tensor",
736 "generator_args": [
737 [
738 "--target-shape",
739 "1,9,21,14,1",
740 "--target-shape",
741 "1,9,11,12,3",
742 "--target-dtype",
743 "int8",
744 "--target-dtype",
745 "int16",
746 "--allow-pooling-and-conv-oversizes"
747 ],
748 [
749 "--target-shape",
750 "1,1,65535,3,1",
751 "--target-shape",
752 "1,1,3,65536,1",
753 "--target-dtype",
754 "int8",
755 "--max-conv-dilation",
756 "1",
757 "--allow-pooling-and-conv-oversizes"
758 ]
759 ],
760 "params": {
761 "shape": [],
762 "type": [
763 "i8xi8",
764 "i16xi8",
765 "i8xi4"
766 ],
767 "kernel": [],
768 "stride": [],
769 "pad": [],
770 "dilation": []
771 },
772 "permutes": [
773 "kernel",
774 "shape",
775 "type",
776 "pad",
777 "stride"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]778 ],
779 "profile": [
780 "tosa-bi",
781 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]782 ]
783 },
784 "depthwise_conv2d": {
785 "group": "tensor",
786 "generator_args": [
787 [
788 "--target-shape",
789 "1,9,44,3",
790 "--target-shape",
791 "1,11,33,4",
792 "--allow-pooling-and-conv-oversizes"
793 ],
794 [
795 "--target-shape",
796 "1,65532,2,1",
797 "--target-shape",
798 "1,4,65537,1",
799 "--target-dtype",
800 "int8",
801 "--max-conv-dilation",
802 "1",
803 "--allow-pooling-and-conv-oversizes"
804 ]
805 ],
806 "params": {
807 "shape": [],
808 "type": [
809 "i8xi8",
810 "i16xi8",
811 "i8xi4"
812 ],
813 "kernel": [],
814 "stride": [],
815 "pad": [],
816 "dilation": []
817 },
818 "permutes": [
819 "kernel",
820 "shape",
821 "type",
822 "pad"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]823 ],
824 "profile": [
825 "tosa-bi",
826 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]827 ]
828 },
829 "equal": {
830 "group": "comparison",
831 "generator_args": [
832 [
833 "--target-dtype",
834 "int32",
835 "--tensor-dim-range",
836 "16,64",
837 "--target-rank",
838 "1",
839 "--target-rank",
840 "2",
841 "--target-rank",
842 "3"
843 ],
844 [
845 "--target-dtype",
846 "int32",
847 "--tensor-dim-range",
848 "1,16",
849 "--target-rank",
850 "4",
851 "--target-rank",
852 "5"
853 ],
854 [
855 "--target-shape",
856 "1,2,65501,1",
857 "--target-shape",
858 "1,65541,1,1,1",
859 "--target-dtype",
860 "int32"
861 ]
862 ],
863 "params": {},
864 "permutes": [
865 "shape",
866 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]867 ],
868 "profile": [
869 "tosa-bi",
870 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]871 ]
872 },
873 "fully_connected": {
874 "group": "tensor",
875 "generator_args": [
876 [
877 "--target-dtype",
878 "int8",
879 "--target-dtype",
880 "int16"
881 ],
882 [
883 "--target-shape",
884 "444,1",
885 "--target-shape",
886 "3,65538",
887 "--tensor-dim-range",
888 "1,2",
889 "--target-dtype",
890 "int8"
891 ],
892 [
893 "--target-shape",
894 "3,16",
895 "--target-shape",
896 "1,23",
897 "--tensor-dim-range",
898 "100,200",
899 "--target-dtype",
900 "int8"
901 ]
902 ],
903 "params": {},
904 "permutes": [
905 "shape",
906 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]907 ],
908 "profile": [
909 "tosa-bi",
910 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]911 ]
912 },
913 "gather": {
914 "group": "scatter_gather",
915 "generator_args": [
916 [
917 "--target-dtype",
918 "int16",
919 "--target-dtype",
920 "int32",
921 "--tensor-dim-range",
922 "4,64"
923 ],
924 [
925 "--target-shape",
926 "2,65533,1",
927 "--target-shape",
928 "1,2,65533",
929 "--target-shape",
930 "5000,1,1",
931 "--target-dtype",
932 "int8"
933 ]
934 ],
935 "params": {},
936 "permutes": [
937 "shape",
938 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]939 ],
940 "profile": [
941 "tosa-bi",
942 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]943 ]
944 },
945 "greater": {
946 "group": "comparison",
947 "generator_args": [
948 [
949 "--filter",
950 "^greater$",
951 "--target-dtype",
952 "int32",
953 "--tensor-dim-range",
954 "10,64",
955 "--target-rank",
956 "1",
957 "--target-rank",
958 "2",
959 "--target-rank",
960 "3"
961 ],
962 [
963 "--filter",
964 "^greater$",
965 "--target-dtype",
966 "int32",
967 "--tensor-dim-range",
968 "1,15",
969 "--target-rank",
970 "4",
971 "--target-rank",
972 "5"
973 ],
974 [
975 "--filter",
976 "^greater$",
977 "--target-shape",
978 "1,2,65530,1",
979 "--target-shape",
980 "1,65539,1,2,1",
981 "--target-dtype",
982 "int32"
983 ]
984 ],
985 "params": {},
986 "permutes": [
987 "shape",
988 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]989 ],
990 "profile": [
991 "tosa-bi",
992 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]993 ]
994 },
995 "greater_equal": {
996 "group": "comparison",
997 "generator_args": [
998 [
999 "--target-dtype",
1000 "int32",
1001 "--tensor-dim-range",
1002 "20,64",
1003 "--target-rank",
1004 "1",
1005 "--target-rank",
1006 "2",
1007 "--target-rank",
1008 "3"
1009 ],
1010 [
1011 "--target-dtype",
1012 "int32",
1013 "--tensor-dim-range",
1014 "1,20",
1015 "--target-rank",
1016 "4",
1017 "--target-rank",
1018 "5"
1019 ],
1020 [
1021 "--target-shape",
1022 "1,2,65533,1",
1023 "--target-shape",
1024 "1,65539,1,3,1",
1025 "--target-dtype",
1026 "int32"
1027 ]
1028 ],
1029 "params": {},
1030 "permutes": [
1031 "shape",
1032 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1033 ],
1034 "profile": [
1035 "tosa-bi",
1036 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1037 ]
1038 },
1039 "identity": {
1040 "group": "data_nodes",
1041 "no_negative_tests": "true",
1042 "generator_args": [
1043 [
1044 "--target-dtype",
1045 "int8",
1046 "--target-dtype",
1047 "int16",
1048 "--target-dtype",
1049 "int32",
1050 "--target-dtype",
1051 "bool",
1052 "--tensor-dim-range",
1053 "1,61",
1054 "--target-rank",
1055 "1",
1056 "--target-rank",
1057 "2",
1058 "--target-rank",
1059 "3"
1060 ],
1061 [
1062 "--target-dtype",
1063 "int8",
1064 "--target-dtype",
1065 "int16",
1066 "--target-dtype",
1067 "int32",
1068 "--target-dtype",
1069 "bool",
1070 "--tensor-dim-range",
1071 "1,15",
1072 "--target-rank",
1073 "4",
1074 "--target-rank",
1075 "5"
1076 ],
1077 [
1078 "--target-shape",
1079 "1,1,65536,1,1",
1080 "--target-shape",
1081 "1,65531,1,2",
1082 "--target-dtype",
1083 "int8"
1084 ]
1085 ],
1086 "params": {},
1087 "permutes": [
1088 "shape",
1089 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1090 ],
1091 "profile": [
1092 "tosa-bi",
1093 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1094 ]
1095 },
1096 "intdiv": {
1097 "group": "ew_binary",
1098 "generator_args": [
1099 [
1100 "--target-dtype",
1101 "int32",
1102 "--tensor-dim-range",
1103 "16,64",
1104 "--target-rank",
1105 "1",
1106 "--target-rank",
1107 "2",
1108 "--target-rank",
1109 "3"
1110 ],
1111 [
1112 "--target-dtype",
1113 "int32",
1114 "--tensor-dim-range",
1115 "1,16",
1116 "--target-rank",
1117 "4",
1118 "--target-rank",
1119 "5"
1120 ],
1121 [
1122 "--target-shape",
1123 "1,65533,1",
1124 "--target-shape",
1125 "1,1,65539,1,1",
1126 "--target-dtype",
1127 "int32"
1128 ]
1129 ],
1130 "params": {},
1131 "permutes": [
1132 "shape",
1133 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1134 ],
1135 "profile": [
1136 "tosa-bi",
1137 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1138 ]
1139 },
1140 "logical_and": {
1141 "group": "ew_binary",
1142 "generator_args": [
1143 [
1144 "--tensor-dim-range",
1145 "1,64",
1146 "--target-rank",
1147 "1",
1148 "--target-rank",
1149 "2",
1150 "--target-rank",
1151 "3"
1152 ],
1153 [
1154 "--tensor-dim-range",
1155 "1,16",
1156 "--target-rank",
1157 "4",
1158 "--target-rank",
1159 "5"
1160 ],
1161 [
1162 "--target-shape",
1163 "1,65538,1",
1164 "--target-shape",
1165 "1,2,1,65536"
1166 ]
1167 ],
1168 "params": {},
1169 "permutes": [
1170 "shape",
1171 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1172 ],
1173 "profile": [
1174 "tosa-bi",
1175 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1176 ]
1177 },
1178 "logical_left_shift": {
1179 "group": "ew_binary",
1180 "generator_args": [
1181 [
1182 "--target-dtype",
1183 "int8",
1184 "--target-dtype",
1185 "int16",
1186 "--target-dtype",
1187 "int32",
1188 "--tensor-dim-range",
1189 "16,64",
1190 "--target-rank",
1191 "1",
1192 "--target-rank",
1193 "2",
1194 "--target-rank",
1195 "3"
1196 ],
1197 [
1198 "--target-dtype",
1199 "int8",
1200 "--target-dtype",
1201 "int16",
1202 "--target-dtype",
1203 "int32",
1204 "--tensor-dim-range",
1205 "1,16",
1206 "--target-rank",
1207 "4",
1208 "--target-rank",
1209 "5"
1210 ],
1211 [
1212 "--target-shape",
1213 "1,65532,1,1",
1214 "--target-shape",
1215 "1,2,1,65538",
1216 "--target-dtype",
1217 "int8"
1218 ]
1219 ],
1220 "params": {},
1221 "permutes": [
1222 "shape",
1223 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1224 ],
1225 "profile": [
1226 "tosa-bi",
1227 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1228 ]
1229 },
1230 "logical_not": {
1231 "group": "ew_unary",
1232 "generator_args": [
1233 [
1234 "--tensor-dim-range",
1235 "2,64",
1236 "--target-rank",
1237 "1",
1238 "--target-rank",
1239 "2",
1240 "--target-rank",
1241 "3"
1242 ],
1243 [
1244 "--tensor-dim-range",
1245 "1,15",
1246 "--target-rank",
1247 "4",
1248 "--target-rank",
1249 "5"
1250 ],
1251 [
1252 "--target-shape",
1253 "1,65537,1",
1254 "--target-shape",
1255 "1,2,1,65535"
1256 ]
1257 ],
1258 "params": {},
1259 "permutes": [
1260 "shape",
1261 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1262 ],
1263 "profile": [
1264 "tosa-bi",
1265 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1266 ]
1267 },
1268 "logical_or": {
1269 "group": "ew_binary",
1270 "generator_args": [
1271 [
1272 "--tensor-dim-range",
1273 "3,64",
1274 "--target-rank",
1275 "1",
1276 "--target-rank",
1277 "2",
1278 "--target-rank",
1279 "3"
1280 ],
1281 [
1282 "--tensor-dim-range",
1283 "1,14",
1284 "--target-rank",
1285 "4",
1286 "--target-rank",
1287 "5"
1288 ],
1289 [
1290 "--target-shape",
1291 "1,65536,1",
1292 "--target-shape",
1293 "1,2,1,65537"
1294 ]
1295 ],
1296 "params": {},
1297 "permutes": [
1298 "shape",
1299 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1300 ],
1301 "profile": [
1302 "tosa-bi",
1303 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1304 ]
1305 },
1306 "logical_right_shift": {
1307 "group": "ew_binary",
1308 "generator_args": [
1309 [
1310 "--target-dtype",
1311 "int8",
1312 "--target-dtype",
1313 "int16",
1314 "--target-dtype",
1315 "int32",
1316 "--tensor-dim-range",
1317 "25,64",
1318 "--target-rank",
1319 "1",
1320 "--target-rank",
1321 "2",
1322 "--target-rank",
1323 "3"
1324 ],
1325 [
1326 "--target-dtype",
1327 "int8",
1328 "--target-dtype",
1329 "int16",
1330 "--target-dtype",
1331 "int32",
1332 "--tensor-dim-range",
1333 "1,25",
1334 "--target-rank",
1335 "4",
1336 "--target-rank",
1337 "5"
1338 ],
1339 [
1340 "--target-shape",
1341 "1,1,65540,3",
1342 "--target-shape",
1343 "2,1,65530,1,1",
1344 "--target-dtype",
1345 "int8"
1346 ]
1347 ],
1348 "params": {},
1349 "permutes": [
1350 "shape",
1351 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1352 ],
1353 "profile": [
1354 "tosa-bi",
1355 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1356 ]
1357 },
1358 "logical_xor": {
1359 "group": "ew_binary",
1360 "generator_args": [
1361 [
1362 "--tensor-dim-range",
1363 "4,64",
1364 "--target-rank",
1365 "1",
1366 "--target-rank",
1367 "2",
1368 "--target-rank",
1369 "3"
1370 ],
1371 [
1372 "--tensor-dim-range",
1373 "3,16",
1374 "--target-rank",
1375 "4",
1376 "--target-rank",
1377 "5"
1378 ],
1379 [
1380 "--target-shape",
1381 "1,65539,1",
1382 "--target-shape",
1383 "1,2,1,65534"
1384 ]
1385 ],
1386 "params": {},
1387 "permutes": [
1388 "shape",
1389 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1390 ],
1391 "profile": [
1392 "tosa-bi",
1393 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1394 ]
1395 },
1396 "matmul": {
1397 "group": "tensor",
1398 "generator_args": [
1399 [
1400 "--target-dtype",
1401 "int8",
1402 "--target-dtype",
1403 "int16",
1404 "--tensor-dim-range",
1405 "16,64"
1406 ],
1407 [
1408 "--target-dtype",
1409 "int8",
1410 "--target-dtype",
1411 "int16",
1412 "--tensor-dim-range",
1413 "1,16"
1414 ],
1415 [
1416 "--target-shape",
1417 "1,65532,1",
1418 "--target-shape",
1419 "1,2,65538",
1420 "--target-shape",
1421 "2000,1,1",
1422 "--target-dtype",
1423 "int8"
1424 ]
1425 ],
1426 "params": {},
1427 "permutes": [
1428 "shape",
1429 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1430 ],
1431 "profile": [
1432 "tosa-bi",
1433 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1434 ]
1435 },
1436 "maximum": {
1437 "group": "ew_binary",
1438 "generator_args": [
1439 [
1440 "--target-dtype",
1441 "int32",
1442 "--tensor-dim-range",
1443 "16,64",
1444 "--target-rank",
1445 "1",
1446 "--target-rank",
1447 "2",
1448 "--target-rank",
1449 "3"
1450 ],
1451 [
1452 "--target-dtype",
1453 "int32",
1454 "--tensor-dim-range",
1455 "1,16",
1456 "--target-rank",
1457 "4",
1458 "--target-rank",
1459 "5"
1460 ],
1461 [
1462 "--target-shape",
1463 "1,2,65533,1",
1464 "--target-shape",
1465 "1,65539,1,1,1",
1466 "--target-dtype",
1467 "int32"
1468 ]
1469 ],
1470 "params": {},
1471 "permutes": [
1472 "shape",
1473 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1474 ],
1475 "profile": [
1476 "tosa-bi",
1477 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1478 ]
1479 },
1480 "max_pool2d": {
1481 "group": "tensor",
1482 "generator_args": [
1483 [
1484 "--target-shape",
1485 "1,4,75,3",
1486 "--target-shape",
1487 "2,11,44,1",
1488 "--allow-pooling-and-conv-oversizes"
1489 ],
1490 [
1491 "--target-shape",
1492 "1,65535,5,1",
1493 "--target-shape",
1494 "1,3,65537,1",
1495 "--target-shape",
1496 "33333,3,2,1",
1497 "--target-dtype",
1498 "int8",
1499 "--allow-pooling-and-conv-oversizes"
1500 ]
1501 ],
1502 "params": {
1503 "shape": [],
1504 "type": [
1505 "i8",
1506 "i16"
1507 ],
1508 "kernel": [],
1509 "stride": [],
1510 "pad": []
1511 },
1512 "permutes": [
1513 "shape",
1514 "type",
1515 "pad"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1516 ],
1517 "profile": [
1518 "tosa-bi",
1519 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1520 ]
1521 },
1522 "minimum": {
1523 "group": "ew_binary",
1524 "generator_args": [
1525 [
1526 "--target-dtype",
1527 "int32",
1528 "--tensor-dim-range",
1529 "16,64",
1530 "--target-rank",
1531 "1",
1532 "--target-rank",
1533 "2",
1534 "--target-rank",
1535 "3"
1536 ],
1537 [
1538 "--target-dtype",
1539 "int32",
1540 "--tensor-dim-range",
1541 "1,16",
1542 "--target-rank",
1543 "4",
1544 "--target-rank",
1545 "5"
1546 ],
1547 [
1548 "--target-shape",
1549 "1,2,65533,1",
1550 "--target-shape",
1551 "1,65539,1,1,1",
1552 "--target-dtype",
1553 "int32"
1554 ]
1555 ],
1556 "params": {},
1557 "permutes": [
1558 "shape",
1559 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1560 ],
1561 "profile": [
1562 "tosa-bi",
1563 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1564 ]
1565 },
1566 "mul": {
1567 "group": "ew_binary",
1568 "generator_args": [
1569 [
1570 "--target-dtype",
1571 "int8",
1572 "--target-dtype",
1573 "int16",
1574 "--target-dtype",
1575 "int32",
1576 "--tensor-dim-range",
1577 "16,64",
1578 "--target-rank",
1579 "1",
1580 "--target-rank",
1581 "2",
1582 "--target-rank",
1583 "3"
1584 ],
1585 [
1586 "--target-dtype",
1587 "int8",
1588 "--target-dtype",
1589 "int16",
1590 "--target-dtype",
1591 "int32",
1592 "--tensor-dim-range",
1593 "1,16",
1594 "--target-rank",
1595 "4",
1596 "--target-rank",
1597 "5"
1598 ],
1599 [
1600 "--target-shape",
1601 "1,2,65500,1",
1602 "--target-shape",
1603 "1,68539,1,1,1",
1604 "--target-dtype",
1605 "int8"
1606 ]
1607 ],
1608 "params": {},
1609 "permutes": [
1610 "shape",
1611 "type",
1612 "perm",
1613 "shift"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1614 ],
1615 "profile": [
1616 "tosa-bi",
1617 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1618 ]
1619 },
1620 "negate": {
1621 "group": "ew_unary",
1622 "generator_args": [
1623 [
1624 "--target-dtype",
1625 "int8",
1626 "--target-dtype",
1627 "int16",
1628 "--target-dtype",
1629 "int32",
1630 "--tensor-dim-range",
1631 "22,64",
1632 "--target-rank",
1633 "1",
1634 "--target-rank",
1635 "2",
1636 "--target-rank",
1637 "3"
1638 ],
1639 [
1640 "--target-dtype",
1641 "int8",
1642 "--target-dtype",
1643 "int16",
1644 "--target-dtype",
1645 "int32",
1646 "--tensor-dim-range",
1647 "1,22",
1648 "--target-rank",
1649 "4",
1650 "--target-rank",
1651 "5"
1652 ],
1653 [
1654 "--target-shape",
1655 "3,1,65540,1",
1656 "--target-shape",
1657 "1,1,2,1,65531",
1658 "--target-dtype",
1659 "int8"
1660 ]
1661 ],
1662 "params": {},
1663 "permutes": [
1664 "shape",
1665 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1666 ],
1667 "profile": [
1668 "tosa-bi",
1669 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1670 ]
1671 },
1672 "pad": {
1673 "group": "data_layout",
1674 "generator_args": [
1675 [
1676 "--tensor-dim-range",
1677 "1,64",
1678 "--target-dtype",
1679 "int8",
1680 "--target-dtype",
1681 "int16",
1682 "--target-dtype",
1683 "int32",
1684 "--target-dtype",
1685 "bool",
1686 "--target-rank",
1687 "1",
1688 "--target-rank",
1689 "2",
1690 "--target-rank",
1691 "3"
1692 ],
1693 [
1694 "--tensor-dim-range",
1695 "1,16",
1696 "--target-dtype",
1697 "int8",
1698 "--target-dtype",
1699 "int16",
1700 "--target-dtype",
1701 "int32",
1702 "--target-dtype",
1703 "bool",
1704 "--target-rank",
1705 "4"
1706 ],
1707 [
1708 "--tensor-dim-range",
1709 "1,16",
1710 "--target-rank",
1711 "5",
1712 "--target-dtype",
1713 "int16"
1714 ],
1715 [
1716 "--target-shape",
1717 "1,1,65537,1",
1718 "--target-dtype",
1719 "int8"
1720 ]
1721 ],
1722 "params": {},
1723 "permutes": [
1724 "shape",
1725 "type"
1726 ],
1727 "sparsity": {
1728 "pad": 15
1729 },
1730 "preselected": [
1731 {
1732 "shape": "6",
1733 "type": "i8",
1734 "pad": "pad00"
1735 }
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1736 ],
1737 "profile": [
1738 "tosa-bi",
1739 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1740 ]
1741 },
1742 "reduce_all": {
1743 "group": "reduction",
1744 "generator_args": [
1745 [
1746 "--tensor-dim-range",
1747 "1,32"
1748 ],
1749 [
1750 "--target-shape",
1751 "65530,1,1,1",
1752 "--target-shape",
1753 "2,65538,1,1"
1754 ]
1755 ],
1756 "params": {},
1757 "permutes": [
1758 "shape",
1759 "type",
1760 "axis"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1761 ],
1762 "profile": [
1763 "tosa-bi",
1764 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1765 ]
1766 },
1767 "reduce_any": {
1768 "group": "reduction",
1769 "generator_args": [
1770 [
1771 "--tensor-dim-range",
1772 "1,32"
1773 ],
1774 [
1775 "--target-shape",
1776 "65530,1,1,1",
1777 "--target-shape",
1778 "2,65538,1,1"
1779 ]
1780 ],
1781 "params": {},
1782 "permutes": [
1783 "shape",
1784 "type",
1785 "axis"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1786 ],
1787 "profile": [
1788 "tosa-bi",
1789 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1790 ]
1791 },
1792 "reduce_max": {
1793 "group": "reduction",
1794 "generator_args": [
1795 [
1796 "--target-dtype",
1797 "int8",
1798 "--target-dtype",
1799 "int16",
1800 "--target-dtype",
1801 "int32",
1802 "--tensor-dim-range",
1803 "1,32"
1804 ],
1805 [
1806 "--target-shape",
1807 "65530,1,1,1",
1808 "--target-shape",
1809 "2,65538,1,1",
1810 "--target-dtype",
1811 "int8"
1812 ]
1813 ],
1814 "params": {},
1815 "permutes": [
1816 "shape",
1817 "type",
1818 "axis"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1819 ],
1820 "profile": [
1821 "tosa-bi",
1822 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1823 ]
1824 },
1825 "reduce_min": {
1826 "group": "reduction",
1827 "generator_args": [
1828 [
1829 "--target-dtype",
1830 "int8",
1831 "--target-dtype",
1832 "int16",
1833 "--target-dtype",
1834 "int32",
1835 "--tensor-dim-range",
1836 "1,24"
1837 ],
1838 [
1839 "--target-shape",
1840 "1,1,65538,1",
1841 "--target-shape",
1842 "2,65535,1,1",
1843 "--target-dtype",
1844 "int8"
1845 ]
1846 ],
1847 "params": {},
1848 "permutes": [
1849 "shape",
1850 "type",
1851 "axis"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1852 ],
1853 "profile": [
1854 "tosa-bi",
1855 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1856 ]
1857 },
1858 "reduce_sum": {
1859 "group": "reduction",
1860 "generator_args": [
1861 [
1862 "--target-dtype",
1863 "int32",
1864 "--tensor-dim-range",
1865 "1,32"
1866 ],
1867 [
1868 "--target-shape",
1869 "3,65532,1,1",
1870 "--target-shape",
1871 "65536,1,1,2",
1872 "--target-dtype",
1873 "int32"
1874 ]
1875 ],
1876 "params": {},
1877 "permutes": [
1878 "shape",
1879 "type",
1880 "axis"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1881 ],
1882 "profile": [
1883 "tosa-bi",
1884 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1885 ]
1886 },
1887 "rescale": {
1888 "group": "type_conversion",
1889 "generator_args": [
1890 [
1891 "--tensor-dim-range",
1892 "16,32",
1893 "--target-rank",
1894 "1",
1895 "--target-rank",
1896 "2",
1897 "--target-rank",
1898 "3"
1899 ],
1900 [
1901 "--tensor-dim-range",
1902 "1,16",
1903 "--target-rank",
Eric Kunzed4fe4bf2022-08-30 05:54:36 +0000[diff] [blame]1904 "4",
1905 "--target-rank",
1906 "5"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1907 ],
1908 [
1909 "--target-shape",
1910 "1,2,65533,1",
1911 "--target-shape",
1912 "1,65539,1,1",
1913 "--target-dtype",
1914 "int8"
1915 ]
1916 ],
1917 "params": {
1918 "scale": [
1919 "sc0",
1920 "sc0",
1921 "sc1",
1922 "sc1",
1923 "sc1",
1924 "sc1"
1925 ],
1926 "double_round": [
1927 "dr0",
1928 "dr0",
1929 "dr0",
1930 "dr0",
1931 "dr1",
1932 "dr1"
1933 ],
1934 "per_channel": [
1935 "pc0",
1936 "pc1",
1937 "pc0",
1938 "pc1",
1939 "pc0",
1940 "pc1",
1941 "pc1"
1942 ]
1943 },
1944 "permutes": [
1945 "shape",
1946 "type",
1947 "output_type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1948 ],
1949 "profile": [
1950 "tosa-bi",
1951 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1952 ]
1953 },
1954 "reshape": {
1955 "group": "data_layout",
1956 "generator_args": [
1957 [
1958 "--target-dtype",
1959 "int8",
1960 "--target-dtype",
1961 "int16",
1962 "--target-dtype",
1963 "int32",
1964 "--target-dtype",
1965 "bool",
1966 "--num-rand-permutations",
1967 "2"
1968 ],
1969 [
1970 "--target-shape",
1971 "1,1,65535,1",
1972 "--target-shape",
1973 "1,65538,1,2",
1974 "--target-dtype",
1975 "int8",
1976 "--num-rand-permutations",
1977 "2"
1978 ]
1979 ],
1980 "params": {},
1981 "permutes": [
1982 "shape",
1983 "type",
1984 "rank",
1985 "perm"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1986 ],
1987 "profile": [
1988 "tosa-bi",
1989 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1990 ]
1991 },
1992 "resize": {
1993 "group": "image",
1994 "generator_args": [
1995 [
1996 "--target-dtype",
1997 "int8",
1998 "--target-dtype",
1999 "int16"
2000 ],
2001 [
2002 "--target-shape",
2003 "1,3,16383,1",
2004 "--target-dtype",
Jeremy Johnsona0e03f32022-06-13 17:48:09 +0100[diff] [blame]2005 "int8",
2006 "--max-resize-output-dim",
2007 "9500"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2008 ]
2009 ],
2010 "params": {},
2011 "permutes": [
2012 "shape",
2013 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2014 ],
2015 "profile": [
2016 "tosa-bi",
2017 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2018 ]
2019 },
2020 "reverse": {
2021 "group": "data_layout",
2022 "generator_args": [
2023 [
2024 "--target-dtype",
2025 "int8",
2026 "--target-dtype",
2027 "int16",
2028 "--target-dtype",
2029 "int32",
2030 "--target-dtype",
2031 "bool",
2032 "--tensor-dim-range",
2033 "1,32"
2034 ],
2035 [
2036 "--target-shape",
2037 "1,1,65535",
2038 "--target-shape",
2039 "1,65540,1,2",
2040 "--target-dtype",
2041 "int8"
2042 ]
2043 ],
2044 "params": {},
2045 "permutes": [
2046 "shape",
2047 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2048 ],
2049 "profile": [
2050 "tosa-bi",
2051 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2052 ]
2053 },
2054 "scatter": {
2055 "group": "scatter_gather",
2056 "generator_args": [
2057 [
2058 "--target-dtype",
2059 "int16",
2060 "--target-dtype",
2061 "int32",
2062 "--tensor-dim-range",
2063 "4,64"
2064 ],
2065 [
2066 "--target-shape",
2067 "1,65540,1",
2068 "--target-shape",
2069 "1,1,65533",
2070 "--target-shape",
2071 "3000,7,1",
2072 "--target-dtype",
2073 "int8"
2074 ]
2075 ],
2076 "params": {},
2077 "permutes": [
2078 "shape",
2079 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2080 ],
2081 "profile": [
2082 "tosa-bi",
2083 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2084 ]
2085 },
2086 "select": {
2087 "group": "ew_ternary",
2088 "generator_args": [
2089 [
2090 "--target-dtype",
2091 "int8",
2092 "--target-dtype",
2093 "int16",
2094 "--target-dtype",
2095 "int32",
2096 "--target-dtype",
2097 "bool",
2098 "--tensor-dim-range",
2099 "16,64",
2100 "--target-rank",
2101 "1",
2102 "--target-rank",
2103 "2",
2104 "--target-rank",
2105 "3"
2106 ],
2107 [
2108 "--target-dtype",
2109 "int8",
2110 "--target-dtype",
2111 "int16",
2112 "--target-dtype",
2113 "int32",
2114 "--target-dtype",
2115 "bool",
2116 "--tensor-dim-range",
2117 "1,16",
2118 "--target-rank",
2119 "4",
2120 "--target-rank",
2121 "5"
2122 ],
2123 [
2124 "--target-shape",
2125 "1,1,65535,1",
2126 "--target-shape",
2127 "1,65538,1,2",
2128 "--target-dtype",
2129 "int8",
2130 "--num-rand-permutations",
2131 "2"
2132 ]
2133 ],
2134 "params": {},
2135 "permutes": [
2136 "shape",
2137 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2138 ],
2139 "profile": [
2140 "tosa-bi",
2141 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2142 ]
2143 },
2144 "slice": {
2145 "group": "data_layout",
2146 "generator_args": [
2147 [
2148 "--target-dtype",
2149 "int8",
2150 "--target-dtype",
2151 "int16",
2152 "--target-dtype",
2153 "int32",
2154 "--target-dtype",
2155 "bool",
2156 "--tensor-dim-range",
2157 "16,64",
2158 "--target-rank",
2159 "1",
2160 "--target-rank",
2161 "2",
2162 "--target-rank",
2163 "3",
2164 "--num-rand-permutations",
2165 "2"
2166 ],
2167 [
2168 "--target-dtype",
2169 "int8",
2170 "--target-dtype",
2171 "int16",
2172 "--target-dtype",
2173 "int32",
2174 "--target-dtype",
2175 "bool",
2176 "--tensor-dim-range",
2177 "1,16",
2178 "--target-rank",
2179 "4",
2180 "--num-rand-permutations",
2181 "2"
2182 ],
2183 [
2184 "--target-shape",
2185 "1,1,65535,1",
2186 "--target-shape",
2187 "1,65538,1,2",
2188 "--target-dtype",
2189 "int8",
2190 "--num-rand-permutations",
2191 "2"
2192 ]
2193 ],
2194 "params": {},
2195 "permutes": [
2196 "shape",
2197 "type",
2198 "perm"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2199 ],
2200 "profile": [
2201 "tosa-bi",
2202 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2203 ]
2204 },
2205 "sub": {
2206 "group": "ew_binary",
2207 "generator_args": [
2208 [
2209 "--target-dtype",
2210 "int32",
2211 "--tensor-dim-range",
2212 "16,64",
2213 "--target-rank",
2214 "1",
2215 "--target-rank",
2216 "2",
2217 "--target-rank",
2218 "3"
2219 ],
2220 [
2221 "--target-dtype",
2222 "int32",
2223 "--tensor-dim-range",
2224 "1,16",
2225 "--target-rank",
2226 "4",
2227 "--target-rank",
2228 "5"
2229 ],
2230 [
2231 "--target-shape",
2232 "1,2,65533,1",
2233 "--target-shape",
2234 "1,65539,1,1,1",
2235 "--target-dtype",
2236 "int32"
2237 ]
2238 ],
2239 "params": {},
2240 "permutes": [
2241 "shape",
2242 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2243 ],
2244 "profile": [
2245 "tosa-bi",
2246 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2247 ]
2248 },
2249 "table": {
2250 "group": "ew_binary",
2251 "generator_args": [
2252 [
2253 "--tensor-dim-range",
2254 "16,64",
2255 "--target-rank",
2256 "1",
2257 "--target-rank",
2258 "2",
2259 "--target-rank",
2260 "3"
2261 ],
2262 [
2263 "--tensor-dim-range",
2264 "1,16",
2265 "--target-rank",
2266 "4",
2267 "--target-rank",
2268 "5"
2269 ],
2270 [
2271 "--target-shape",
2272 "1,2,65536,1",
2273 "--target-shape",
2274 "1,65537,1,1,3",
2275 "--target-dtype",
2276 "int8"
2277 ]
2278 ],
2279 "params": {},
2280 "permutes": [
2281 "shape",
2282 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2283 ],
2284 "profile": [
2285 "tosa-bi",
2286 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2287 ]
2288 },
2289 "tile": {
2290 "group": "data_layout",
2291 "generator_args": [
2292 [
2293 "--target-dtype",
2294 "int16",
2295 "--target-dtype",
2296 "int32",
2297 "--tensor-dim-range",
2298 "4,32"
2299 ],
2300 [
2301 "--target-shape",
2302 "65533",
2303 "--target-shape",
2304 "1,2,65540,1",
2305 "--target-dtype",
2306 "int8"
2307 ]
2308 ],
2309 "params": {},
2310 "permutes": [
2311 "shape",
2312 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2313 ],
2314 "profile": [
2315 "tosa-bi",
2316 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2317 ]
2318 },
2319 "transpose": {
2320 "group": "data_layout",
2321 "generator_args": [
2322 [
2323 "--filter",
2324 "^transpose$",
2325 "--target-dtype",
2326 "int8",
2327 "--target-rank",
2328 "1",
2329 "--target-rank",
2330 "2",
2331 "--num-rand-permutations",
2332 "2"
2333 ],
2334 [
2335 "--filter",
2336 "^transpose$",
2337 "--target-dtype",
2338 "int8",
2339 "--target-dtype",
2340 "int16",
2341 "--target-dtype",
2342 "int32",
2343 "--target-dtype",
2344 "bool",
2345 "--target-rank",
2346 "3",
2347 "--target-rank",
2348 "4",
2349 "--num-rand-permutations",
2350 "2"
2351 ],
2352 [
2353 "--filter",
2354 "^transpose$",
2355 "--target-shape",
2356 "1,65537,1,2",
2357 "--target-shape",
2358 "65535,1,1,1",
2359 "--target-dtype",
2360 "int8",
2361 "--num-rand-permutations",
2362 "1"
2363 ]
2364 ],
2365 "params": {},
2366 "permutes": [
2367 "shape",
2368 "type",
2369 "perm"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2370 ],
2371 "profile": [
2372 "tosa-bi",
2373 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2374 ]
2375 },
2376 "transpose_conv2d": {
2377 "group": "tensor",
2378 "generator_args": [
2379 [
2380 "--target-shape",
2381 "1,49,33,1",
2382 "--target-shape",
2383 "2,11,33,3",
2384 "--target-dtype",
2385 "int16",
2386 "--target-dtype",
2387 "int8",
2388 "--allow-pooling-and-conv-oversizes"
2389 ],
2390 [
2391 "--target-shape",
2392 "1,65536,1,1",
2393 "--target-dtype",
2394 "int8",
2395 "--max-conv-dilation",
2396 "1"
2397 ]
2398 ],
2399 "params": {
2400 "shape": [],
2401 "type": [
2402 "i8xi8",
2403 "i16xi8",
2404 "i8xi4"
2405 ],
2406 "kernel": [],
2407 "stride": [],
2408 "pad": [],
2409 "out_shape": []
2410 },
2411 "permutes": [
2412 "shape",
2413 "type",
2414 "stride",
2415 "pad"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2416 ],
2417 "profile": [
2418 "tosa-bi",
2419 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2420 ]
2421 },
2422 "while_loop": {
2423 "group": "control_flow",
2424 "generator_args": [
2425 [
2426 "--target-dtype",
2427 "int32"
2428 ],
2429 [
2430 "--target-shape",
2431 "3,22,17,1,2",
2432 "--target-shape",
2433 "1,65537,2,1",
2434 "--target-dtype",
2435 "int32"
2436 ]
2437 ],
2438 "params": {},
2439 "permutes": [
2440 "shape",
2441 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2442 ],
2443 "profile": [
2444 "tosa-bi",
2445 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2446 ]
2447 }
2448}