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