Gitiles
Code Review Sign In
review.mlplatform.org / tosa / reference_model / a0e03f3f9a44d79367e2624f571009e4657c775a / . / verif / conformance / tosa_base_profile_ops_info.json
blob: d05b881ff4a33e6a41dd44c73e861dfc7e13895e [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",
1904 "4"
1905 ],
1906 [
1907 "--target-shape",
1908 "1,2,65533,1",
1909 "--target-shape",
1910 "1,65539,1,1",
1911 "--target-dtype",
1912 "int8"
1913 ]
1914 ],
1915 "params": {
1916 "scale": [
1917 "sc0",
1918 "sc0",
1919 "sc1",
1920 "sc1",
1921 "sc1",
1922 "sc1"
1923 ],
1924 "double_round": [
1925 "dr0",
1926 "dr0",
1927 "dr0",
1928 "dr0",
1929 "dr1",
1930 "dr1"
1931 ],
1932 "per_channel": [
1933 "pc0",
1934 "pc1",
1935 "pc0",
1936 "pc1",
1937 "pc0",
1938 "pc1",
1939 "pc1"
1940 ]
1941 },
1942 "permutes": [
1943 "shape",
1944 "type",
1945 "output_type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1946 ],
1947 "profile": [
1948 "tosa-bi",
1949 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1950 ]
1951 },
1952 "reshape": {
1953 "group": "data_layout",
1954 "generator_args": [
1955 [
1956 "--target-dtype",
1957 "int8",
1958 "--target-dtype",
1959 "int16",
1960 "--target-dtype",
1961 "int32",
1962 "--target-dtype",
1963 "bool",
1964 "--num-rand-permutations",
1965 "2"
1966 ],
1967 [
1968 "--target-shape",
1969 "1,1,65535,1",
1970 "--target-shape",
1971 "1,65538,1,2",
1972 "--target-dtype",
1973 "int8",
1974 "--num-rand-permutations",
1975 "2"
1976 ]
1977 ],
1978 "params": {},
1979 "permutes": [
1980 "shape",
1981 "type",
1982 "rank",
1983 "perm"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1984 ],
1985 "profile": [
1986 "tosa-bi",
1987 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1988 ]
1989 },
1990 "resize": {
1991 "group": "image",
1992 "generator_args": [
1993 [
1994 "--target-dtype",
1995 "int8",
1996 "--target-dtype",
1997 "int16"
1998 ],
1999 [
2000 "--target-shape",
2001 "1,3,16383,1",
2002 "--target-dtype",
Jeremy Johnsona0e03f32022-06-13 17:48:09 +0100[diff] [blame^]2003 "int8",
2004 "--max-resize-output-dim",
2005 "9500"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2006 ]
2007 ],
2008 "params": {},
2009 "permutes": [
2010 "shape",
2011 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2012 ],
2013 "profile": [
2014 "tosa-bi",
2015 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2016 ]
2017 },
2018 "reverse": {
2019 "group": "data_layout",
2020 "generator_args": [
2021 [
2022 "--target-dtype",
2023 "int8",
2024 "--target-dtype",
2025 "int16",
2026 "--target-dtype",
2027 "int32",
2028 "--target-dtype",
2029 "bool",
2030 "--tensor-dim-range",
2031 "1,32"
2032 ],
2033 [
2034 "--target-shape",
2035 "1,1,65535",
2036 "--target-shape",
2037 "1,65540,1,2",
2038 "--target-dtype",
2039 "int8"
2040 ]
2041 ],
2042 "params": {},
2043 "permutes": [
2044 "shape",
2045 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2046 ],
2047 "profile": [
2048 "tosa-bi",
2049 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2050 ]
2051 },
2052 "scatter": {
2053 "group": "scatter_gather",
2054 "generator_args": [
2055 [
2056 "--target-dtype",
2057 "int16",
2058 "--target-dtype",
2059 "int32",
2060 "--tensor-dim-range",
2061 "4,64"
2062 ],
2063 [
2064 "--target-shape",
2065 "1,65540,1",
2066 "--target-shape",
2067 "1,1,65533",
2068 "--target-shape",
2069 "3000,7,1",
2070 "--target-dtype",
2071 "int8"
2072 ]
2073 ],
2074 "params": {},
2075 "permutes": [
2076 "shape",
2077 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2078 ],
2079 "profile": [
2080 "tosa-bi",
2081 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2082 ]
2083 },
2084 "select": {
2085 "group": "ew_ternary",
2086 "generator_args": [
2087 [
2088 "--target-dtype",
2089 "int8",
2090 "--target-dtype",
2091 "int16",
2092 "--target-dtype",
2093 "int32",
2094 "--target-dtype",
2095 "bool",
2096 "--tensor-dim-range",
2097 "16,64",
2098 "--target-rank",
2099 "1",
2100 "--target-rank",
2101 "2",
2102 "--target-rank",
2103 "3"
2104 ],
2105 [
2106 "--target-dtype",
2107 "int8",
2108 "--target-dtype",
2109 "int16",
2110 "--target-dtype",
2111 "int32",
2112 "--target-dtype",
2113 "bool",
2114 "--tensor-dim-range",
2115 "1,16",
2116 "--target-rank",
2117 "4",
2118 "--target-rank",
2119 "5"
2120 ],
2121 [
2122 "--target-shape",
2123 "1,1,65535,1",
2124 "--target-shape",
2125 "1,65538,1,2",
2126 "--target-dtype",
2127 "int8",
2128 "--num-rand-permutations",
2129 "2"
2130 ]
2131 ],
2132 "params": {},
2133 "permutes": [
2134 "shape",
2135 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2136 ],
2137 "profile": [
2138 "tosa-bi",
2139 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2140 ]
2141 },
2142 "slice": {
2143 "group": "data_layout",
2144 "generator_args": [
2145 [
2146 "--target-dtype",
2147 "int8",
2148 "--target-dtype",
2149 "int16",
2150 "--target-dtype",
2151 "int32",
2152 "--target-dtype",
2153 "bool",
2154 "--tensor-dim-range",
2155 "16,64",
2156 "--target-rank",
2157 "1",
2158 "--target-rank",
2159 "2",
2160 "--target-rank",
2161 "3",
2162 "--num-rand-permutations",
2163 "2"
2164 ],
2165 [
2166 "--target-dtype",
2167 "int8",
2168 "--target-dtype",
2169 "int16",
2170 "--target-dtype",
2171 "int32",
2172 "--target-dtype",
2173 "bool",
2174 "--tensor-dim-range",
2175 "1,16",
2176 "--target-rank",
2177 "4",
2178 "--num-rand-permutations",
2179 "2"
2180 ],
2181 [
2182 "--target-shape",
2183 "1,1,65535,1",
2184 "--target-shape",
2185 "1,65538,1,2",
2186 "--target-dtype",
2187 "int8",
2188 "--num-rand-permutations",
2189 "2"
2190 ]
2191 ],
2192 "params": {},
2193 "permutes": [
2194 "shape",
2195 "type",
2196 "perm"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2197 ],
2198 "profile": [
2199 "tosa-bi",
2200 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2201 ]
2202 },
2203 "sub": {
2204 "group": "ew_binary",
2205 "generator_args": [
2206 [
2207 "--target-dtype",
2208 "int32",
2209 "--tensor-dim-range",
2210 "16,64",
2211 "--target-rank",
2212 "1",
2213 "--target-rank",
2214 "2",
2215 "--target-rank",
2216 "3"
2217 ],
2218 [
2219 "--target-dtype",
2220 "int32",
2221 "--tensor-dim-range",
2222 "1,16",
2223 "--target-rank",
2224 "4",
2225 "--target-rank",
2226 "5"
2227 ],
2228 [
2229 "--target-shape",
2230 "1,2,65533,1",
2231 "--target-shape",
2232 "1,65539,1,1,1",
2233 "--target-dtype",
2234 "int32"
2235 ]
2236 ],
2237 "params": {},
2238 "permutes": [
2239 "shape",
2240 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2241 ],
2242 "profile": [
2243 "tosa-bi",
2244 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2245 ]
2246 },
2247 "table": {
2248 "group": "ew_binary",
2249 "generator_args": [
2250 [
2251 "--tensor-dim-range",
2252 "16,64",
2253 "--target-rank",
2254 "1",
2255 "--target-rank",
2256 "2",
2257 "--target-rank",
2258 "3"
2259 ],
2260 [
2261 "--tensor-dim-range",
2262 "1,16",
2263 "--target-rank",
2264 "4",
2265 "--target-rank",
2266 "5"
2267 ],
2268 [
2269 "--target-shape",
2270 "1,2,65536,1",
2271 "--target-shape",
2272 "1,65537,1,1,3",
2273 "--target-dtype",
2274 "int8"
2275 ]
2276 ],
2277 "params": {},
2278 "permutes": [
2279 "shape",
2280 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2281 ],
2282 "profile": [
2283 "tosa-bi",
2284 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2285 ]
2286 },
2287 "tile": {
2288 "group": "data_layout",
2289 "generator_args": [
2290 [
2291 "--target-dtype",
2292 "int16",
2293 "--target-dtype",
2294 "int32",
2295 "--tensor-dim-range",
2296 "4,32"
2297 ],
2298 [
2299 "--target-shape",
2300 "65533",
2301 "--target-shape",
2302 "1,2,65540,1",
2303 "--target-dtype",
2304 "int8"
2305 ]
2306 ],
2307 "params": {},
2308 "permutes": [
2309 "shape",
2310 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2311 ],
2312 "profile": [
2313 "tosa-bi",
2314 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2315 ]
2316 },
2317 "transpose": {
2318 "group": "data_layout",
2319 "generator_args": [
2320 [
2321 "--filter",
2322 "^transpose$",
2323 "--target-dtype",
2324 "int8",
2325 "--target-rank",
2326 "1",
2327 "--target-rank",
2328 "2",
2329 "--num-rand-permutations",
2330 "2"
2331 ],
2332 [
2333 "--filter",
2334 "^transpose$",
2335 "--target-dtype",
2336 "int8",
2337 "--target-dtype",
2338 "int16",
2339 "--target-dtype",
2340 "int32",
2341 "--target-dtype",
2342 "bool",
2343 "--target-rank",
2344 "3",
2345 "--target-rank",
2346 "4",
2347 "--num-rand-permutations",
2348 "2"
2349 ],
2350 [
2351 "--filter",
2352 "^transpose$",
2353 "--target-shape",
2354 "1,65537,1,2",
2355 "--target-shape",
2356 "65535,1,1,1",
2357 "--target-dtype",
2358 "int8",
2359 "--num-rand-permutations",
2360 "1"
2361 ]
2362 ],
2363 "params": {},
2364 "permutes": [
2365 "shape",
2366 "type",
2367 "perm"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2368 ],
2369 "profile": [
2370 "tosa-bi",
2371 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2372 ]
2373 },
2374 "transpose_conv2d": {
2375 "group": "tensor",
2376 "generator_args": [
2377 [
2378 "--target-shape",
2379 "1,49,33,1",
2380 "--target-shape",
2381 "2,11,33,3",
2382 "--target-dtype",
2383 "int16",
2384 "--target-dtype",
2385 "int8",
2386 "--allow-pooling-and-conv-oversizes"
2387 ],
2388 [
2389 "--target-shape",
2390 "1,65536,1,1",
2391 "--target-dtype",
2392 "int8",
2393 "--max-conv-dilation",
2394 "1"
2395 ]
2396 ],
2397 "params": {
2398 "shape": [],
2399 "type": [
2400 "i8xi8",
2401 "i16xi8",
2402 "i8xi4"
2403 ],
2404 "kernel": [],
2405 "stride": [],
2406 "pad": [],
2407 "out_shape": []
2408 },
2409 "permutes": [
2410 "shape",
2411 "type",
2412 "stride",
2413 "pad"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2414 ],
2415 "profile": [
2416 "tosa-bi",
2417 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2418 ]
2419 },
2420 "while_loop": {
2421 "group": "control_flow",
2422 "generator_args": [
2423 [
2424 "--target-dtype",
2425 "int32"
2426 ],
2427 [
2428 "--target-shape",
2429 "3,22,17,1,2",
2430 "--target-shape",
2431 "1,65537,2,1",
2432 "--target-dtype",
2433 "int32"
2434 ]
2435 ],
2436 "params": {},
2437 "permutes": [
2438 "shape",
2439 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2440 ],
2441 "profile": [
2442 "tosa-bi",
2443 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2444 ]
2445 }
2446}