Gitiles
Code Review Sign In
review.mlplatform.org / tosa / reference_model / 417d8746c0335f99c2d3ec72580cb3cf610b01bc / . / verif / conformance / tosa_base_profile_ops_info.json
blob: ff88293dbb6e5ea6bc558045f44917cc61ccf2f5 [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",
206 "3,11,44,3",
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",
755 "1,65535,4,1",
756 "--target-shape",
757 "1,5,65536,1",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]758 "--max-conv-dilation",
759 "1",
760 "--allow-pooling-and-conv-oversizes"
761 ]
762 ],
763 "params": {
764 "shape": [],
765 "type": [
766 "i8xi8",
767 "i16xi8",
768 "i8xi4"
769 ],
770 "kernel": [],
771 "stride": [],
772 "pad": [],
773 "dilation": []
774 },
775 "permutes": [
776 "kernel",
777 "shape",
778 "type",
779 "pad"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]780 ],
781 "profile": [
782 "tosa-bi",
783 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]784 ]
785 },
786 "conv3d": {
787 "group": "tensor",
788 "generator_args": [
789 [
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]790 "--target-dtype",
791 "int8",
792 "--target-dtype",
793 "int16",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]794 "--target-shape",
795 "1,9,21,14,1",
796 "--target-shape",
797 "1,9,11,12,3",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]798 "--allow-pooling-and-conv-oversizes"
799 ],
800 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]801 "--target-dtype",
802 "int8",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]803 "--target-shape",
804 "1,1,65535,3,1",
805 "--target-shape",
806 "1,1,3,65536,1",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]807 "--max-conv-dilation",
808 "1",
809 "--allow-pooling-and-conv-oversizes"
810 ]
811 ],
812 "params": {
813 "shape": [],
814 "type": [
815 "i8xi8",
816 "i16xi8",
817 "i8xi4"
818 ],
819 "kernel": [],
820 "stride": [],
821 "pad": [],
822 "dilation": []
823 },
824 "permutes": [
825 "kernel",
826 "shape",
827 "type",
828 "pad",
829 "stride"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]830 ],
831 "profile": [
832 "tosa-bi",
833 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]834 ]
835 },
836 "depthwise_conv2d": {
837 "group": "tensor",
838 "generator_args": [
839 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]840 "--target-dtype",
841 "int8",
842 "--target-dtype",
843 "int16",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]844 "--target-shape",
845 "1,9,44,3",
846 "--target-shape",
847 "1,11,33,4",
848 "--allow-pooling-and-conv-oversizes"
849 ],
850 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]851 "--target-dtype",
852 "int8",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]853 "--target-shape",
854 "1,65532,2,1",
855 "--target-shape",
856 "1,4,65537,1",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]857 "--max-conv-dilation",
858 "1",
859 "--allow-pooling-and-conv-oversizes"
860 ]
861 ],
862 "params": {
863 "shape": [],
864 "type": [
865 "i8xi8",
866 "i16xi8",
867 "i8xi4"
868 ],
869 "kernel": [],
870 "stride": [],
871 "pad": [],
872 "dilation": []
873 },
874 "permutes": [
875 "kernel",
876 "shape",
877 "type",
878 "pad"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]879 ],
880 "profile": [
881 "tosa-bi",
882 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]883 ]
884 },
885 "equal": {
886 "group": "comparison",
887 "generator_args": [
888 [
889 "--target-dtype",
890 "int32",
891 "--tensor-dim-range",
892 "16,64",
893 "--target-rank",
894 "1",
895 "--target-rank",
896 "2",
897 "--target-rank",
898 "3"
899 ],
900 [
901 "--target-dtype",
902 "int32",
903 "--tensor-dim-range",
904 "1,16",
905 "--target-rank",
906 "4",
907 "--target-rank",
908 "5"
909 ],
910 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]911 "--target-dtype",
912 "int32",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]913 "--target-shape",
914 "1,2,65501,1",
915 "--target-shape",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]916 "1,65541,1,1,1"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]917 ]
918 ],
919 "params": {},
920 "permutes": [
921 "shape",
922 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]923 ],
924 "profile": [
925 "tosa-bi",
926 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]927 ]
928 },
929 "fully_connected": {
930 "group": "tensor",
931 "generator_args": [
932 [
933 "--target-dtype",
934 "int8",
935 "--target-dtype",
936 "int16"
937 ],
938 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]939 "--target-dtype",
940 "int8",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]941 "--target-shape",
942 "444,1",
943 "--target-shape",
944 "3,65538",
945 "--tensor-dim-range",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]946 "1,2"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]947 ],
948 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]949 "--target-dtype",
950 "int8",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]951 "--target-shape",
952 "3,16",
953 "--target-shape",
954 "1,23",
955 "--tensor-dim-range",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]956 "100,200"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]957 ]
958 ],
959 "params": {},
960 "permutes": [
961 "shape",
962 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]963 ],
964 "profile": [
965 "tosa-bi",
966 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]967 ]
968 },
969 "gather": {
970 "group": "scatter_gather",
971 "generator_args": [
972 [
973 "--target-dtype",
974 "int16",
975 "--target-dtype",
976 "int32",
977 "--tensor-dim-range",
978 "4,64"
979 ],
980 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]981 "--target-dtype",
982 "int8",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]983 "--target-shape",
984 "2,65533,1",
985 "--target-shape",
986 "1,2,65533",
987 "--target-shape",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]988 "5000,1,1"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]989 ]
990 ],
991 "params": {},
992 "permutes": [
993 "shape",
994 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]995 ],
996 "profile": [
997 "tosa-bi",
998 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]999 ]
1000 },
1001 "greater": {
1002 "group": "comparison",
1003 "generator_args": [
1004 [
1005 "--filter",
1006 "^greater$",
1007 "--target-dtype",
1008 "int32",
1009 "--tensor-dim-range",
1010 "10,64",
1011 "--target-rank",
1012 "1",
1013 "--target-rank",
1014 "2",
1015 "--target-rank",
1016 "3"
1017 ],
1018 [
1019 "--filter",
1020 "^greater$",
1021 "--target-dtype",
1022 "int32",
1023 "--tensor-dim-range",
1024 "1,15",
1025 "--target-rank",
1026 "4",
1027 "--target-rank",
1028 "5"
1029 ],
1030 [
1031 "--filter",
1032 "^greater$",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1033 "--target-dtype",
1034 "int32",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1035 "--target-shape",
1036 "1,2,65530,1",
1037 "--target-shape",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1038 "1,65539,1,2,1"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1039 ]
1040 ],
1041 "params": {},
1042 "permutes": [
1043 "shape",
1044 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1045 ],
1046 "profile": [
1047 "tosa-bi",
1048 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1049 ]
1050 },
1051 "greater_equal": {
1052 "group": "comparison",
1053 "generator_args": [
1054 [
1055 "--target-dtype",
1056 "int32",
1057 "--tensor-dim-range",
1058 "20,64",
1059 "--target-rank",
1060 "1",
1061 "--target-rank",
1062 "2",
1063 "--target-rank",
1064 "3"
1065 ],
1066 [
1067 "--target-dtype",
1068 "int32",
1069 "--tensor-dim-range",
1070 "1,20",
1071 "--target-rank",
1072 "4",
1073 "--target-rank",
1074 "5"
1075 ],
1076 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1077 "--target-dtype",
1078 "int32",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1079 "--target-shape",
1080 "1,2,65533,1",
1081 "--target-shape",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1082 "1,65539,1,3,1"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1083 ]
1084 ],
1085 "params": {},
1086 "permutes": [
1087 "shape",
1088 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1089 ],
1090 "profile": [
1091 "tosa-bi",
1092 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1093 ]
1094 },
1095 "identity": {
1096 "group": "data_nodes",
1097 "no_negative_tests": "true",
1098 "generator_args": [
1099 [
1100 "--target-dtype",
1101 "int8",
1102 "--target-dtype",
1103 "int16",
1104 "--target-dtype",
1105 "int32",
1106 "--target-dtype",
1107 "bool",
1108 "--tensor-dim-range",
1109 "1,61",
1110 "--target-rank",
1111 "1",
1112 "--target-rank",
1113 "2",
1114 "--target-rank",
1115 "3"
1116 ],
1117 [
1118 "--target-dtype",
1119 "int8",
1120 "--target-dtype",
1121 "int16",
1122 "--target-dtype",
1123 "int32",
1124 "--target-dtype",
1125 "bool",
1126 "--tensor-dim-range",
1127 "1,15",
1128 "--target-rank",
1129 "4",
1130 "--target-rank",
1131 "5"
1132 ],
1133 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1134 "--target-dtype",
1135 "int8",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1136 "--target-shape",
1137 "1,1,65536,1,1",
1138 "--target-shape",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1139 "1,65531,1,2"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1140 ]
1141 ],
1142 "params": {},
1143 "permutes": [
1144 "shape",
1145 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1146 ],
1147 "profile": [
1148 "tosa-bi",
1149 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1150 ]
1151 },
1152 "intdiv": {
1153 "group": "ew_binary",
1154 "generator_args": [
1155 [
1156 "--target-dtype",
1157 "int32",
1158 "--tensor-dim-range",
1159 "16,64",
1160 "--target-rank",
1161 "1",
1162 "--target-rank",
1163 "2",
1164 "--target-rank",
1165 "3"
1166 ],
1167 [
1168 "--target-dtype",
1169 "int32",
1170 "--tensor-dim-range",
1171 "1,16",
1172 "--target-rank",
1173 "4",
1174 "--target-rank",
1175 "5"
1176 ],
1177 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1178 "--target-dtype",
1179 "int32",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1180 "--target-shape",
1181 "1,65533,1",
1182 "--target-shape",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1183 "1,1,65539,1,1"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1184 ]
1185 ],
1186 "params": {},
1187 "permutes": [
1188 "shape",
1189 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1190 ],
1191 "profile": [
1192 "tosa-bi",
1193 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1194 ]
1195 },
1196 "logical_and": {
1197 "group": "ew_binary",
1198 "generator_args": [
1199 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1200 "--target-dtype",
1201 "bool",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1202 "--tensor-dim-range",
1203 "1,64",
1204 "--target-rank",
1205 "1",
1206 "--target-rank",
1207 "2",
1208 "--target-rank",
1209 "3"
1210 ],
1211 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1212 "--target-dtype",
1213 "bool",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1214 "--tensor-dim-range",
1215 "1,16",
1216 "--target-rank",
1217 "4",
1218 "--target-rank",
1219 "5"
1220 ],
1221 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1222 "--target-dtype",
1223 "bool",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1224 "--target-shape",
1225 "1,65538,1",
1226 "--target-shape",
1227 "1,2,1,65536"
1228 ]
1229 ],
1230 "params": {},
1231 "permutes": [
1232 "shape",
1233 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1234 ],
1235 "profile": [
1236 "tosa-bi",
1237 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1238 ]
1239 },
1240 "logical_left_shift": {
1241 "group": "ew_binary",
1242 "generator_args": [
1243 [
1244 "--target-dtype",
1245 "int8",
1246 "--target-dtype",
1247 "int16",
1248 "--target-dtype",
1249 "int32",
1250 "--tensor-dim-range",
1251 "16,64",
1252 "--target-rank",
1253 "1",
1254 "--target-rank",
1255 "2",
1256 "--target-rank",
1257 "3"
1258 ],
1259 [
1260 "--target-dtype",
1261 "int8",
1262 "--target-dtype",
1263 "int16",
1264 "--target-dtype",
1265 "int32",
1266 "--tensor-dim-range",
1267 "1,16",
1268 "--target-rank",
1269 "4",
1270 "--target-rank",
1271 "5"
1272 ],
1273 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1274 "--target-dtype",
1275 "int8",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1276 "--target-shape",
1277 "1,65532,1,1",
1278 "--target-shape",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1279 "1,2,1,65538"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1280 ]
1281 ],
1282 "params": {},
1283 "permutes": [
1284 "shape",
1285 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1286 ],
1287 "profile": [
1288 "tosa-bi",
1289 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1290 ]
1291 },
1292 "logical_not": {
1293 "group": "ew_unary",
1294 "generator_args": [
1295 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1296 "--target-dtype",
1297 "bool",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1298 "--tensor-dim-range",
1299 "2,64",
1300 "--target-rank",
1301 "1",
1302 "--target-rank",
1303 "2",
1304 "--target-rank",
1305 "3"
1306 ],
1307 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1308 "--target-dtype",
1309 "bool",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1310 "--tensor-dim-range",
1311 "1,15",
1312 "--target-rank",
1313 "4",
1314 "--target-rank",
1315 "5"
1316 ],
1317 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1318 "--target-dtype",
1319 "bool",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1320 "--target-shape",
1321 "1,65537,1",
1322 "--target-shape",
1323 "1,2,1,65535"
1324 ]
1325 ],
1326 "params": {},
1327 "permutes": [
1328 "shape",
1329 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1330 ],
1331 "profile": [
1332 "tosa-bi",
1333 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1334 ]
1335 },
1336 "logical_or": {
1337 "group": "ew_binary",
1338 "generator_args": [
1339 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1340 "--target-dtype",
1341 "bool",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1342 "--tensor-dim-range",
1343 "3,64",
1344 "--target-rank",
1345 "1",
1346 "--target-rank",
1347 "2",
1348 "--target-rank",
1349 "3"
1350 ],
1351 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1352 "--target-dtype",
1353 "bool",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1354 "--tensor-dim-range",
1355 "1,14",
1356 "--target-rank",
1357 "4",
1358 "--target-rank",
1359 "5"
1360 ],
1361 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1362 "--target-dtype",
1363 "bool",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1364 "--target-shape",
1365 "1,65536,1",
1366 "--target-shape",
1367 "1,2,1,65537"
1368 ]
1369 ],
1370 "params": {},
1371 "permutes": [
1372 "shape",
1373 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1374 ],
1375 "profile": [
1376 "tosa-bi",
1377 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1378 ]
1379 },
1380 "logical_right_shift": {
1381 "group": "ew_binary",
1382 "generator_args": [
1383 [
1384 "--target-dtype",
1385 "int8",
1386 "--target-dtype",
1387 "int16",
1388 "--target-dtype",
1389 "int32",
1390 "--tensor-dim-range",
1391 "25,64",
1392 "--target-rank",
1393 "1",
1394 "--target-rank",
1395 "2",
1396 "--target-rank",
1397 "3"
1398 ],
1399 [
1400 "--target-dtype",
1401 "int8",
1402 "--target-dtype",
1403 "int16",
1404 "--target-dtype",
1405 "int32",
1406 "--tensor-dim-range",
1407 "1,25",
1408 "--target-rank",
1409 "4",
1410 "--target-rank",
1411 "5"
1412 ],
1413 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1414 "--target-dtype",
1415 "int8",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1416 "--target-shape",
1417 "1,1,65540,3",
1418 "--target-shape",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1419 "2,1,65530,1,1"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1420 ]
1421 ],
1422 "params": {},
1423 "permutes": [
1424 "shape",
1425 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1426 ],
1427 "profile": [
1428 "tosa-bi",
1429 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1430 ]
1431 },
1432 "logical_xor": {
1433 "group": "ew_binary",
1434 "generator_args": [
1435 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1436 "--target-dtype",
1437 "bool",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1438 "--tensor-dim-range",
1439 "4,64",
1440 "--target-rank",
1441 "1",
1442 "--target-rank",
1443 "2",
1444 "--target-rank",
1445 "3"
1446 ],
1447 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1448 "--target-dtype",
1449 "bool",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1450 "--tensor-dim-range",
1451 "3,16",
1452 "--target-rank",
1453 "4",
1454 "--target-rank",
1455 "5"
1456 ],
1457 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1458 "--target-dtype",
1459 "bool",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1460 "--target-shape",
1461 "1,65539,1",
1462 "--target-shape",
1463 "1,2,1,65534"
1464 ]
1465 ],
1466 "params": {},
1467 "permutes": [
1468 "shape",
1469 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1470 ],
1471 "profile": [
1472 "tosa-bi",
1473 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1474 ]
1475 },
1476 "matmul": {
1477 "group": "tensor",
1478 "generator_args": [
1479 [
1480 "--target-dtype",
1481 "int8",
1482 "--target-dtype",
1483 "int16",
1484 "--tensor-dim-range",
1485 "16,64"
1486 ],
1487 [
1488 "--target-dtype",
1489 "int8",
1490 "--target-dtype",
1491 "int16",
1492 "--tensor-dim-range",
1493 "1,16"
1494 ],
1495 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1496 "--target-dtype",
1497 "int8",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1498 "--target-shape",
1499 "1,65532,1",
1500 "--target-shape",
1501 "1,2,65538",
1502 "--target-shape",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1503 "2000,1,1"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1504 ]
1505 ],
1506 "params": {},
1507 "permutes": [
1508 "shape",
1509 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1510 ],
1511 "profile": [
1512 "tosa-bi",
1513 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1514 ]
1515 },
1516 "maximum": {
1517 "group": "ew_binary",
1518 "generator_args": [
1519 [
1520 "--target-dtype",
1521 "int32",
1522 "--tensor-dim-range",
1523 "16,64",
1524 "--target-rank",
1525 "1",
1526 "--target-rank",
1527 "2",
1528 "--target-rank",
1529 "3"
1530 ],
1531 [
1532 "--target-dtype",
1533 "int32",
1534 "--tensor-dim-range",
1535 "1,16",
1536 "--target-rank",
1537 "4",
1538 "--target-rank",
1539 "5"
1540 ],
1541 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1542 "--target-dtype",
1543 "int32",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1544 "--target-shape",
1545 "1,2,65533,1",
1546 "--target-shape",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1547 "1,65539,1,1,1"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1548 ]
1549 ],
1550 "params": {},
1551 "permutes": [
1552 "shape",
1553 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1554 ],
1555 "profile": [
1556 "tosa-bi",
1557 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1558 ]
1559 },
1560 "max_pool2d": {
1561 "group": "tensor",
1562 "generator_args": [
1563 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1564 "--target-dtype",
1565 "int8",
1566 "--target-dtype",
1567 "int16",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1568 "--target-shape",
1569 "1,4,75,3",
1570 "--target-shape",
1571 "2,11,44,1",
1572 "--allow-pooling-and-conv-oversizes"
1573 ],
1574 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1575 "--target-dtype",
1576 "int8",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1577 "--target-shape",
1578 "1,65535,5,1",
1579 "--target-shape",
1580 "1,3,65537,1",
1581 "--target-shape",
1582 "33333,3,2,1",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1583 "--allow-pooling-and-conv-oversizes"
1584 ]
1585 ],
1586 "params": {
1587 "shape": [],
1588 "type": [
1589 "i8",
1590 "i16"
1591 ],
1592 "kernel": [],
1593 "stride": [],
1594 "pad": []
1595 },
1596 "permutes": [
1597 "shape",
1598 "type",
1599 "pad"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1600 ],
1601 "profile": [
1602 "tosa-bi",
1603 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1604 ]
1605 },
1606 "minimum": {
1607 "group": "ew_binary",
1608 "generator_args": [
1609 [
1610 "--target-dtype",
1611 "int32",
1612 "--tensor-dim-range",
1613 "16,64",
1614 "--target-rank",
1615 "1",
1616 "--target-rank",
1617 "2",
1618 "--target-rank",
1619 "3"
1620 ],
1621 [
1622 "--target-dtype",
1623 "int32",
1624 "--tensor-dim-range",
1625 "1,16",
1626 "--target-rank",
1627 "4",
1628 "--target-rank",
1629 "5"
1630 ],
1631 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1632 "--target-dtype",
1633 "int32",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1634 "--target-shape",
1635 "1,2,65533,1",
1636 "--target-shape",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1637 "1,65539,1,1,1"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1638 ]
1639 ],
1640 "params": {},
1641 "permutes": [
1642 "shape",
1643 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1644 ],
1645 "profile": [
1646 "tosa-bi",
1647 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1648 ]
1649 },
1650 "mul": {
1651 "group": "ew_binary",
1652 "generator_args": [
1653 [
1654 "--target-dtype",
1655 "int8",
1656 "--target-dtype",
1657 "int16",
1658 "--target-dtype",
1659 "int32",
1660 "--tensor-dim-range",
1661 "16,64",
1662 "--target-rank",
1663 "1",
1664 "--target-rank",
1665 "2",
1666 "--target-rank",
1667 "3"
1668 ],
1669 [
1670 "--target-dtype",
1671 "int8",
1672 "--target-dtype",
1673 "int16",
1674 "--target-dtype",
1675 "int32",
1676 "--tensor-dim-range",
1677 "1,16",
1678 "--target-rank",
1679 "4",
1680 "--target-rank",
1681 "5"
1682 ],
1683 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1684 "--target-dtype",
1685 "int8",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1686 "--target-shape",
1687 "1,2,65500,1",
1688 "--target-shape",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1689 "1,68539,1,1,1"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1690 ]
1691 ],
1692 "params": {},
1693 "permutes": [
1694 "shape",
1695 "type",
1696 "perm",
1697 "shift"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1698 ],
1699 "profile": [
1700 "tosa-bi",
1701 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1702 ]
1703 },
1704 "negate": {
1705 "group": "ew_unary",
1706 "generator_args": [
1707 [
1708 "--target-dtype",
1709 "int8",
1710 "--target-dtype",
1711 "int16",
1712 "--target-dtype",
1713 "int32",
1714 "--tensor-dim-range",
1715 "22,64",
1716 "--target-rank",
1717 "1",
1718 "--target-rank",
1719 "2",
1720 "--target-rank",
1721 "3"
1722 ],
1723 [
1724 "--target-dtype",
1725 "int8",
1726 "--target-dtype",
1727 "int16",
1728 "--target-dtype",
1729 "int32",
1730 "--tensor-dim-range",
1731 "1,22",
1732 "--target-rank",
1733 "4",
1734 "--target-rank",
1735 "5"
1736 ],
1737 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1738 "--target-dtype",
1739 "int8",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1740 "--target-shape",
1741 "3,1,65540,1",
1742 "--target-shape",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1743 "1,1,2,1,65531"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1744 ]
1745 ],
1746 "params": {},
1747 "permutes": [
1748 "shape",
1749 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1750 ],
1751 "profile": [
1752 "tosa-bi",
1753 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1754 ]
1755 },
1756 "pad": {
1757 "group": "data_layout",
1758 "generator_args": [
1759 [
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1760 "--target-dtype",
1761 "int8",
1762 "--target-dtype",
1763 "int16",
1764 "--target-dtype",
1765 "int32",
1766 "--target-dtype",
1767 "bool",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1768 "--tensor-dim-range",
1769 "1,64",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1770 "--target-rank",
1771 "1",
1772 "--target-rank",
1773 "2",
1774 "--target-rank",
1775 "3"
1776 ],
1777 [
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1778 "--target-dtype",
1779 "int8",
1780 "--target-dtype",
1781 "int16",
1782 "--target-dtype",
1783 "int32",
1784 "--target-dtype",
1785 "bool",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1786 "--tensor-dim-range",
1787 "1,16",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1788 "--target-rank",
1789 "4"
1790 ],
1791 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1792 "--target-dtype",
1793 "int16",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1794 "--tensor-dim-range",
1795 "1,16",
1796 "--target-rank",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1797 "5"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1798 ],
1799 [
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1800 "--target-dtype",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1801 "int8",
1802 "--target-shape",
1803 "1,1,65537,1"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1804 ]
1805 ],
1806 "params": {},
1807 "permutes": [
1808 "shape",
1809 "type"
1810 ],
1811 "sparsity": {
1812 "pad": 15
1813 },
1814 "preselected": [
1815 {
1816 "shape": "6",
1817 "type": "i8",
1818 "pad": "pad00"
1819 }
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1820 ],
1821 "profile": [
1822 "tosa-bi",
1823 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1824 ]
1825 },
1826 "reduce_all": {
1827 "group": "reduction",
1828 "generator_args": [
1829 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1830 "--target-dtype",
1831 "bool",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1832 "--tensor-dim-range",
1833 "1,32"
1834 ],
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 "--target-shape",
1839 "65530,1,1,1",
1840 "--target-shape",
1841 "2,65538,1,1"
1842 ]
1843 ],
1844 "params": {},
1845 "permutes": [
1846 "shape",
1847 "type",
1848 "axis"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1849 ],
1850 "profile": [
1851 "tosa-bi",
1852 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1853 ]
1854 },
1855 "reduce_any": {
1856 "group": "reduction",
1857 "generator_args": [
1858 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1859 "--target-dtype",
1860 "bool",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1861 "--tensor-dim-range",
1862 "1,32"
1863 ],
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 "--target-shape",
1868 "65530,1,1,1",
1869 "--target-shape",
1870 "2,65538,1,1"
1871 ]
1872 ],
1873 "params": {},
1874 "permutes": [
1875 "shape",
1876 "type",
1877 "axis"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1878 ],
1879 "profile": [
1880 "tosa-bi",
1881 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1882 ]
1883 },
1884 "reduce_max": {
1885 "group": "reduction",
1886 "generator_args": [
1887 [
1888 "--target-dtype",
1889 "int8",
1890 "--target-dtype",
1891 "int16",
1892 "--target-dtype",
1893 "int32",
1894 "--tensor-dim-range",
1895 "1,32"
1896 ],
1897 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1898 "--target-dtype",
1899 "int8",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1900 "--target-shape",
1901 "65530,1,1,1",
1902 "--target-shape",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1903 "2,65538,1,1"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1904 ]
1905 ],
1906 "params": {},
1907 "permutes": [
1908 "shape",
1909 "type",
1910 "axis"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1911 ],
1912 "profile": [
1913 "tosa-bi",
1914 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1915 ]
1916 },
1917 "reduce_min": {
1918 "group": "reduction",
1919 "generator_args": [
1920 [
1921 "--target-dtype",
1922 "int8",
1923 "--target-dtype",
1924 "int16",
1925 "--target-dtype",
1926 "int32",
1927 "--tensor-dim-range",
1928 "1,24"
1929 ],
1930 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1931 "--target-dtype",
1932 "int8",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1933 "--target-shape",
1934 "1,1,65538,1",
1935 "--target-shape",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1936 "2,65535,1,1"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1937 ]
1938 ],
1939 "params": {},
1940 "permutes": [
1941 "shape",
1942 "type",
1943 "axis"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1944 ],
1945 "profile": [
1946 "tosa-bi",
1947 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1948 ]
1949 },
1950 "reduce_sum": {
1951 "group": "reduction",
1952 "generator_args": [
1953 [
1954 "--target-dtype",
1955 "int32",
1956 "--tensor-dim-range",
1957 "1,32"
1958 ],
1959 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1960 "--target-dtype",
1961 "int32",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1962 "--target-shape",
1963 "3,65532,1,1",
1964 "--target-shape",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1965 "65536,1,1,2"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1966 ]
1967 ],
1968 "params": {},
1969 "permutes": [
1970 "shape",
1971 "type",
1972 "axis"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]1973 ],
1974 "profile": [
1975 "tosa-bi",
1976 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1977 ]
1978 },
1979 "rescale": {
1980 "group": "type_conversion",
1981 "generator_args": [
1982 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]1983 "--target-dtype",
1984 "int8",
1985 "--target-dtype",
1986 "int16",
1987 "--target-dtype",
1988 "int32",
1989 "--target-dtype",
1990 "int48",
1991 "--target-dtype",
1992 "uint8",
1993 "--target-dtype",
1994 "uint16",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]1995 "--tensor-dim-range",
1996 "16,32",
1997 "--target-rank",
1998 "1",
1999 "--target-rank",
2000 "2",
2001 "--target-rank",
2002 "3"
2003 ],
2004 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]2005 "--target-dtype",
2006 "int8",
2007 "--target-dtype",
2008 "int16",
2009 "--target-dtype",
2010 "int32",
2011 "--target-dtype",
2012 "int48",
2013 "--target-dtype",
2014 "uint8",
2015 "--target-dtype",
2016 "uint16",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2017 "--tensor-dim-range",
2018 "1,16",
2019 "--target-rank",
Eric Kunzed4fe4bf2022-08-30 05:54:36 +0000[diff] [blame]2020 "4",
2021 "--target-rank",
2022 "5"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2023 ],
2024 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]2025 "--target-dtype",
2026 "int8",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2027 "--target-shape",
2028 "1,2,65533,1",
2029 "--target-shape",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]2030 "1,65539,1,1"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2031 ]
2032 ],
2033 "params": {
2034 "scale": [
2035 "sc0",
2036 "sc0",
2037 "sc1",
2038 "sc1",
2039 "sc1",
2040 "sc1"
2041 ],
2042 "double_round": [
2043 "dr0",
2044 "dr0",
2045 "dr0",
2046 "dr0",
2047 "dr1",
2048 "dr1"
2049 ],
2050 "per_channel": [
2051 "pc0",
2052 "pc1",
2053 "pc0",
2054 "pc1",
2055 "pc0",
2056 "pc1",
2057 "pc1"
2058 ]
2059 },
2060 "permutes": [
2061 "shape",
2062 "type",
2063 "output_type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2064 ],
2065 "profile": [
2066 "tosa-bi",
2067 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2068 ]
2069 },
2070 "reshape": {
2071 "group": "data_layout",
2072 "generator_args": [
2073 [
2074 "--target-dtype",
2075 "int8",
2076 "--target-dtype",
2077 "int16",
2078 "--target-dtype",
2079 "int32",
2080 "--target-dtype",
2081 "bool",
2082 "--num-rand-permutations",
2083 "2"
2084 ],
2085 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]2086 "--target-dtype",
2087 "int8",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2088 "--target-shape",
2089 "1,1,65535,1",
2090 "--target-shape",
2091 "1,65538,1,2",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2092 "--num-rand-permutations",
2093 "2"
2094 ]
2095 ],
2096 "params": {},
2097 "permutes": [
2098 "shape",
2099 "type",
2100 "rank",
2101 "perm"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2102 ],
2103 "profile": [
2104 "tosa-bi",
2105 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2106 ]
2107 },
2108 "resize": {
2109 "group": "image",
2110 "generator_args": [
2111 [
2112 "--target-dtype",
2113 "int8",
2114 "--target-dtype",
2115 "int16"
2116 ],
2117 [
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2118 "--target-dtype",
Jeremy Johnsona0e03f32022-06-13 17:48:09 +0100[diff] [blame]2119 "int8",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]2120 "--target-shape",
2121 "1,3,16383,1",
Jeremy Johnsona0e03f32022-06-13 17:48:09 +0100[diff] [blame]2122 "--max-resize-output-dim",
2123 "9500"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2124 ]
2125 ],
2126 "params": {},
2127 "permutes": [
2128 "shape",
2129 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2130 ],
2131 "profile": [
2132 "tosa-bi",
2133 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2134 ]
2135 },
2136 "reverse": {
2137 "group": "data_layout",
2138 "generator_args": [
2139 [
2140 "--target-dtype",
2141 "int8",
2142 "--target-dtype",
2143 "int16",
2144 "--target-dtype",
2145 "int32",
2146 "--target-dtype",
2147 "bool",
2148 "--tensor-dim-range",
2149 "1,32"
2150 ],
2151 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]2152 "--target-dtype",
2153 "int8",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2154 "--target-shape",
2155 "1,1,65535",
2156 "--target-shape",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]2157 "1,65540,1,2"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2158 ]
2159 ],
2160 "params": {},
2161 "permutes": [
2162 "shape",
2163 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2164 ],
2165 "profile": [
2166 "tosa-bi",
2167 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2168 ]
2169 },
2170 "scatter": {
2171 "group": "scatter_gather",
2172 "generator_args": [
2173 [
2174 "--target-dtype",
2175 "int16",
2176 "--target-dtype",
2177 "int32",
2178 "--tensor-dim-range",
2179 "4,64"
2180 ],
2181 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]2182 "--target-dtype",
2183 "int8",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2184 "--target-shape",
2185 "1,65540,1",
2186 "--target-shape",
2187 "1,1,65533",
2188 "--target-shape",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]2189 "3000,7,1"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2190 ]
2191 ],
2192 "params": {},
2193 "permutes": [
2194 "shape",
2195 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2196 ],
2197 "profile": [
2198 "tosa-bi",
2199 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2200 ]
2201 },
2202 "select": {
2203 "group": "ew_ternary",
2204 "generator_args": [
2205 [
2206 "--target-dtype",
2207 "int8",
2208 "--target-dtype",
2209 "int16",
2210 "--target-dtype",
2211 "int32",
2212 "--target-dtype",
2213 "bool",
2214 "--tensor-dim-range",
2215 "16,64",
2216 "--target-rank",
2217 "1",
2218 "--target-rank",
2219 "2",
2220 "--target-rank",
2221 "3"
2222 ],
2223 [
2224 "--target-dtype",
2225 "int8",
2226 "--target-dtype",
2227 "int16",
2228 "--target-dtype",
2229 "int32",
2230 "--target-dtype",
2231 "bool",
2232 "--tensor-dim-range",
2233 "1,16",
2234 "--target-rank",
2235 "4",
2236 "--target-rank",
2237 "5"
2238 ],
2239 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]2240 "--target-dtype",
2241 "int8",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2242 "--target-shape",
2243 "1,1,65535,1",
2244 "--target-shape",
2245 "1,65538,1,2",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2246 "--num-rand-permutations",
2247 "2"
2248 ]
2249 ],
2250 "params": {},
2251 "permutes": [
2252 "shape",
2253 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2254 ],
2255 "profile": [
2256 "tosa-bi",
2257 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2258 ]
2259 },
2260 "slice": {
2261 "group": "data_layout",
2262 "generator_args": [
2263 [
2264 "--target-dtype",
2265 "int8",
2266 "--target-dtype",
2267 "int16",
2268 "--target-dtype",
2269 "int32",
2270 "--target-dtype",
2271 "bool",
2272 "--tensor-dim-range",
2273 "16,64",
2274 "--target-rank",
2275 "1",
2276 "--target-rank",
2277 "2",
2278 "--target-rank",
2279 "3",
2280 "--num-rand-permutations",
2281 "2"
2282 ],
2283 [
2284 "--target-dtype",
2285 "int8",
2286 "--target-dtype",
2287 "int16",
2288 "--target-dtype",
2289 "int32",
2290 "--target-dtype",
2291 "bool",
2292 "--tensor-dim-range",
2293 "1,16",
2294 "--target-rank",
2295 "4",
2296 "--num-rand-permutations",
2297 "2"
2298 ],
2299 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]2300 "--target-dtype",
2301 "int8",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2302 "--target-shape",
2303 "1,1,65535,1",
2304 "--target-shape",
2305 "1,65538,1,2",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2306 "--num-rand-permutations",
2307 "2"
2308 ]
2309 ],
2310 "params": {},
2311 "permutes": [
2312 "shape",
2313 "type",
2314 "perm"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2315 ],
2316 "profile": [
2317 "tosa-bi",
2318 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2319 ]
2320 },
2321 "sub": {
2322 "group": "ew_binary",
2323 "generator_args": [
2324 [
2325 "--target-dtype",
2326 "int32",
2327 "--tensor-dim-range",
2328 "16,64",
2329 "--target-rank",
2330 "1",
2331 "--target-rank",
2332 "2",
2333 "--target-rank",
2334 "3"
2335 ],
2336 [
2337 "--target-dtype",
2338 "int32",
2339 "--tensor-dim-range",
2340 "1,16",
2341 "--target-rank",
2342 "4",
2343 "--target-rank",
2344 "5"
2345 ],
2346 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]2347 "--target-dtype",
2348 "int32",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2349 "--target-shape",
2350 "1,2,65533,1",
2351 "--target-shape",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]2352 "1,65539,1,1,1"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2353 ]
2354 ],
2355 "params": {},
2356 "permutes": [
2357 "shape",
2358 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2359 ],
2360 "profile": [
2361 "tosa-bi",
2362 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2363 ]
2364 },
2365 "table": {
2366 "group": "ew_binary",
2367 "generator_args": [
2368 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]2369 "--target-dtype",
2370 "int8",
2371 "--target-dtype",
2372 "int16",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2373 "--tensor-dim-range",
2374 "16,64",
2375 "--target-rank",
2376 "1",
2377 "--target-rank",
2378 "2",
2379 "--target-rank",
2380 "3"
2381 ],
2382 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]2383 "--target-dtype",
2384 "int8",
2385 "--target-dtype",
2386 "int16",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2387 "--tensor-dim-range",
2388 "1,16",
2389 "--target-rank",
2390 "4",
2391 "--target-rank",
2392 "5"
2393 ],
2394 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]2395 "--target-dtype",
2396 "int8",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2397 "--target-shape",
2398 "1,2,65536,1",
2399 "--target-shape",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]2400 "1,65537,1,1,3"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2401 ]
2402 ],
2403 "params": {},
2404 "permutes": [
2405 "shape",
2406 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2407 ],
2408 "profile": [
2409 "tosa-bi",
2410 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2411 ]
2412 },
2413 "tile": {
2414 "group": "data_layout",
2415 "generator_args": [
2416 [
2417 "--target-dtype",
2418 "int16",
2419 "--target-dtype",
2420 "int32",
2421 "--tensor-dim-range",
2422 "4,32"
2423 ],
2424 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]2425 "--target-dtype",
Jeremy Johnson3b0544c2022-10-18 16:32:19 +0100[diff] [blame]2426 "bool",
2427 "--tensor-dim-range",
2428 "1,16"
2429 ],
2430 [
2431 "--target-dtype",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]2432 "int8",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2433 "--target-shape",
2434 "65533",
2435 "--target-shape",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]2436 "1,2,65540,1"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2437 ]
2438 ],
2439 "params": {},
2440 "permutes": [
2441 "shape",
2442 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2443 ],
2444 "profile": [
2445 "tosa-bi",
2446 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2447 ]
2448 },
2449 "transpose": {
2450 "group": "data_layout",
2451 "generator_args": [
2452 [
2453 "--filter",
2454 "^transpose$",
2455 "--target-dtype",
2456 "int8",
2457 "--target-rank",
2458 "1",
2459 "--target-rank",
2460 "2",
2461 "--num-rand-permutations",
2462 "2"
2463 ],
2464 [
2465 "--filter",
2466 "^transpose$",
2467 "--target-dtype",
2468 "int8",
2469 "--target-dtype",
2470 "int16",
2471 "--target-dtype",
2472 "int32",
2473 "--target-dtype",
2474 "bool",
2475 "--target-rank",
2476 "3",
2477 "--target-rank",
2478 "4",
2479 "--num-rand-permutations",
2480 "2"
2481 ],
2482 [
2483 "--filter",
2484 "^transpose$",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]2485 "--target-dtype",
2486 "int8",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2487 "--target-shape",
2488 "1,65537,1,2",
2489 "--target-shape",
2490 "65535,1,1,1",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2491 "--num-rand-permutations",
2492 "1"
2493 ]
2494 ],
2495 "params": {},
2496 "permutes": [
2497 "shape",
2498 "type",
2499 "perm"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2500 ],
2501 "profile": [
2502 "tosa-bi",
2503 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2504 ]
2505 },
2506 "transpose_conv2d": {
2507 "group": "tensor",
2508 "generator_args": [
2509 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]2510 "--target-dtype",
2511 "int8",
2512 "--target-dtype",
2513 "int16",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2514 "--target-shape",
2515 "1,49,33,1",
2516 "--target-shape",
2517 "2,11,33,3",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2518 "--allow-pooling-and-conv-oversizes"
2519 ],
2520 [
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2521 "--target-dtype",
2522 "int8",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]2523 "--target-shape",
2524 "1,65536,1,1",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2525 "--max-conv-dilation",
2526 "1"
2527 ]
2528 ],
2529 "params": {
2530 "shape": [],
2531 "type": [
2532 "i8xi8",
2533 "i16xi8",
2534 "i8xi4"
2535 ],
2536 "kernel": [],
2537 "stride": [],
2538 "pad": [],
2539 "out_shape": []
2540 },
2541 "permutes": [
2542 "shape",
2543 "type",
2544 "stride",
2545 "pad"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2546 ],
2547 "profile": [
2548 "tosa-bi",
2549 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2550 ]
2551 },
2552 "while_loop": {
2553 "group": "control_flow",
2554 "generator_args": [
2555 [
2556 "--target-dtype",
2557 "int32"
2558 ],
2559 [
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]2560 "--target-dtype",
2561 "int32",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2562 "--target-shape",
2563 "3,22,17,1,2",
2564 "--target-shape",
Jeremy Johnson93d43902022-09-27 12:26:14 +0100[diff] [blame]2565 "1,65537,2,1"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2566 ]
2567 ],
2568 "params": {},
2569 "permutes": [
2570 "shape",
2571 "type"
Jeremy Johnson88588622022-07-12 16:42:29 +0100[diff] [blame]2572 ],
2573 "profile": [
2574 "tosa-bi",
2575 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100[diff] [blame]2576 ]
2577 }
2578}