Gitiles
Code Review Sign In
review.mlplatform.org / tosa / reference_model / 52ac9d6e4afa916acc4cab0617b940f4ab7aaf4f / . / verif / conformance / tosa_main_profile_ops_info.json
blob: 0614dfc4c1ce37702b1a790d062a1af3c4fb3557 [file] [log] [blame]
Jeremy Johnsone4b08ff2022-09-15 10:38:17 +0100[diff] [blame]1{
James Ward52ac9d62022-11-16 12:53:49 +0000[diff] [blame^]2 "argmax": {
3 "group": "tensor",
4 "generator_args": [
5 [
6 "--target-dtype",
7 "fp32",
8 "--target-dtype",
9 "fp16",
10 "--target-dtype",
11 "bf16",
12 "--fp-values-range",
13 "-2.0,2.0",
14 "--tensor-dim-range",
15 "32,64",
16 "--target-rank",
17 "1",
18 "--target-rank",
19 "2",
20 "--target-rank",
21 "3",
22 "--num-rand-permutations",
23 "2"
24 ],
25 [
26 "--target-dtype",
27 "fp32",
28 "--target-dtype",
29 "fp16",
30 "--target-dtype",
31 "bf16",
32 "--fp-values-range",
33 "-2.0,2.0",
34 "--tensor-dim-range",
35 "1,32",
36 "--target-rank",
37 "4",
38 "--num-rand-permutations",
39 "2"
40 ],
41 [
42 "--target-dtype",
43 "fp32",
44 "--fp-values-range",
45 "-2.0,2.0",
46 "--target-shape",
47 "1,3,65535,1",
48 "--target-shape",
49 "1,65537,1,2",
50 "--num-rand-permutations",
51 "2"
52 ]
53 ],
54 "params": {
55 },
56 "permutes": [
57 "shape",
58 "type"
59 ],
60 "profile": [
61 "tosa-mi"
62 ]
63 },
Jeremy Johnsone4b08ff2022-09-15 10:38:17 +0100[diff] [blame]64 "avg_pool2d": {
65 "group": "tensor",
66 "generator_args": [
67 [
68 "--target-dtype",
69 "fp32",
70 "--target-dtype",
71 "fp16",
72 "--target-dtype",
73 "bf16",
74 "--fp-values-range",
75 "-2.0,2.0",
76 "--target-shape",
77 "1,49,42,28",
78 "--target-shape",
79 "3,11,44,3",
80 "--allow-pooling-and-conv-oversizes"
81 ],
82 [
83 "--target-dtype",
84 "fp32",
85 "--fp-values-range",
86 "-2.0,2.0",
87 "--target-shape",
88 "1,65535,5,1",
89 "--target-shape",
90 "1,3,65537,1",
91 "--allow-pooling-and-conv-oversizes"
92 ]
93 ],
94 "params": {
95 },
96 "permutes": [
97 "shape",
98 "type",
99 "accum_type"
100 ],
101 "sparsity": {
102 "pad": 17
103 },
104 "profile": [
105 "tosa-mi"
106 ]
107 },
108 "conv2d": {
109 "group": "tensor",
110 "generator_args": [
111 [
112 "--target-dtype",
113 "fp32",
114 "--target-dtype",
115 "fp16",
116 "--target-dtype",
117 "bf16",
118 "--fp-values-range",
119 "-2.0,2.0",
120 "--target-shape",
121 "1,49,42,28",
122 "--target-shape",
123 "1,11,44,13",
124 "--allow-pooling-and-conv-oversizes"
125 ],
126 [
127 "--target-dtype",
128 "fp32",
129 "--fp-values-range",
130 "-2.0,2.0",
131 "--target-shape",
132 "1,65535,4,1",
133 "--target-shape",
134 "1,5,65536,1",
135 "--max-conv-dilation",
136 "1",
137 "--allow-pooling-and-conv-oversizes"
138 ]
139 ],
140 "params": {
141 "shape": [],
142 "type": [],
143 "kernel": [],
144 "stride": [],
145 "pad": [],
146 "dilation": []
147 },
148 "permutes": [
149 "kernel",
150 "shape",
151 "type",
152 "pad"
153 ],
154 "profile": [
155 "tosa-mi"
156 ]
James Ward52ac9d62022-11-16 12:53:49 +0000[diff] [blame^]157 },
158 "max_pool2d": {
159 "group": "tensor",
160 "generator_args": [
161 [
162 "--target-dtype",
163 "fp32",
164 "--target-dtype",
165 "fp16",
166 "--target-dtype",
167 "bf16",
168 "--fp-values-range",
169 "-2.0,2.0",
170 "--target-shape",
171 "1,47,37,25",
172 "--target-shape",
173 "1,15,13,38",
174 "--allow-pooling-and-conv-oversizes"
175 ],
176 [
177 "--target-dtype",
178 "fp32",
179 "--fp-values-range",
180 "-2.0,2.0",
181 "--target-shape",
182 "1,65534,4,1",
183 "--target-shape",
184 "1,6,65536,1",
185 "--max-conv-dilation",
186 "1",
187 "--allow-pooling-and-conv-oversizes"
188 ]
189 ],
190 "params": {
191 "shape": [],
192 "type": [
193 "f32",
194 "f16",
195 "bf16"
196 ],
197 "stride": [],
198 "pad": []
199 },
200 "permutes": [
201 "shape",
202 "type",
203 "pad"
204 ],
205 "sparsity": {
206 "pad": 11
207 },
208 "profile": [
209 "tosa-mi"
210 ]
Jeremy Johnsone4b08ff2022-09-15 10:38:17 +0100[diff] [blame]211 }
212}