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