MLIR 24.0.0git
TosaComplianceData.h.inc
Go to the documentation of this file.
1// The profile-based compliance content below is auto-generated by the script
2// `tools/genspec.py` in https://github.com/arm/tosa-specification
3profileComplianceMap = {
4 {"tosa.argmax",
5 {{{Profile::pro_int}, {{{i8T, i32T}, SpecificationVersion::V_1_0}}},
6 {{Profile::pro_fp},
7 {{{fp16T, i32T}, SpecificationVersion::V_1_0},
8 {{fp32T, i32T}, SpecificationVersion::V_1_0}}}}},
9 {"tosa.avg_pool2d",
10 {{{Profile::pro_int},
11 {{{i8T, i8T, i8T, i32T, i8T}, SpecificationVersion::V_1_0}}},
12 {{Profile::pro_fp},
13 {{{fp16T, fp16T, fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
14 {{fp16T, fp16T, fp16T, fp32T, fp16T}, SpecificationVersion::V_1_0},
15 {{fp32T, fp32T, fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
16 {"tosa.avg_pool2d_adaptive",
17 {{{Profile::pro_int},
18 {{{i8T, i8T, i8T, i32T, i8T}, SpecificationVersion::V_1_1_DRAFT}}},
19 {{Profile::pro_fp},
20 {{{fp16T, fp16T, fp16T, fp16T, fp16T},
21 SpecificationVersion::V_1_1_DRAFT},
22 {{fp16T, fp16T, fp16T, fp32T, fp16T},
23 SpecificationVersion::V_1_1_DRAFT},
24 {{fp32T, fp32T, fp32T, fp32T, fp32T},
25 SpecificationVersion::V_1_1_DRAFT}}}}},
26 {"tosa.conv2d",
27 {{{Profile::pro_int},
28 {{{i8T, i8T, i32T, i8T, i8T, i32T, i32T}, SpecificationVersion::V_1_0}}},
29 {{Profile::pro_fp},
30 {{{fp16T, fp16T, fp16T, fp16T, fp16T, fp16T, fp16T},
31 SpecificationVersion::V_1_0},
32 {{fp16T, fp16T, fp16T, fp16T, fp16T, fp32T, fp16T},
33 SpecificationVersion::V_1_0},
34 {{fp32T, fp32T, fp32T, fp32T, fp32T, fp32T, fp32T},
35 SpecificationVersion::V_1_0}}}}},
36 {"tosa.conv3d",
37 {{{Profile::pro_int},
38 {{{i8T, i8T, i32T, i8T, i8T, i32T, i32T}, SpecificationVersion::V_1_0}}},
39 {{Profile::pro_fp},
40 {{{fp16T, fp16T, fp16T, fp16T, fp16T, fp16T, fp16T},
41 SpecificationVersion::V_1_0},
42 {{fp16T, fp16T, fp16T, fp16T, fp16T, fp32T, fp16T},
43 SpecificationVersion::V_1_0},
44 {{fp32T, fp32T, fp32T, fp32T, fp32T, fp32T, fp32T},
45 SpecificationVersion::V_1_0}}}}},
46 {"tosa.depthwise_conv2d",
47 {{{Profile::pro_int},
48 {{{i8T, i8T, i32T, i8T, i8T, i32T, i32T}, SpecificationVersion::V_1_0}}},
49 {{Profile::pro_fp},
50 {{{fp16T, fp16T, fp16T, fp16T, fp16T, fp16T, fp16T},
51 SpecificationVersion::V_1_0},
52 {{fp16T, fp16T, fp16T, fp16T, fp16T, fp32T, fp16T},
53 SpecificationVersion::V_1_0},
54 {{fp32T, fp32T, fp32T, fp32T, fp32T, fp32T, fp32T},
55 SpecificationVersion::V_1_0}}}}},
56 {"tosa.matmul",
57 {{{Profile::pro_int},
58 {{{i8T, i8T, i8T, i8T, i32T}, SpecificationVersion::V_1_0}}},
59 {{Profile::pro_fp},
60 {{{fp16T, fp16T, fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
61 {{fp16T, fp16T, fp16T, fp16T, fp32T}, SpecificationVersion::V_1_0},
62 {{fp32T, fp32T, fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
63 {"tosa.matmul_t",
64 {{{Profile::pro_int},
65 {{{i8T, i8T, i8T, i8T, i32T}, SpecificationVersion::V_1_1_DRAFT}}},
66 {{Profile::pro_fp},
67 {{{fp16T, fp16T, fp16T, fp16T, fp16T},
68 SpecificationVersion::V_1_1_DRAFT},
69 {{fp16T, fp16T, fp16T, fp16T, fp32T},
70 SpecificationVersion::V_1_1_DRAFT},
71 {{fp16T, fp32T, fp16T, fp32T, fp32T},
72 SpecificationVersion::V_1_1_DRAFT},
73 {{fp32T, fp16T, fp32T, fp16T, fp32T},
74 SpecificationVersion::V_1_1_DRAFT},
75 {{fp16T, fp16T, fp16T, fp16T, fp32T},
76 SpecificationVersion::V_1_1_DRAFT},
77 {{fp32T, fp32T, fp32T, fp32T, fp32T},
78 SpecificationVersion::V_1_1_DRAFT}}}}},
79 {"tosa.max_pool2d",
80 {{{Profile::pro_int}, {{{i8T, i8T}, SpecificationVersion::V_1_0}}},
81 {{Profile::pro_fp},
82 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
83 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
84 {"tosa.max_pool2d_adaptive",
85 {{{Profile::pro_int}, {{{i8T, i8T}, SpecificationVersion::V_1_1_DRAFT}}},
86 {{Profile::pro_fp},
87 {{{fp16T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
88 {{fp32T, fp32T}, SpecificationVersion::V_1_1_DRAFT}}}}},
89 {"tosa.transpose_conv2d",
90 {{{Profile::pro_int},
91 {{{i8T, i8T, i32T, i8T, i8T, i32T, i32T}, SpecificationVersion::V_1_0}}},
92 {{Profile::pro_fp},
93 {{{fp16T, fp16T, fp16T, fp16T, fp16T, fp16T, fp16T},
94 SpecificationVersion::V_1_0},
95 {{fp16T, fp16T, fp16T, fp16T, fp16T, fp32T, fp16T},
96 SpecificationVersion::V_1_0},
97 {{fp32T, fp32T, fp32T, fp32T, fp32T, fp32T, fp32T},
98 SpecificationVersion::V_1_0}}}}},
99 {"tosa.clamp",
100 {{{Profile::pro_int}, {{{i8T, i8T}, SpecificationVersion::V_1_0}}},
101 {{Profile::pro_fp},
102 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
103 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
104 {"tosa.erf",
105 {{{Profile::pro_fp},
106 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
107 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
108 {"tosa.sigmoid",
109 {{{Profile::pro_fp},
110 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
111 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
112 {"tosa.tanh",
113 {{{Profile::pro_fp},
114 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
115 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
116 {"tosa.add",
117 {{{Profile::pro_int, Profile::pro_fp},
118 {{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}},
119 anyOf},
120 {{Profile::pro_fp},
121 {{{fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
122 {{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
123 {"tosa.arithmetic_right_shift",
124 {{{Profile::pro_int},
125 {{{i8T, i8T, i8T}, SpecificationVersion::V_1_0},
126 {{i16T, i16T, i16T}, SpecificationVersion::V_1_0},
127 {{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}}}},
128 {"tosa.bitwise_and",
129 {{{Profile::pro_int},
130 {{{i8T, i8T, i8T}, SpecificationVersion::V_1_0},
131 {{i16T, i16T, i16T}, SpecificationVersion::V_1_0},
132 {{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}}}},
133 {"tosa.bitwise_or",
134 {{{Profile::pro_int},
135 {{{i8T, i8T, i8T}, SpecificationVersion::V_1_0},
136 {{i16T, i16T, i16T}, SpecificationVersion::V_1_0},
137 {{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}}}},
138 {"tosa.bitwise_xor",
139 {{{Profile::pro_int},
140 {{{i8T, i8T, i8T}, SpecificationVersion::V_1_0},
141 {{i16T, i16T, i16T}, SpecificationVersion::V_1_0},
142 {{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}}}},
143 {"tosa.intdiv",
144 {{{Profile::pro_int, Profile::pro_fp},
145 {{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}},
146 anyOf}}},
147 {"tosa.logical_and",
148 {{{Profile::pro_int, Profile::pro_fp},
149 {{{boolT, boolT, boolT}, SpecificationVersion::V_1_0}},
150 anyOf}}},
151 {"tosa.logical_left_shift",
152 {{{Profile::pro_int, Profile::pro_fp},
153 {{{i8T, i8T, i8T}, SpecificationVersion::V_1_0},
154 {{i16T, i16T, i16T}, SpecificationVersion::V_1_0},
155 {{i32T, i32T, i32T}, SpecificationVersion::V_1_0}},
156 anyOf}}},
157 {"tosa.logical_right_shift",
158 {{{Profile::pro_int, Profile::pro_fp},
159 {{{i8T, i8T, i8T}, SpecificationVersion::V_1_0},
160 {{i16T, i16T, i16T}, SpecificationVersion::V_1_0},
161 {{i32T, i32T, i32T}, SpecificationVersion::V_1_0}},
162 anyOf}}},
163 {"tosa.logical_or",
164 {{{Profile::pro_int, Profile::pro_fp},
165 {{{boolT, boolT, boolT}, SpecificationVersion::V_1_0}},
166 anyOf}}},
167 {"tosa.logical_xor",
168 {{{Profile::pro_int, Profile::pro_fp},
169 {{{boolT, boolT, boolT}, SpecificationVersion::V_1_0}},
170 anyOf}}},
171 {"tosa.maximum",
172 {{{Profile::pro_int}, {{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}},
173 {{Profile::pro_fp},
174 {{{fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
175 {{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
176 {"tosa.minimum",
177 {{{Profile::pro_int}, {{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}},
178 {{Profile::pro_fp},
179 {{{fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
180 {{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
181 {"tosa.mul",
182 {{{Profile::pro_int},
183 {{{i8T, i8T, i32T}, SpecificationVersion::V_1_0},
184 {{i16T, i16T, i32T}, SpecificationVersion::V_1_0}}},
185 {{Profile::pro_int, Profile::pro_fp},
186 {{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}},
187 anyOf},
188 {{Profile::pro_fp},
189 {{{fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
190 {{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
191 {"tosa.pow",
192 {{{Profile::pro_fp},
193 {{{fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
194 {{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
195 {"tosa.sub",
196 {{{Profile::pro_int, Profile::pro_fp},
197 {{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}},
198 anyOf},
199 {{Profile::pro_fp},
200 {{{fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
201 {{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
202 {"tosa.table",
203 {{{Profile::pro_int}, {{{i8T, i8T, i8T}, SpecificationVersion::V_1_0}}}}},
204 {"tosa.abs",
205 {{{Profile::pro_int}, {{{i32T, i32T}, SpecificationVersion::V_1_0}}},
206 {{Profile::pro_fp},
207 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
208 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
209 {"tosa.bitwise_not",
210 {{{Profile::pro_int},
211 {{{i8T, i8T}, SpecificationVersion::V_1_0},
212 {{i16T, i16T}, SpecificationVersion::V_1_0},
213 {{i32T, i32T}, SpecificationVersion::V_1_0}}}}},
214 {"tosa.ceil",
215 {{{Profile::pro_fp},
216 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
217 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
218 {"tosa.clz",
219 {{{Profile::pro_int}, {{{i32T, i32T}, SpecificationVersion::V_1_0}}}}},
220 {"tosa.cos",
221 {{{Profile::pro_fp},
222 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
223 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
224 {"tosa.exp",
225 {{{Profile::pro_fp},
226 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
227 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
228 {"tosa.floor",
229 {{{Profile::pro_fp},
230 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
231 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
232 {"tosa.log",
233 {{{Profile::pro_fp},
234 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
235 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
236 {"tosa.logical_not",
237 {{{Profile::pro_int, Profile::pro_fp},
238 {{{boolT, boolT}, SpecificationVersion::V_1_0}},
239 anyOf}}},
240 {"tosa.negate",
241 {{{Profile::pro_int},
242 {{{i8T, i8T, i8T, i8T}, SpecificationVersion::V_1_0},
243 {{i16T, i16T, i16T, i16T}, SpecificationVersion::V_1_0},
244 {{i32T, i32T, i32T, i32T}, SpecificationVersion::V_1_0}}},
245 {{Profile::pro_fp},
246 {{{fp16T, fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
247 {{fp32T, fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
248 {"tosa.reciprocal",
249 {{{Profile::pro_fp},
250 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
251 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
252 {"tosa.rsqrt",
253 {{{Profile::pro_fp},
254 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
255 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
256 {"tosa.sin",
257 {{{Profile::pro_fp},
258 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
259 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
260 {"tosa.select",
261 {{{Profile::pro_int, Profile::pro_fp},
262 {{{boolT, boolT, boolT}, SpecificationVersion::V_1_0}},
263 anyOf},
264 {{Profile::pro_int},
265 {{{i8T, i8T, i8T}, SpecificationVersion::V_1_0},
266 {{i16T, i16T, i16T}, SpecificationVersion::V_1_0},
267 {{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}},
268 {{Profile::pro_fp},
269 {{{fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
270 {{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
271 {"tosa.equal",
272 {{{Profile::pro_int},
273 {{{i32T, i32T, boolT}, SpecificationVersion::V_1_0}}},
274 {{Profile::pro_fp},
275 {{{fp16T, fp16T, boolT}, SpecificationVersion::V_1_0},
276 {{fp32T, fp32T, boolT}, SpecificationVersion::V_1_0}}}}},
277 {"tosa.greater",
278 {{{Profile::pro_int},
279 {{{i32T, i32T, boolT}, SpecificationVersion::V_1_0}}},
280 {{Profile::pro_fp},
281 {{{fp16T, fp16T, boolT}, SpecificationVersion::V_1_0},
282 {{fp32T, fp32T, boolT}, SpecificationVersion::V_1_0}}}}},
283 {"tosa.greater_equal",
284 {{{Profile::pro_int},
285 {{{i32T, i32T, boolT}, SpecificationVersion::V_1_0}}},
286 {{Profile::pro_fp},
287 {{{fp16T, fp16T, boolT}, SpecificationVersion::V_1_0},
288 {{fp32T, fp32T, boolT}, SpecificationVersion::V_1_0}}}}},
289 {"tosa.reduce_all",
290 {{{Profile::pro_int, Profile::pro_fp},
291 {{{boolT, boolT}, SpecificationVersion::V_1_0}},
292 anyOf}}},
293 {"tosa.reduce_any",
294 {{{Profile::pro_int, Profile::pro_fp},
295 {{{boolT, boolT}, SpecificationVersion::V_1_0}},
296 anyOf}}},
297 {"tosa.reduce_max",
298 {{{Profile::pro_int},
299 {{{i8T, i8T}, SpecificationVersion::V_1_0},
300 {{i16T, i16T}, SpecificationVersion::V_1_0},
301 {{i32T, i32T}, SpecificationVersion::V_1_0}}},
302 {{Profile::pro_fp},
303 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
304 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
305 {"tosa.reduce_min",
306 {{{Profile::pro_int},
307 {{{i8T, i8T}, SpecificationVersion::V_1_0},
308 {{i16T, i16T}, SpecificationVersion::V_1_0},
309 {{i32T, i32T}, SpecificationVersion::V_1_0}}},
310 {{Profile::pro_fp},
311 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
312 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
313 {"tosa.reduce_product",
314 {{{Profile::pro_fp},
315 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
316 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
317 {"tosa.reduce_sum",
318 {{{Profile::pro_int}, {{{i32T, i32T}, SpecificationVersion::V_1_0}}},
319 {{Profile::pro_fp},
320 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
321 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
322 {"tosa.concat",
323 {{{Profile::pro_int, Profile::pro_fp},
324 {{{boolT, boolT}, SpecificationVersion::V_1_0}},
325 anyOf},
326 {{Profile::pro_int},
327 {{{i8T, i8T}, SpecificationVersion::V_1_0},
328 {{i32T, i32T}, SpecificationVersion::V_1_0}}},
329 {{Profile::pro_fp},
330 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
331 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
332 {"tosa.pad",
333 {{{Profile::pro_int, Profile::pro_fp},
334 {{{boolT, boolT, boolT}, SpecificationVersion::V_1_0}},
335 anyOf},
336 {{Profile::pro_int},
337 {{{i8T, i8T, i8T}, SpecificationVersion::V_1_0},
338 {{i16T, i16T, i16T}, SpecificationVersion::V_1_0},
339 {{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}},
340 {{Profile::pro_fp},
341 {{{fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
342 {{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
343 {"tosa.reshape",
344 {{{Profile::pro_int, Profile::pro_fp},
345 {{{boolT, boolT}, SpecificationVersion::V_1_0}},
346 anyOf},
347 {{Profile::pro_int},
348 {{{i8T, i8T}, SpecificationVersion::V_1_0},
349 {{i16T, i16T}, SpecificationVersion::V_1_0},
350 {{i32T, i32T}, SpecificationVersion::V_1_0}}},
351 {{Profile::pro_fp},
352 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
353 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
354 {"tosa.reverse",
355 {{{Profile::pro_int, Profile::pro_fp},
356 {{{boolT, boolT}, SpecificationVersion::V_1_0}},
357 anyOf},
358 {{Profile::pro_int},
359 {{{i8T, i8T}, SpecificationVersion::V_1_0},
360 {{i16T, i16T}, SpecificationVersion::V_1_0},
361 {{i32T, i32T}, SpecificationVersion::V_1_0}}},
362 {{Profile::pro_fp},
363 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
364 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
365 {"tosa.slice",
366 {{{Profile::pro_int, Profile::pro_fp},
367 {{{boolT, boolT}, SpecificationVersion::V_1_0}},
368 anyOf},
369 {{Profile::pro_int},
370 {{{i8T, i8T}, SpecificationVersion::V_1_0},
371 {{i16T, i16T}, SpecificationVersion::V_1_0},
372 {{i32T, i32T}, SpecificationVersion::V_1_0}}},
373 {{Profile::pro_fp},
374 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
375 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
376 {"tosa.tile",
377 {{{Profile::pro_int, Profile::pro_fp},
378 {{{boolT, boolT}, SpecificationVersion::V_1_0}},
379 anyOf},
380 {{Profile::pro_int},
381 {{{i8T, i8T}, SpecificationVersion::V_1_0},
382 {{i16T, i16T}, SpecificationVersion::V_1_0},
383 {{i32T, i32T}, SpecificationVersion::V_1_0}}},
384 {{Profile::pro_fp},
385 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
386 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
387 {"tosa.transpose",
388 {{{Profile::pro_int, Profile::pro_fp},
389 {{{boolT, boolT}, SpecificationVersion::V_1_0}},
390 anyOf},
391 {{Profile::pro_int},
392 {{{i8T, i8T}, SpecificationVersion::V_1_0},
393 {{i16T, i16T}, SpecificationVersion::V_1_0},
394 {{i32T, i32T}, SpecificationVersion::V_1_0}}},
395 {{Profile::pro_fp},
396 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
397 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
398 {"tosa.gather",
399 {{{Profile::pro_int},
400 {{{i8T, i32T, i8T}, SpecificationVersion::V_1_0},
401 {{i16T, i32T, i16T}, SpecificationVersion::V_1_0},
402 {{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}},
403 {{Profile::pro_fp},
404 {{{i8T, i32T, i8T}, SpecificationVersion::V_1_1_DRAFT},
405 {{i16T, i32T, i16T}, SpecificationVersion::V_1_1_DRAFT},
406 {{i32T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT},
407 {{fp16T, i32T, fp16T}, SpecificationVersion::V_1_0},
408 {{fp32T, i32T, fp32T}, SpecificationVersion::V_1_0}}},
409 {{Profile::pro_fp, Profile::pro_int},
410 {{{boolT, i32T, boolT}, SpecificationVersion::V_1_1_DRAFT},
411 {{i8T, i64T, i8T}, SpecificationVersion::V_1_1_DRAFT},
412 {{i16T, i64T, i16T}, SpecificationVersion::V_1_1_DRAFT},
413 {{i32T, i64T, i32T}, SpecificationVersion::V_1_1_DRAFT}},
414 anyOf}}},
415 {"tosa.row_gather",
416 {{{Profile::pro_int},
417 {{{boolT, i32T, boolT}, SpecificationVersion::V_1_1_DRAFT},
418 {{i8T, i32T, i8T}, SpecificationVersion::V_1_1_DRAFT},
419 {{i16T, i32T, i16T}, SpecificationVersion::V_1_1_DRAFT},
420 {{i32T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT},
421 {{i64T, i32T, i64T}, SpecificationVersion::V_1_1_DRAFT},
422 {{boolT, i64T, boolT}, SpecificationVersion::V_1_1_DRAFT},
423 {{i8T, i64T, i8T}, SpecificationVersion::V_1_1_DRAFT},
424 {{i16T, i64T, i16T}, SpecificationVersion::V_1_1_DRAFT},
425 {{i32T, i64T, i32T}, SpecificationVersion::V_1_1_DRAFT},
426 {{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
427 {{Profile::pro_fp},
428 {{{boolT, i32T, boolT}, SpecificationVersion::V_1_1_DRAFT},
429 {{i8T, i32T, i8T}, SpecificationVersion::V_1_1_DRAFT},
430 {{i16T, i32T, i16T}, SpecificationVersion::V_1_1_DRAFT},
431 {{i32T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT},
432 {{i64T, i32T, i64T}, SpecificationVersion::V_1_1_DRAFT},
433 {{fp16T, i32T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
434 {{fp32T, i32T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
435 {{boolT, i64T, boolT}, SpecificationVersion::V_1_1_DRAFT},
436 {{i8T, i64T, i8T}, SpecificationVersion::V_1_1_DRAFT},
437 {{i16T, i64T, i16T}, SpecificationVersion::V_1_1_DRAFT},
438 {{i32T, i64T, i32T}, SpecificationVersion::V_1_1_DRAFT},
439 {{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT},
440 {{fp16T, i64T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
441 {{fp32T, i64T, fp32T}, SpecificationVersion::V_1_1_DRAFT}}}}},
442 {"tosa.row_gather_block_scaled",
443 {{{Profile::pro_int},
444 {{{i8T, i32T, i32T, i8T}, SpecificationVersion::V_1_1_DRAFT},
445 {{i16T, i32T, i32T, i16T}, SpecificationVersion::V_1_1_DRAFT},
446 {{i32T, i32T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT}}},
447 {{Profile::pro_fp},
448 {{{i8T, i32T, i32T, i8T}, SpecificationVersion::V_1_1_DRAFT},
449 {{i16T, i32T, i32T, i16T}, SpecificationVersion::V_1_1_DRAFT},
450 {{i32T, i32T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT},
451 {{fp16T, i32T, i32T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
452 {{fp32T, i32T, i32T, fp32T}, SpecificationVersion::V_1_1_DRAFT}}},
453 {{Profile::pro_fp, Profile::pro_int},
454 {{{boolT, i32T, i32T, boolT}, SpecificationVersion::V_1_1_DRAFT},
455 {{i8T, i64T, i32T, i8T}, SpecificationVersion::V_1_1_DRAFT},
456 {{i16T, i64T, i32T, i16T}, SpecificationVersion::V_1_1_DRAFT},
457 {{i32T, i64T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT},
458 {{i64T, i64T, i32T, i64T}, SpecificationVersion::V_1_1_DRAFT},
459 {{boolT, i64T, i32T, boolT}, SpecificationVersion::V_1_1_DRAFT}},
460 anyOf}}},
461 {"tosa.scatter",
462 {{{Profile::pro_int},
463 {{{i8T, i32T, i8T, i8T}, SpecificationVersion::V_1_0},
464 {{i16T, i32T, i16T, i16T}, SpecificationVersion::V_1_0},
465 {{i32T, i32T, i32T, i32T}, SpecificationVersion::V_1_0}}},
466 {{Profile::pro_fp},
467 {{{i8T, i32T, i8T, i8T}, SpecificationVersion::V_1_1_DRAFT},
468 {{i16T, i32T, i16T, i16T}, SpecificationVersion::V_1_1_DRAFT},
469 {{i32T, i32T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT},
470 {{fp16T, i32T, fp16T, fp16T}, SpecificationVersion::V_1_0},
471 {{fp32T, i32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}},
472 {{Profile::pro_fp, Profile::pro_int},
473 {{{boolT, i32T, boolT, boolT}, SpecificationVersion::V_1_1_DRAFT},
474 {{i8T, i64T, i8T, i8T}, SpecificationVersion::V_1_1_DRAFT},
475 {{i16T, i64T, i16T, i16T}, SpecificationVersion::V_1_1_DRAFT},
476 {{i32T, i64T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT}},
477 anyOf}}},
478 {"tosa.resize",
479 {{{Profile::pro_int},
480 {{{i8T, i32T}, SpecificationVersion::V_1_0},
481 {{i8T, i8T}, SpecificationVersion::V_1_0}}},
482 {{Profile::pro_fp},
483 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
484 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
485 {"tosa.cast",
486 {{{Profile::pro_int},
487 {{{boolT, i8T}, SpecificationVersion::V_1_0},
488 {{boolT, i16T}, SpecificationVersion::V_1_0},
489 {{boolT, i32T}, SpecificationVersion::V_1_0},
490 {{i8T, boolT}, SpecificationVersion::V_1_0},
491 {{i8T, i16T}, SpecificationVersion::V_1_0},
492 {{i8T, i32T}, SpecificationVersion::V_1_0},
493 {{i16T, boolT}, SpecificationVersion::V_1_0},
494 {{i16T, i8T}, SpecificationVersion::V_1_0},
495 {{i16T, i32T}, SpecificationVersion::V_1_0},
496 {{i32T, boolT}, SpecificationVersion::V_1_0},
497 {{i32T, i8T}, SpecificationVersion::V_1_0},
498 {{i32T, i16T}, SpecificationVersion::V_1_0}}},
499 {{Profile::pro_fp},
500 {{{boolT, fp32T}, SpecificationVersion::V_1_1_DRAFT},
501 {{i8T, fp16T}, SpecificationVersion::V_1_0},
502 {{i8T, fp32T}, SpecificationVersion::V_1_0},
503 {{i16T, fp16T}, SpecificationVersion::V_1_0},
504 {{i16T, fp32T}, SpecificationVersion::V_1_0},
505 {{i32T, fp16T}, SpecificationVersion::V_1_0},
506 {{i32T, fp32T}, SpecificationVersion::V_1_0},
507 {{fp16T, i8T}, SpecificationVersion::V_1_0},
508 {{fp16T, i16T}, SpecificationVersion::V_1_0},
509 {{fp16T, i32T}, SpecificationVersion::V_1_0},
510 {{fp16T, fp32T}, SpecificationVersion::V_1_0},
511 {{fp32T, i8T}, SpecificationVersion::V_1_0},
512 {{fp32T, i16T}, SpecificationVersion::V_1_0},
513 {{fp32T, i32T}, SpecificationVersion::V_1_0},
514 {{fp32T, fp16T}, SpecificationVersion::V_1_0},
515 {{fp32T, boolT}, SpecificationVersion::V_1_1_DRAFT}}}}},
516 {"tosa.rescale",
517 {{{Profile::pro_int},
518 {{{i8T, i8T, i8T, i8T}, SpecificationVersion::V_1_0},
519 {{i8T, i8T, i16T, i16T}, SpecificationVersion::V_1_0},
520 {{i8T, i8T, i32T, i32T}, SpecificationVersion::V_1_0},
521 {{i16T, i16T, i8T, i8T}, SpecificationVersion::V_1_0},
522 {{i16T, i16T, i16T, i16T}, SpecificationVersion::V_1_0},
523 {{i16T, i16T, i32T, i32T}, SpecificationVersion::V_1_0},
524 {{i32T, i32T, i8T, i8T}, SpecificationVersion::V_1_0},
525 {{i32T, i32T, i16T, i16T}, SpecificationVersion::V_1_0},
526 {{i32T, i32T, i32T, i32T}, SpecificationVersion::V_1_0}}}}},
527 {"tosa.const",
528 {{{Profile::pro_int, Profile::pro_fp},
529 {{{boolT}, SpecificationVersion::V_1_0},
530 {{i8T}, SpecificationVersion::V_1_0},
531 {{i16T}, SpecificationVersion::V_1_0},
532 {{i32T}, SpecificationVersion::V_1_0}},
533 anyOf},
534 {{Profile::pro_fp},
535 {{{fp16T}, SpecificationVersion::V_1_0},
536 {{fp32T}, SpecificationVersion::V_1_0}}}}},
537 {"tosa.identity",
538 {{{Profile::pro_int, Profile::pro_fp},
539 {{{boolT, boolT}, SpecificationVersion::V_1_0},
540 {{i8T, i8T}, SpecificationVersion::V_1_0},
541 {{i16T, i16T}, SpecificationVersion::V_1_0},
542 {{i32T, i32T}, SpecificationVersion::V_1_0}},
543 anyOf},
544 {{Profile::pro_fp},
545 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
546 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
547 {"tosa.variable",
548 {{{Profile::pro_int}, {{{i8T}, SpecificationVersion::V_1_0}}},
549 {{Profile::pro_fp},
550 {{{fp16T}, SpecificationVersion::V_1_0},
551 {{fp32T}, SpecificationVersion::V_1_0}}}}},
552 {"tosa.variable_write",
553 {{{Profile::pro_int}, {{{i8T}, SpecificationVersion::V_1_0}}},
554 {{Profile::pro_fp},
555 {{{fp16T}, SpecificationVersion::V_1_0},
556 {{fp32T}, SpecificationVersion::V_1_0}}}}},
557 {"tosa.variable_read",
558 {{{Profile::pro_int}, {{{i8T}, SpecificationVersion::V_1_0}}},
559 {{Profile::pro_fp},
560 {{{fp16T}, SpecificationVersion::V_1_0},
561 {{fp32T}, SpecificationVersion::V_1_0}}}}},
562};
563
564extensionComplianceMap = {
565 {"tosa.argmax",
566 {{{Extension::int16}, {{{i16T, i32T}, SpecificationVersion::V_1_0}}},
567 {{Extension::int64},
568 {{{i8T, i64T}, SpecificationVersion::V_1_1_DRAFT},
569 {{i16T, i64T}, SpecificationVersion::V_1_1_DRAFT},
570 {{i32T, i64T}, SpecificationVersion::V_1_1_DRAFT},
571 {{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT},
572 {{fp16T, i64T}, SpecificationVersion::V_1_1_DRAFT},
573 {{fp32T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
574 {{Extension::fp8e4m3}, {{{fp8e4m3T, i32T}, SpecificationVersion::V_1_0}}},
575 {{Extension::fp8e5m2}, {{{fp8e5m2T, i32T}, SpecificationVersion::V_1_0}}},
576 {{Extension::bf16}, {{{bf16T, i32T}, SpecificationVersion::V_1_0}}},
577 {{Extension::fp8e4m3, Extension::int64},
578 {{{fp8e4m3T, i64T}, SpecificationVersion::V_1_1_DRAFT}},
579 allOf},
580 {{Extension::fp8e5m2, Extension::int64},
581 {{{fp8e5m2T, i64T}, SpecificationVersion::V_1_1_DRAFT}},
582 allOf},
583 {{Extension::bf16, Extension::int64},
584 {{{bf16T, i64T}, SpecificationVersion::V_1_1_DRAFT}},
585 allOf}}},
586 {"tosa.avg_pool2d",
587 {{{Extension::int16},
588 {{{i16T, i16T, i16T, i32T, i16T}, SpecificationVersion::V_1_0}}},
589 {{Extension::fp8e4m3},
590 {{{fp8e4m3T, fp8e4m3T, fp8e4m3T, fp16T, fp8e4m3T},
591 SpecificationVersion::V_1_0}}},
592 {{Extension::fp8e5m2},
593 {{{fp8e5m2T, fp8e5m2T, fp8e5m2T, fp16T, fp8e5m2T},
594 SpecificationVersion::V_1_0}}},
595 {{Extension::bf16},
596 {{{bf16T, bf16T, bf16T, fp32T, bf16T}, SpecificationVersion::V_1_0}}}}},
597 {"tosa.avg_pool2d_adaptive",
598 {{{Extension::int16},
599 {{{i16T, i16T, i16T, i32T, i16T}, SpecificationVersion::V_1_1_DRAFT}}},
600 {{Extension::fp8e4m3},
601 {{{fp8e4m3T, fp8e4m3T, fp8e4m3T, fp16T, fp8e4m3T},
602 SpecificationVersion::V_1_1_DRAFT}}},
603 {{Extension::fp8e5m2},
604 {{{fp8e5m2T, fp8e5m2T, fp8e5m2T, fp16T, fp8e5m2T},
605 SpecificationVersion::V_1_1_DRAFT}}},
606 {{Extension::bf16},
607 {{{bf16T, bf16T, bf16T, fp32T, bf16T},
608 SpecificationVersion::V_1_1_DRAFT}}}}},
609 {"tosa.conv2d",
610 {{{Extension::int4},
611 {{{i8T, i4T, i32T, i8T, i4T, i32T, i32T}, SpecificationVersion::V_1_0}}},
612 {{Extension::int16},
613 {{{i16T, i8T, i48T, i16T, i8T, i48T, i48T},
614 SpecificationVersion::V_1_0}}},
615 {{Extension::fp8e4m3},
616 {{{fp8e4m3T, fp8e4m3T, fp16T, fp8e4m3T, fp8e4m3T, fp16T, fp16T},
617 SpecificationVersion::V_1_0},
618 {{fp8e4m3T, fp8e4m3T, fp32T, fp8e4m3T, fp8e4m3T, fp32T, fp32T},
619 SpecificationVersion::V_1_1_DRAFT}}},
620 {{Extension::fp8e5m2},
621 {{{fp8e5m2T, fp8e5m2T, fp16T, fp8e5m2T, fp8e5m2T, fp16T, fp16T},
622 SpecificationVersion::V_1_0},
623 {{fp8e5m2T, fp8e5m2T, fp32T, fp8e5m2T, fp8e5m2T, fp32T, fp32T},
624 SpecificationVersion::V_1_1_DRAFT}}},
625 {{Extension::bf16},
626 {{{bf16T, bf16T, bf16T, bf16T, bf16T, fp32T, bf16T},
627 SpecificationVersion::V_1_0}}}}},
628 {"tosa.conv2d_block_scaled",
629 {{{Extension::mxfp_conv},
630 {{{fp4e2m1T, fp8ue8m0T, fp4e2m1T, fp8ue8m0T, fp32T, fp32T},
631 SpecificationVersion::V_1_1_DRAFT},
632 {{fp6e2m3T, fp8ue8m0T, fp6e2m3T, fp8ue8m0T, fp32T, fp32T},
633 SpecificationVersion::V_1_1_DRAFT},
634 {{fp6e3m2T, fp8ue8m0T, fp6e3m2T, fp8ue8m0T, fp32T, fp32T},
635 SpecificationVersion::V_1_1_DRAFT},
636 {{fp8e4m3T, fp8ue8m0T, fp8e4m3T, fp8ue8m0T, fp32T, fp32T},
637 SpecificationVersion::V_1_1_DRAFT},
638 {{fp8e5m2T, fp8ue8m0T, fp8e5m2T, fp8ue8m0T, fp32T, fp32T},
639 SpecificationVersion::V_1_1_DRAFT}}}}},
640 {"tosa.conv3d",
641 {{{Extension::int4},
642 {{{i8T, i4T, i32T, i8T, i4T, i32T, i32T}, SpecificationVersion::V_1_0}}},
643 {{Extension::int16},
644 {{{i16T, i8T, i48T, i16T, i8T, i48T, i48T},
645 SpecificationVersion::V_1_0}}},
646 {{Extension::fp8e4m3},
647 {{{fp8e4m3T, fp8e4m3T, fp16T, fp8e4m3T, fp8e4m3T, fp16T, fp16T},
648 SpecificationVersion::V_1_0},
649 {{fp8e4m3T, fp8e4m3T, fp32T, fp8e4m3T, fp8e4m3T, fp32T, fp32T},
650 SpecificationVersion::V_1_1_DRAFT}}},
651 {{Extension::fp8e5m2},
652 {{{fp8e5m2T, fp8e5m2T, fp16T, fp8e5m2T, fp8e5m2T, fp16T, fp16T},
653 SpecificationVersion::V_1_0},
654 {{fp8e5m2T, fp8e5m2T, fp32T, fp8e5m2T, fp8e5m2T, fp32T, fp32T},
655 SpecificationVersion::V_1_1_DRAFT}}},
656 {{Extension::bf16},
657 {{{bf16T, bf16T, bf16T, bf16T, bf16T, fp32T, bf16T},
658 SpecificationVersion::V_1_0}}}}},
659 {"tosa.depthwise_conv2d",
660 {{{Extension::int4},
661 {{{i8T, i4T, i32T, i8T, i4T, i32T, i32T}, SpecificationVersion::V_1_0}}},
662 {{Extension::int16},
663 {{{i16T, i8T, i48T, i16T, i8T, i48T, i48T},
664 SpecificationVersion::V_1_0}}},
665 {{Extension::fp8e4m3},
666 {{{fp8e4m3T, fp8e4m3T, fp16T, fp8e4m3T, fp8e4m3T, fp16T, fp16T},
667 SpecificationVersion::V_1_0},
668 {{fp8e4m3T, fp8e4m3T, fp32T, fp8e4m3T, fp8e4m3T, fp32T, fp32T},
669 SpecificationVersion::V_1_1_DRAFT}}},
670 {{Extension::fp8e5m2},
671 {{{fp8e5m2T, fp8e5m2T, fp16T, fp8e5m2T, fp8e5m2T, fp16T, fp16T},
672 SpecificationVersion::V_1_0},
673 {{fp8e5m2T, fp8e5m2T, fp32T, fp8e5m2T, fp8e5m2T, fp32T, fp32T},
674 SpecificationVersion::V_1_1_DRAFT}}},
675 {{Extension::bf16},
676 {{{bf16T, bf16T, bf16T, bf16T, bf16T, fp32T, bf16T},
677 SpecificationVersion::V_1_0}}}}},
678 {"tosa.fft2d",
679 {{{Extension::fft},
680 {{{fp32T, fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
681 {"tosa.matmul",
682 {{{Extension::int16},
683 {{{i16T, i16T, i16T, i16T, i48T}, SpecificationVersion::V_1_0}}},
684 {{Extension::fp8e4m3},
685 {{{fp8e4m3T, fp8e4m3T, fp8e4m3T, fp8e4m3T, fp16T},
686 SpecificationVersion::V_1_0},
687 {{fp8e4m3T, fp8e4m3T, fp8e4m3T, fp8e4m3T, fp32T},
688 SpecificationVersion::V_1_1_DRAFT}}},
689 {{Extension::fp8e5m2},
690 {{{fp8e5m2T, fp8e5m2T, fp8e5m2T, fp8e5m2T, fp16T},
691 SpecificationVersion::V_1_0},
692 {{fp8e5m2T, fp8e5m2T, fp8e5m2T, fp8e5m2T, fp32T},
693 SpecificationVersion::V_1_1_DRAFT}}},
694 {{Extension::fp8e4m3, Extension::fp8e5m2},
695 {{{fp8e4m3T, fp8e5m2T, fp8e4m3T, fp8e5m2T, fp16T},
696 SpecificationVersion::V_1_1_DRAFT},
697 {{fp8e4m3T, fp8e5m2T, fp8e4m3T, fp8e5m2T, fp32T},
698 SpecificationVersion::V_1_1_DRAFT},
699 {{fp8e5m2T, fp8e4m3T, fp8e5m2T, fp8e4m3T, fp16T},
700 SpecificationVersion::V_1_1_DRAFT},
701 {{fp8e5m2T, fp8e4m3T, fp8e5m2T, fp8e4m3T, fp32T},
702 SpecificationVersion::V_1_1_DRAFT}},
703 allOf},
704 {{Extension::bf16},
705 {{{bf16T, bf16T, bf16T, bf16T, fp32T}, SpecificationVersion::V_1_0}}}}},
706 {"tosa.matmul_t",
707 {{{Extension::int16},
708 {{{i16T, i16T, i16T, i16T, i48T}, SpecificationVersion::V_1_1_DRAFT}}},
709 {{Extension::fp8e4m3},
710 {{{fp8e4m3T, fp8e4m3T, fp8e4m3T, fp8e4m3T, fp16T},
711 SpecificationVersion::V_1_1_DRAFT},
712 {{fp8e4m3T, fp16T, fp8e4m3T, fp16T, fp16T},
713 SpecificationVersion::V_1_1_DRAFT},
714 {{fp16T, fp8e4m3T, fp16T, fp8e4m3T, fp16T},
715 SpecificationVersion::V_1_1_DRAFT},
716 {{fp8e4m3T, fp8e4m3T, fp8e4m3T, fp8e4m3T, fp32T},
717 SpecificationVersion::V_1_1_DRAFT},
718 {{fp8e4m3T, fp16T, fp8e4m3T, fp16T, fp32T},
719 SpecificationVersion::V_1_1_DRAFT},
720 {{fp8e4m3T, fp32T, fp8e4m3T, fp32T, fp32T},
721 SpecificationVersion::V_1_1_DRAFT},
722 {{fp16T, fp8e4m3T, fp16T, fp8e4m3T, fp32T},
723 SpecificationVersion::V_1_1_DRAFT},
724 {{fp32T, fp8e4m3T, fp32T, fp8e4m3T, fp32T},
725 SpecificationVersion::V_1_1_DRAFT}}},
726 {{Extension::fp8e4m3, Extension::fp8e5m2},
727 {{{fp8e4m3T, fp8e5m2T, fp8e4m3T, fp8e5m2T, fp16T},
728 SpecificationVersion::V_1_1_DRAFT},
729 {{fp8e5m2T, fp8e4m3T, fp8e5m2T, fp8e4m3T, fp16T},
730 SpecificationVersion::V_1_1_DRAFT},
731 {{fp8e4m3T, fp8e5m2T, fp8e4m3T, fp8e5m2T, fp32T},
732 SpecificationVersion::V_1_1_DRAFT},
733 {{fp8e5m2T, fp8e4m3T, fp8e5m2T, fp8e4m3T, fp32T},
734 SpecificationVersion::V_1_1_DRAFT}},
735 allOf},
736 {{Extension::bf16, Extension::fp8e4m3},
737 {{{fp8e4m3T, bf16T, fp8e4m3T, bf16T, fp16T},
738 SpecificationVersion::V_1_1_DRAFT},
739 {{bf16T, fp8e4m3T, bf16T, fp8e4m3T, fp16T},
740 SpecificationVersion::V_1_1_DRAFT},
741 {{fp8e4m3T, bf16T, fp8e4m3T, bf16T, fp32T},
742 SpecificationVersion::V_1_1_DRAFT},
743 {{bf16T, fp8e4m3T, bf16T, fp8e4m3T, fp32T},
744 SpecificationVersion::V_1_1_DRAFT}},
745 allOf},
746 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp8e4m3},
747 {{{fp8e4m3T, bs32_fp8ue8m0_fp8e4m3T, fp8e4m3T, fp32T, fp16T},
748 SpecificationVersion::V_1_1_DRAFT},
749 {{bs32_fp8ue8m0_fp8e4m3T, fp8e4m3T, fp32T, fp8e4m3T, fp16T},
750 SpecificationVersion::V_1_1_DRAFT},
751 {{fp8e4m3T, bs32_fp8ue8m0_fp8e4m3T, fp8e4m3T, fp32T, fp32T},
752 SpecificationVersion::V_1_1_DRAFT},
753 {{bs32_fp8ue8m0_fp8e4m3T, fp8e4m3T, fp32T, fp8e4m3T, fp32T},
754 SpecificationVersion::V_1_1_DRAFT}},
755 allOf},
756 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp8e5m2},
757 {{{fp8e4m3T, bs32_fp8ue8m0_fp8e5m2T, fp8e4m3T, fp32T, fp16T},
758 SpecificationVersion::V_1_1_DRAFT},
759 {{bs32_fp8ue8m0_fp8e5m2T, fp8e4m3T, fp32T, fp8e4m3T, fp16T},
760 SpecificationVersion::V_1_1_DRAFT},
761 {{fp8e4m3T, bs32_fp8ue8m0_fp8e5m2T, fp8e4m3T, fp32T, fp32T},
762 SpecificationVersion::V_1_1_DRAFT},
763 {{bs32_fp8ue8m0_fp8e5m2T, fp8e4m3T, fp32T, fp8e4m3T, fp32T},
764 SpecificationVersion::V_1_1_DRAFT}},
765 allOf},
766 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp6e3m2},
767 {{{fp8e4m3T, bs32_fp8ue8m0_fp6e3m2T, fp8e4m3T, fp32T, fp16T},
768 SpecificationVersion::V_1_1_DRAFT},
769 {{bs32_fp8ue8m0_fp6e3m2T, fp8e4m3T, fp32T, fp8e4m3T, fp16T},
770 SpecificationVersion::V_1_1_DRAFT},
771 {{fp8e4m3T, bs32_fp8ue8m0_fp6e3m2T, fp8e4m3T, fp32T, fp32T},
772 SpecificationVersion::V_1_1_DRAFT},
773 {{bs32_fp8ue8m0_fp6e3m2T, fp8e4m3T, fp32T, fp8e4m3T, fp32T},
774 SpecificationVersion::V_1_1_DRAFT}},
775 allOf},
776 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp6e2m3},
777 {{{fp8e4m3T, bs32_fp8ue8m0_fp6e2m3T, fp8e4m3T, fp32T, fp16T},
778 SpecificationVersion::V_1_1_DRAFT},
779 {{bs32_fp8ue8m0_fp6e2m3T, fp8e4m3T, fp32T, fp8e4m3T, fp16T},
780 SpecificationVersion::V_1_1_DRAFT},
781 {{fp8e4m3T, bs32_fp8ue8m0_fp6e2m3T, fp8e4m3T, fp32T, fp32T},
782 SpecificationVersion::V_1_1_DRAFT},
783 {{bs32_fp8ue8m0_fp6e2m3T, fp8e4m3T, fp32T, fp8e4m3T, fp32T},
784 SpecificationVersion::V_1_1_DRAFT}},
785 allOf},
786 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp4e2m1},
787 {{{fp8e4m3T, bs32_fp8ue8m0_fp4e2m1T, fp8e4m3T, fp32T, fp16T},
788 SpecificationVersion::V_1_1_DRAFT},
789 {{bs32_fp8ue8m0_fp4e2m1T, fp8e4m3T, fp32T, fp8e4m3T, fp16T},
790 SpecificationVersion::V_1_1_DRAFT},
791 {{fp8e4m3T, bs32_fp8ue8m0_fp4e2m1T, fp8e4m3T, fp32T, fp32T},
792 SpecificationVersion::V_1_1_DRAFT},
793 {{bs32_fp8ue8m0_fp4e2m1T, fp8e4m3T, fp32T, fp8e4m3T, fp32T},
794 SpecificationVersion::V_1_1_DRAFT}},
795 allOf},
796 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_int8},
797 {{{fp8e4m3T, bs32_fp8ue8m0_mxint8T, fp8e4m3T, fp32T, fp16T},
798 SpecificationVersion::V_1_1_DRAFT},
799 {{bs32_fp8ue8m0_mxint8T, fp8e4m3T, fp32T, fp8e4m3T, fp16T},
800 SpecificationVersion::V_1_1_DRAFT},
801 {{fp8e4m3T, bs32_fp8ue8m0_mxint8T, fp8e4m3T, fp32T, fp32T},
802 SpecificationVersion::V_1_1_DRAFT},
803 {{bs32_fp8ue8m0_mxint8T, fp8e4m3T, fp32T, fp8e4m3T, fp32T},
804 SpecificationVersion::V_1_1_DRAFT}},
805 allOf},
806 {{Extension::fp8e5m2},
807 {{{fp8e5m2T, fp8e5m2T, fp8e5m2T, fp8e5m2T, fp16T},
808 SpecificationVersion::V_1_1_DRAFT},
809 {{fp8e5m2T, fp16T, fp8e5m2T, fp16T, fp16T},
810 SpecificationVersion::V_1_1_DRAFT},
811 {{fp16T, fp8e5m2T, fp16T, fp8e5m2T, fp16T},
812 SpecificationVersion::V_1_1_DRAFT},
813 {{fp8e5m2T, fp8e5m2T, fp8e5m2T, fp8e5m2T, fp32T},
814 SpecificationVersion::V_1_1_DRAFT},
815 {{fp8e5m2T, fp16T, fp8e5m2T, fp16T, fp32T},
816 SpecificationVersion::V_1_1_DRAFT},
817 {{fp8e5m2T, fp32T, fp8e5m2T, fp32T, fp32T},
818 SpecificationVersion::V_1_1_DRAFT},
819 {{fp16T, fp8e5m2T, fp16T, fp8e5m2T, fp32T},
820 SpecificationVersion::V_1_1_DRAFT},
821 {{fp32T, fp8e5m2T, fp32T, fp8e5m2T, fp32T},
822 SpecificationVersion::V_1_1_DRAFT}}},
823 {{Extension::bf16, Extension::fp8e5m2},
824 {{{fp8e5m2T, bf16T, fp8e5m2T, bf16T, fp16T},
825 SpecificationVersion::V_1_1_DRAFT},
826 {{bf16T, fp8e5m2T, bf16T, fp8e5m2T, fp16T},
827 SpecificationVersion::V_1_1_DRAFT},
828 {{fp8e5m2T, bf16T, fp8e5m2T, bf16T, fp32T},
829 SpecificationVersion::V_1_1_DRAFT},
830 {{bf16T, fp8e5m2T, bf16T, fp8e5m2T, fp32T},
831 SpecificationVersion::V_1_1_DRAFT}},
832 allOf},
833 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp8e4m3},
834 {{{fp8e5m2T, bs32_fp8ue8m0_fp8e4m3T, fp8e5m2T, fp32T, fp16T},
835 SpecificationVersion::V_1_1_DRAFT},
836 {{bs32_fp8ue8m0_fp8e4m3T, fp8e5m2T, fp32T, fp8e5m2T, fp16T},
837 SpecificationVersion::V_1_1_DRAFT},
838 {{fp8e5m2T, bs32_fp8ue8m0_fp8e4m3T, fp8e5m2T, fp32T, fp32T},
839 SpecificationVersion::V_1_1_DRAFT},
840 {{bs32_fp8ue8m0_fp8e4m3T, fp8e5m2T, fp32T, fp8e5m2T, fp32T},
841 SpecificationVersion::V_1_1_DRAFT}},
842 allOf},
843 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp8e5m2},
844 {{{fp8e5m2T, bs32_fp8ue8m0_fp8e5m2T, fp8e5m2T, fp32T, fp16T},
845 SpecificationVersion::V_1_1_DRAFT},
846 {{bs32_fp8ue8m0_fp8e5m2T, fp8e5m2T, fp32T, fp8e5m2T, fp16T},
847 SpecificationVersion::V_1_1_DRAFT},
848 {{fp8e5m2T, bs32_fp8ue8m0_fp8e5m2T, fp8e5m2T, fp32T, fp32T},
849 SpecificationVersion::V_1_1_DRAFT},
850 {{bs32_fp8ue8m0_fp8e5m2T, fp8e5m2T, fp32T, fp8e5m2T, fp32T},
851 SpecificationVersion::V_1_1_DRAFT}},
852 allOf},
853 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp6e3m2},
854 {{{fp8e5m2T, bs32_fp8ue8m0_fp6e3m2T, fp8e5m2T, fp32T, fp16T},
855 SpecificationVersion::V_1_1_DRAFT},
856 {{bs32_fp8ue8m0_fp6e3m2T, fp8e5m2T, fp32T, fp8e5m2T, fp16T},
857 SpecificationVersion::V_1_1_DRAFT},
858 {{fp8e5m2T, bs32_fp8ue8m0_fp6e3m2T, fp8e5m2T, fp32T, fp32T},
859 SpecificationVersion::V_1_1_DRAFT},
860 {{bs32_fp8ue8m0_fp6e3m2T, fp8e5m2T, fp32T, fp8e5m2T, fp32T},
861 SpecificationVersion::V_1_1_DRAFT}},
862 allOf},
863 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp6e2m3},
864 {{{fp8e5m2T, bs32_fp8ue8m0_fp6e2m3T, fp8e5m2T, fp32T, fp16T},
865 SpecificationVersion::V_1_1_DRAFT},
866 {{bs32_fp8ue8m0_fp6e2m3T, fp8e5m2T, fp32T, fp8e5m2T, fp16T},
867 SpecificationVersion::V_1_1_DRAFT},
868 {{fp8e5m2T, bs32_fp8ue8m0_fp6e2m3T, fp8e5m2T, fp32T, fp32T},
869 SpecificationVersion::V_1_1_DRAFT},
870 {{bs32_fp8ue8m0_fp6e2m3T, fp8e5m2T, fp32T, fp8e5m2T, fp32T},
871 SpecificationVersion::V_1_1_DRAFT}},
872 allOf},
873 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp4e2m1},
874 {{{fp8e5m2T, bs32_fp8ue8m0_fp4e2m1T, fp8e5m2T, fp32T, fp16T},
875 SpecificationVersion::V_1_1_DRAFT},
876 {{bs32_fp8ue8m0_fp4e2m1T, fp8e5m2T, fp32T, fp8e5m2T, fp16T},
877 SpecificationVersion::V_1_1_DRAFT},
878 {{fp8e5m2T, bs32_fp8ue8m0_fp4e2m1T, fp8e5m2T, fp32T, fp32T},
879 SpecificationVersion::V_1_1_DRAFT},
880 {{bs32_fp8ue8m0_fp4e2m1T, fp8e5m2T, fp32T, fp8e5m2T, fp32T},
881 SpecificationVersion::V_1_1_DRAFT}},
882 allOf},
883 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_int8},
884 {{{fp8e5m2T, bs32_fp8ue8m0_mxint8T, fp8e5m2T, fp32T, fp16T},
885 SpecificationVersion::V_1_1_DRAFT},
886 {{bs32_fp8ue8m0_mxint8T, fp8e5m2T, fp32T, fp8e5m2T, fp16T},
887 SpecificationVersion::V_1_1_DRAFT},
888 {{fp8e5m2T, bs32_fp8ue8m0_mxint8T, fp8e5m2T, fp32T, fp32T},
889 SpecificationVersion::V_1_1_DRAFT},
890 {{bs32_fp8ue8m0_mxint8T, fp8e5m2T, fp32T, fp8e5m2T, fp32T},
891 SpecificationVersion::V_1_1_DRAFT}},
892 allOf},
893 {{Extension::bf16},
894 {{{fp16T, bf16T, fp16T, bf16T, fp16T},
895 SpecificationVersion::V_1_1_DRAFT},
896 {{bf16T, fp16T, bf16T, fp16T, fp16T},
897 SpecificationVersion::V_1_1_DRAFT},
898 {{bf16T, bf16T, bf16T, bf16T, fp16T},
899 SpecificationVersion::V_1_1_DRAFT},
900 {{fp16T, bf16T, fp16T, bf16T, fp32T},
901 SpecificationVersion::V_1_1_DRAFT},
902 {{bf16T, fp16T, bf16T, fp16T, fp32T},
903 SpecificationVersion::V_1_1_DRAFT},
904 {{bf16T, fp32T, bf16T, fp32T, fp32T},
905 SpecificationVersion::V_1_1_DRAFT},
906 {{fp32T, bf16T, fp32T, bf16T, fp32T},
907 SpecificationVersion::V_1_1_DRAFT},
908 {{bf16T, bf16T, bf16T, bf16T, fp32T},
909 SpecificationVersion::V_1_1_DRAFT}}},
910 {{Extension::mx_common, Extension::mx_fp8e4m3},
911 {{{fp16T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp32T, fp16T},
912 SpecificationVersion::V_1_1_DRAFT},
913 {{bs32_fp8ue8m0_fp8e4m3T, fp16T, fp32T, fp16T, fp16T},
914 SpecificationVersion::V_1_1_DRAFT},
915 {{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp16T},
916 SpecificationVersion::V_1_1_DRAFT},
917 {{fp16T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp32T, fp32T},
918 SpecificationVersion::V_1_1_DRAFT},
919 {{fp32T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp32T},
920 SpecificationVersion::V_1_1_DRAFT},
921 {{bs32_fp8ue8m0_fp8e4m3T, fp16T, fp32T, fp16T, fp32T},
922 SpecificationVersion::V_1_1_DRAFT},
923 {{bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp32T, fp32T},
924 SpecificationVersion::V_1_1_DRAFT},
925 {{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp32T},
926 SpecificationVersion::V_1_1_DRAFT}},
927 allOf},
928 {{Extension::mx_common, Extension::mx_fp8e5m2},
929 {{{fp16T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp32T, fp16T},
930 SpecificationVersion::V_1_1_DRAFT},
931 {{bs32_fp8ue8m0_fp8e5m2T, fp16T, fp32T, fp16T, fp16T},
932 SpecificationVersion::V_1_1_DRAFT},
933 {{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp16T},
934 SpecificationVersion::V_1_1_DRAFT},
935 {{fp16T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp32T, fp32T},
936 SpecificationVersion::V_1_1_DRAFT},
937 {{fp32T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp32T},
938 SpecificationVersion::V_1_1_DRAFT},
939 {{bs32_fp8ue8m0_fp8e5m2T, fp16T, fp32T, fp16T, fp32T},
940 SpecificationVersion::V_1_1_DRAFT},
941 {{bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp32T, fp32T},
942 SpecificationVersion::V_1_1_DRAFT},
943 {{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp32T},
944 SpecificationVersion::V_1_1_DRAFT}},
945 allOf},
946 {{Extension::mx_common, Extension::mx_fp6e3m2},
947 {{{fp16T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp32T, fp16T},
948 SpecificationVersion::V_1_1_DRAFT},
949 {{bs32_fp8ue8m0_fp6e3m2T, fp16T, fp32T, fp16T, fp16T},
950 SpecificationVersion::V_1_1_DRAFT},
951 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp16T},
952 SpecificationVersion::V_1_1_DRAFT},
953 {{fp16T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp32T, fp32T},
954 SpecificationVersion::V_1_1_DRAFT},
955 {{fp32T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp32T},
956 SpecificationVersion::V_1_1_DRAFT},
957 {{bs32_fp8ue8m0_fp6e3m2T, fp16T, fp32T, fp16T, fp32T},
958 SpecificationVersion::V_1_1_DRAFT},
959 {{bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp32T, fp32T},
960 SpecificationVersion::V_1_1_DRAFT},
961 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp32T},
962 SpecificationVersion::V_1_1_DRAFT}},
963 allOf},
964 {{Extension::mx_common, Extension::mx_fp6e2m3},
965 {{{fp16T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp32T, fp16T},
966 SpecificationVersion::V_1_1_DRAFT},
967 {{bs32_fp8ue8m0_fp6e2m3T, fp16T, fp32T, fp16T, fp16T},
968 SpecificationVersion::V_1_1_DRAFT},
969 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp16T},
970 SpecificationVersion::V_1_1_DRAFT},
971 {{fp16T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp32T, fp32T},
972 SpecificationVersion::V_1_1_DRAFT},
973 {{fp32T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp32T},
974 SpecificationVersion::V_1_1_DRAFT},
975 {{bs32_fp8ue8m0_fp6e2m3T, fp16T, fp32T, fp16T, fp32T},
976 SpecificationVersion::V_1_1_DRAFT},
977 {{bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp32T, fp32T},
978 SpecificationVersion::V_1_1_DRAFT},
979 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp32T},
980 SpecificationVersion::V_1_1_DRAFT}},
981 allOf},
982 {{Extension::mx_common, Extension::mx_fp4e2m1},
983 {{{fp16T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp32T, fp16T},
984 SpecificationVersion::V_1_1_DRAFT},
985 {{bs32_fp8ue8m0_fp4e2m1T, fp16T, fp32T, fp16T, fp16T},
986 SpecificationVersion::V_1_1_DRAFT},
987 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp16T},
988 SpecificationVersion::V_1_1_DRAFT},
989 {{fp16T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp32T, fp32T},
990 SpecificationVersion::V_1_1_DRAFT},
991 {{fp32T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp32T},
992 SpecificationVersion::V_1_1_DRAFT},
993 {{bs32_fp8ue8m0_fp4e2m1T, fp16T, fp32T, fp16T, fp32T},
994 SpecificationVersion::V_1_1_DRAFT},
995 {{bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp32T, fp32T},
996 SpecificationVersion::V_1_1_DRAFT},
997 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp32T},
998 SpecificationVersion::V_1_1_DRAFT}},
999 allOf},
1000 {{Extension::mx_common, Extension::mx_int8},
1001 {{{fp16T, bs32_fp8ue8m0_mxint8T, fp16T, fp32T, fp16T},
1002 SpecificationVersion::V_1_1_DRAFT},
1003 {{bs32_fp8ue8m0_mxint8T, fp16T, fp32T, fp16T, fp16T},
1004 SpecificationVersion::V_1_1_DRAFT},
1005 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp16T},
1006 SpecificationVersion::V_1_1_DRAFT},
1007 {{fp16T, bs32_fp8ue8m0_mxint8T, fp16T, fp32T, fp32T},
1008 SpecificationVersion::V_1_1_DRAFT},
1009 {{fp32T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp32T},
1010 SpecificationVersion::V_1_1_DRAFT},
1011 {{bs32_fp8ue8m0_mxint8T, fp16T, fp32T, fp16T, fp32T},
1012 SpecificationVersion::V_1_1_DRAFT},
1013 {{bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp32T, fp32T},
1014 SpecificationVersion::V_1_1_DRAFT},
1015 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp32T},
1016 SpecificationVersion::V_1_1_DRAFT}},
1017 allOf},
1018 {{Extension::bf16, Extension::mx_common, Extension::mx_fp8e4m3},
1019 {{{bf16T, bs32_fp8ue8m0_fp8e4m3T, bf16T, fp32T, fp16T},
1020 SpecificationVersion::V_1_1_DRAFT},
1021 {{bs32_fp8ue8m0_fp8e4m3T, bf16T, fp32T, bf16T, fp16T},
1022 SpecificationVersion::V_1_1_DRAFT},
1023 {{bf16T, bs32_fp8ue8m0_fp8e4m3T, bf16T, fp32T, fp32T},
1024 SpecificationVersion::V_1_1_DRAFT},
1025 {{bs32_fp8ue8m0_fp8e4m3T, bf16T, fp32T, bf16T, fp32T},
1026 SpecificationVersion::V_1_1_DRAFT}},
1027 allOf},
1028 {{Extension::bf16, Extension::mx_common, Extension::mx_fp8e5m2},
1029 {{{bf16T, bs32_fp8ue8m0_fp8e5m2T, bf16T, fp32T, fp16T},
1030 SpecificationVersion::V_1_1_DRAFT},
1031 {{bs32_fp8ue8m0_fp8e5m2T, bf16T, fp32T, bf16T, fp16T},
1032 SpecificationVersion::V_1_1_DRAFT},
1033 {{bf16T, bs32_fp8ue8m0_fp8e5m2T, bf16T, fp32T, fp32T},
1034 SpecificationVersion::V_1_1_DRAFT},
1035 {{bs32_fp8ue8m0_fp8e5m2T, bf16T, fp32T, bf16T, fp32T},
1036 SpecificationVersion::V_1_1_DRAFT}},
1037 allOf},
1038 {{Extension::bf16, Extension::mx_common, Extension::mx_fp6e3m2},
1039 {{{bf16T, bs32_fp8ue8m0_fp6e3m2T, bf16T, fp32T, fp16T},
1040 SpecificationVersion::V_1_1_DRAFT},
1041 {{bs32_fp8ue8m0_fp6e3m2T, bf16T, fp32T, bf16T, fp16T},
1042 SpecificationVersion::V_1_1_DRAFT},
1043 {{bf16T, bs32_fp8ue8m0_fp6e3m2T, bf16T, fp32T, fp32T},
1044 SpecificationVersion::V_1_1_DRAFT},
1045 {{bs32_fp8ue8m0_fp6e3m2T, bf16T, fp32T, bf16T, fp32T},
1046 SpecificationVersion::V_1_1_DRAFT}},
1047 allOf},
1048 {{Extension::bf16, Extension::mx_common, Extension::mx_fp6e2m3},
1049 {{{bf16T, bs32_fp8ue8m0_fp6e2m3T, bf16T, fp32T, fp16T},
1050 SpecificationVersion::V_1_1_DRAFT},
1051 {{bs32_fp8ue8m0_fp6e2m3T, bf16T, fp32T, bf16T, fp16T},
1052 SpecificationVersion::V_1_1_DRAFT},
1053 {{bf16T, bs32_fp8ue8m0_fp6e2m3T, bf16T, fp32T, fp32T},
1054 SpecificationVersion::V_1_1_DRAFT},
1055 {{bs32_fp8ue8m0_fp6e2m3T, bf16T, fp32T, bf16T, fp32T},
1056 SpecificationVersion::V_1_1_DRAFT}},
1057 allOf},
1058 {{Extension::bf16, Extension::mx_common, Extension::mx_fp4e2m1},
1059 {{{bf16T, bs32_fp8ue8m0_fp4e2m1T, bf16T, fp32T, fp16T},
1060 SpecificationVersion::V_1_1_DRAFT},
1061 {{bs32_fp8ue8m0_fp4e2m1T, bf16T, fp32T, bf16T, fp16T},
1062 SpecificationVersion::V_1_1_DRAFT},
1063 {{bf16T, bs32_fp8ue8m0_fp4e2m1T, bf16T, fp32T, fp32T},
1064 SpecificationVersion::V_1_1_DRAFT},
1065 {{bs32_fp8ue8m0_fp4e2m1T, bf16T, fp32T, bf16T, fp32T},
1066 SpecificationVersion::V_1_1_DRAFT}},
1067 allOf},
1068 {{Extension::bf16, Extension::mx_common, Extension::mx_int8},
1069 {{{bf16T, bs32_fp8ue8m0_mxint8T, bf16T, fp32T, fp16T},
1070 SpecificationVersion::V_1_1_DRAFT},
1071 {{bs32_fp8ue8m0_mxint8T, bf16T, fp32T, bf16T, fp16T},
1072 SpecificationVersion::V_1_1_DRAFT},
1073 {{bf16T, bs32_fp8ue8m0_mxint8T, bf16T, fp32T, fp32T},
1074 SpecificationVersion::V_1_1_DRAFT},
1075 {{bs32_fp8ue8m0_mxint8T, bf16T, fp32T, bf16T, fp32T},
1076 SpecificationVersion::V_1_1_DRAFT}},
1077 allOf},
1078 {{Extension::mx_common, Extension::mx_fp8e4m3, Extension::mx_fp8e5m2},
1079 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp16T},
1080 SpecificationVersion::V_1_1_DRAFT},
1081 {{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp16T},
1082 SpecificationVersion::V_1_1_DRAFT},
1083 {{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp32T},
1084 SpecificationVersion::V_1_1_DRAFT},
1085 {{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp32T},
1086 SpecificationVersion::V_1_1_DRAFT}},
1087 allOf},
1088 {{Extension::mx_common, Extension::mx_fp6e3m2, Extension::mx_fp8e4m3},
1089 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp16T},
1090 SpecificationVersion::V_1_1_DRAFT},
1091 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp16T},
1092 SpecificationVersion::V_1_1_DRAFT},
1093 {{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp32T},
1094 SpecificationVersion::V_1_1_DRAFT},
1095 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp32T},
1096 SpecificationVersion::V_1_1_DRAFT}},
1097 allOf},
1098 {{Extension::mx_common, Extension::mx_fp6e2m3, Extension::mx_fp8e4m3},
1099 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp16T},
1100 SpecificationVersion::V_1_1_DRAFT},
1101 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp16T},
1102 SpecificationVersion::V_1_1_DRAFT},
1103 {{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp32T},
1104 SpecificationVersion::V_1_1_DRAFT},
1105 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp32T},
1106 SpecificationVersion::V_1_1_DRAFT}},
1107 allOf},
1108 {{Extension::mx_common, Extension::mx_fp4e2m1, Extension::mx_fp8e4m3},
1109 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp16T},
1110 SpecificationVersion::V_1_1_DRAFT},
1111 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp16T},
1112 SpecificationVersion::V_1_1_DRAFT},
1113 {{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp32T},
1114 SpecificationVersion::V_1_1_DRAFT},
1115 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp32T},
1116 SpecificationVersion::V_1_1_DRAFT}},
1117 allOf},
1118 {{Extension::mx_common, Extension::mx_fp8e4m3, Extension::mx_int8},
1119 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp16T},
1120 SpecificationVersion::V_1_1_DRAFT},
1121 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp16T},
1122 SpecificationVersion::V_1_1_DRAFT},
1123 {{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp32T},
1124 SpecificationVersion::V_1_1_DRAFT},
1125 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp32T},
1126 SpecificationVersion::V_1_1_DRAFT}},
1127 allOf},
1128 {{Extension::mx_common, Extension::mx_fp6e3m2, Extension::mx_fp8e5m2},
1129 {{{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp16T},
1130 SpecificationVersion::V_1_1_DRAFT},
1131 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp16T},
1132 SpecificationVersion::V_1_1_DRAFT},
1133 {{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp32T},
1134 SpecificationVersion::V_1_1_DRAFT},
1135 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp32T},
1136 SpecificationVersion::V_1_1_DRAFT}},
1137 allOf},
1138 {{Extension::mx_common, Extension::mx_fp6e2m3, Extension::mx_fp8e5m2},
1139 {{{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp16T},
1140 SpecificationVersion::V_1_1_DRAFT},
1141 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp16T},
1142 SpecificationVersion::V_1_1_DRAFT},
1143 {{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp32T},
1144 SpecificationVersion::V_1_1_DRAFT},
1145 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp32T},
1146 SpecificationVersion::V_1_1_DRAFT}},
1147 allOf},
1148 {{Extension::mx_common, Extension::mx_fp4e2m1, Extension::mx_fp8e5m2},
1149 {{{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp16T},
1150 SpecificationVersion::V_1_1_DRAFT},
1151 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp16T},
1152 SpecificationVersion::V_1_1_DRAFT},
1153 {{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp32T},
1154 SpecificationVersion::V_1_1_DRAFT},
1155 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp32T},
1156 SpecificationVersion::V_1_1_DRAFT}},
1157 allOf},
1158 {{Extension::mx_common, Extension::mx_fp8e5m2, Extension::mx_int8},
1159 {{{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp16T},
1160 SpecificationVersion::V_1_1_DRAFT},
1161 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp16T},
1162 SpecificationVersion::V_1_1_DRAFT},
1163 {{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp32T},
1164 SpecificationVersion::V_1_1_DRAFT},
1165 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp32T},
1166 SpecificationVersion::V_1_1_DRAFT}},
1167 allOf},
1168 {{Extension::mx_common, Extension::mx_fp6e2m3, Extension::mx_fp6e3m2},
1169 {{{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp16T},
1170 SpecificationVersion::V_1_1_DRAFT},
1171 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp16T},
1172 SpecificationVersion::V_1_1_DRAFT},
1173 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp32T},
1174 SpecificationVersion::V_1_1_DRAFT},
1175 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp32T},
1176 SpecificationVersion::V_1_1_DRAFT}},
1177 allOf},
1178 {{Extension::mx_common, Extension::mx_fp4e2m1, Extension::mx_fp6e3m2},
1179 {{{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp16T},
1180 SpecificationVersion::V_1_1_DRAFT},
1181 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp16T},
1182 SpecificationVersion::V_1_1_DRAFT},
1183 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp32T},
1184 SpecificationVersion::V_1_1_DRAFT},
1185 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp32T},
1186 SpecificationVersion::V_1_1_DRAFT}},
1187 allOf},
1188 {{Extension::mx_common, Extension::mx_fp6e3m2, Extension::mx_int8},
1189 {{{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp16T},
1190 SpecificationVersion::V_1_1_DRAFT},
1191 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp16T},
1192 SpecificationVersion::V_1_1_DRAFT},
1193 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp32T},
1194 SpecificationVersion::V_1_1_DRAFT},
1195 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp32T},
1196 SpecificationVersion::V_1_1_DRAFT}},
1197 allOf},
1198 {{Extension::mx_common, Extension::mx_fp4e2m1, Extension::mx_fp6e2m3},
1199 {{{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp16T},
1200 SpecificationVersion::V_1_1_DRAFT},
1201 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp16T},
1202 SpecificationVersion::V_1_1_DRAFT},
1203 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp32T},
1204 SpecificationVersion::V_1_1_DRAFT},
1205 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp32T},
1206 SpecificationVersion::V_1_1_DRAFT}},
1207 allOf},
1208 {{Extension::mx_common, Extension::mx_fp6e2m3, Extension::mx_int8},
1209 {{{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp16T},
1210 SpecificationVersion::V_1_1_DRAFT},
1211 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp16T},
1212 SpecificationVersion::V_1_1_DRAFT},
1213 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp32T},
1214 SpecificationVersion::V_1_1_DRAFT},
1215 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp32T},
1216 SpecificationVersion::V_1_1_DRAFT}},
1217 allOf},
1218 {{Extension::mx_common, Extension::mx_fp4e2m1, Extension::mx_int8},
1219 {{{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp16T},
1220 SpecificationVersion::V_1_1_DRAFT},
1221 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp16T},
1222 SpecificationVersion::V_1_1_DRAFT},
1223 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp32T},
1224 SpecificationVersion::V_1_1_DRAFT},
1225 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp32T},
1226 SpecificationVersion::V_1_1_DRAFT}},
1227 allOf}}},
1228 {"tosa.matmul_t_block_scaled",
1229 {{{Extension::mxfp},
1230 {{{fp4e2m1T, fp8ue8m0T, fp4e2m1T, fp8ue8m0T, fp32T},
1231 SpecificationVersion::V_1_1_DRAFT},
1232 {{fp6e2m3T, fp8ue8m0T, fp6e2m3T, fp8ue8m0T, fp32T},
1233 SpecificationVersion::V_1_1_DRAFT},
1234 {{fp6e3m2T, fp8ue8m0T, fp6e3m2T, fp8ue8m0T, fp32T},
1235 SpecificationVersion::V_1_1_DRAFT},
1236 {{fp8e4m3T, fp8ue8m0T, fp8e4m3T, fp8ue8m0T, fp32T},
1237 SpecificationVersion::V_1_1_DRAFT},
1238 {{fp8e5m2T, fp8ue8m0T, fp8e5m2T, fp8ue8m0T, fp32T},
1239 SpecificationVersion::V_1_1_DRAFT},
1240 {{mxint8T, fp8ue8m0T, mxint8T, fp8ue8m0T, fp32T},
1241 SpecificationVersion::V_1_1_DRAFT}}}}},
1242 {"tosa.max_pool2d",
1243 {{{Extension::int16}, {{{i16T, i16T}, SpecificationVersion::V_1_0}}},
1244 {{Extension::fp8e4m3},
1245 {{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
1246 {{Extension::fp8e5m2},
1247 {{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
1248 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1249 {"tosa.max_pool2d_adaptive",
1250 {{{Extension::int16}, {{{i16T, i16T}, SpecificationVersion::V_1_1_DRAFT}}},
1251 {{Extension::fp8e4m3},
1252 {{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT}}},
1253 {{Extension::fp8e5m2},
1254 {{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT}}},
1255 {{Extension::bf16},
1256 {{{bf16T, bf16T}, SpecificationVersion::V_1_1_DRAFT}}}}},
1257 {"tosa.rfft2d",
1258 {{{Extension::fft},
1259 {{{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
1260 {"tosa.transpose_conv2d",
1261 {{{Extension::int4},
1262 {{{i8T, i4T, i32T, i8T, i4T, i32T, i32T}, SpecificationVersion::V_1_0}}},
1263 {{Extension::int16},
1264 {{{i16T, i8T, i48T, i16T, i8T, i48T, i48T},
1265 SpecificationVersion::V_1_0}}},
1266 {{Extension::fp8e4m3},
1267 {{{fp8e4m3T, fp8e4m3T, fp16T, fp8e4m3T, fp8e4m3T, fp16T, fp16T},
1268 SpecificationVersion::V_1_0},
1269 {{fp8e4m3T, fp8e4m3T, fp32T, fp8e4m3T, fp8e4m3T, fp32T, fp32T},
1270 SpecificationVersion::V_1_1_DRAFT}}},
1271 {{Extension::fp8e5m2},
1272 {{{fp8e5m2T, fp8e5m2T, fp16T, fp8e5m2T, fp8e5m2T, fp16T, fp16T},
1273 SpecificationVersion::V_1_0},
1274 {{fp8e5m2T, fp8e5m2T, fp32T, fp8e5m2T, fp8e5m2T, fp32T, fp32T},
1275 SpecificationVersion::V_1_1_DRAFT}}},
1276 {{Extension::bf16},
1277 {{{bf16T, bf16T, bf16T, bf16T, bf16T, fp32T, bf16T},
1278 SpecificationVersion::V_1_0}}}}},
1279 {"tosa.clamp",
1280 {{{Extension::int16}, {{{i16T, i16T}, SpecificationVersion::V_1_0}}},
1281 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1282 {"tosa.erf",
1283 {{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1284 {"tosa.sigmoid",
1285 {{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1286 {"tosa.tanh",
1287 {{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1288 {"tosa.add",
1289 {{{Extension::int64},
1290 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
1291 {{Extension::bf16},
1292 {{{bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1293 {"tosa.arithmetic_right_shift",
1294 {{{Extension::int64},
1295 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
1296 {"tosa.bitwise_and",
1297 {{{Extension::int64},
1298 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
1299 {"tosa.bitwise_or",
1300 {{{Extension::int64},
1301 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
1302 {"tosa.bitwise_xor",
1303 {{{Extension::int64},
1304 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
1305 {"tosa.intdiv",
1306 {{{Extension::int64},
1307 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
1308 {"tosa.logical_left_shift",
1309 {{{Extension::int64},
1310 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
1311 {"tosa.logical_right_shift",
1312 {{{Extension::int64},
1313 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
1314 {"tosa.maximum",
1315 {{{Extension::int64},
1316 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
1317 {{Extension::bf16},
1318 {{{bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1319 {"tosa.minimum",
1320 {{{Extension::int64},
1321 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
1322 {{Extension::bf16},
1323 {{{bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1324 {"tosa.mul",
1325 {{{Extension::int64},
1326 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
1327 {{Extension::bf16},
1328 {{{bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1329 {"tosa.pow",
1330 {{{Extension::bf16},
1331 {{{bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1332 {"tosa.sub",
1333 {{{Extension::int64},
1334 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
1335 {{Extension::bf16},
1336 {{{bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1337 {"tosa.table",
1338 {{{Extension::int16},
1339 {{{i16T, i16T, i32T}, SpecificationVersion::V_1_0}}}}},
1340 {"tosa.abs",
1341 {{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
1342 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1343 {"tosa.bitwise_not",
1344 {{{Extension::int64},
1345 {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
1346 {"tosa.ceil",
1347 {{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1348 {"tosa.clz",
1349 {{{Extension::int64},
1350 {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
1351 {"tosa.cos",
1352 {{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1353 {"tosa.exp",
1354 {{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1355 {"tosa.floor",
1356 {{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1357 {"tosa.log",
1358 {{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1359 {"tosa.negate",
1360 {{{Extension::int64},
1361 {{{i64T, i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
1362 {{Extension::bf16},
1363 {{{bf16T, bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1364 {"tosa.reciprocal",
1365 {{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1366 {"tosa.rsqrt",
1367 {{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1368 {"tosa.sin",
1369 {{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1370 {"tosa.select",
1371 {{{Extension::int64},
1372 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
1373 {{Extension::bf16},
1374 {{{bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1375 {"tosa.equal",
1376 {{{Extension::int64},
1377 {{{i64T, i64T, boolT}, SpecificationVersion::V_1_1_DRAFT}}},
1378 {{Extension::bf16},
1379 {{{bf16T, bf16T, boolT}, SpecificationVersion::V_1_0}}}}},
1380 {"tosa.greater",
1381 {{{Extension::int64},
1382 {{{i64T, i64T, boolT}, SpecificationVersion::V_1_1_DRAFT}}},
1383 {{Extension::bf16},
1384 {{{bf16T, bf16T, boolT}, SpecificationVersion::V_1_0}}}}},
1385 {"tosa.greater_equal",
1386 {{{Extension::int64},
1387 {{{i64T, i64T, boolT}, SpecificationVersion::V_1_1_DRAFT}}},
1388 {{Extension::bf16},
1389 {{{bf16T, bf16T, boolT}, SpecificationVersion::V_1_0}}}}},
1390 {"tosa.reduce_max",
1391 {{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
1392 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1393 {"tosa.reduce_min",
1394 {{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
1395 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1396 {"tosa.reduce_product",
1397 {{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1398 {"tosa.reduce_sum",
1399 {{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
1400 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1401 {"tosa.concat",
1402 {{{Extension::int16}, {{{i16T, i16T}, SpecificationVersion::V_1_0}}},
1403 {{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
1404 {{Extension::fp8e4m3},
1405 {{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
1406 {{Extension::fp8e5m2},
1407 {{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
1408 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}},
1409 {{Extension::mx_common, Extension::mx_fp8e4m3},
1410 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp8e4m3T},
1411 SpecificationVersion::V_1_1_DRAFT}},
1412 allOf},
1413 {{Extension::mx_common, Extension::mx_fp8e5m2},
1414 {{{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp8e5m2T},
1415 SpecificationVersion::V_1_1_DRAFT}},
1416 allOf},
1417 {{Extension::mx_common, Extension::mx_fp6e3m2},
1418 {{{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp6e3m2T},
1419 SpecificationVersion::V_1_1_DRAFT}},
1420 allOf},
1421 {{Extension::mx_common, Extension::mx_fp6e2m3},
1422 {{{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp6e2m3T},
1423 SpecificationVersion::V_1_1_DRAFT}},
1424 allOf},
1425 {{Extension::mx_common, Extension::mx_fp4e2m1},
1426 {{{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp4e2m1T},
1427 SpecificationVersion::V_1_1_DRAFT}},
1428 allOf},
1429 {{Extension::mx_common, Extension::mx_int8},
1430 {{{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_mxint8T},
1431 SpecificationVersion::V_1_1_DRAFT}},
1432 allOf}}},
1433 {"tosa.pad",
1434 {{{Extension::int64},
1435 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
1436 {{Extension::fp8e4m3},
1437 {{{fp8e4m3T, fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
1438 {{Extension::fp8e5m2},
1439 {{{fp8e5m2T, fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
1440 {{Extension::bf16},
1441 {{{bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1442 {"tosa.reshape",
1443 {{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
1444 {{Extension::fp8e4m3},
1445 {{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
1446 {{Extension::fp8e5m2},
1447 {{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
1448 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}},
1449 {{Extension::mx_common, Extension::mx_fp8e4m3},
1450 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp8e4m3T},
1451 SpecificationVersion::V_1_1_DRAFT}},
1452 allOf},
1453 {{Extension::mx_common, Extension::mx_fp8e5m2},
1454 {{{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp8e5m2T},
1455 SpecificationVersion::V_1_1_DRAFT}},
1456 allOf},
1457 {{Extension::mx_common, Extension::mx_fp6e3m2},
1458 {{{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp6e3m2T},
1459 SpecificationVersion::V_1_1_DRAFT}},
1460 allOf},
1461 {{Extension::mx_common, Extension::mx_fp6e2m3},
1462 {{{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp6e2m3T},
1463 SpecificationVersion::V_1_1_DRAFT}},
1464 allOf},
1465 {{Extension::mx_common, Extension::mx_fp4e2m1},
1466 {{{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp4e2m1T},
1467 SpecificationVersion::V_1_1_DRAFT}},
1468 allOf},
1469 {{Extension::mx_common, Extension::mx_int8},
1470 {{{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_mxint8T},
1471 SpecificationVersion::V_1_1_DRAFT}},
1472 allOf}}},
1473 {"tosa.reverse",
1474 {{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
1475 {{Extension::fp8e4m3},
1476 {{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
1477 {{Extension::fp8e5m2},
1478 {{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
1479 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1480 {"tosa.slice",
1481 {{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
1482 {{Extension::fp8e4m3},
1483 {{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
1484 {{Extension::fp8e5m2},
1485 {{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
1486 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1487 {"tosa.tile",
1488 {{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
1489 {{Extension::fp8e4m3},
1490 {{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
1491 {{Extension::fp8e5m2},
1492 {{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
1493 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1494 {"tosa.transpose",
1495 {{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
1496 {{Extension::fp8e4m3},
1497 {{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
1498 {{Extension::fp8e5m2},
1499 {{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
1500 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1501 {"tosa.gather",
1502 {{{Extension::fp8e4m3},
1503 {{{fp8e4m3T, i32T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
1504 {{Extension::fp8e5m2},
1505 {{{fp8e5m2T, i32T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
1506 {{Extension::bf16},
1507 {{{bf16T, i32T, bf16T}, SpecificationVersion::V_1_0}}},
1508 {{Extension::int64},
1509 {{{i8T, i64T, i8T}, SpecificationVersion::V_1_1_DRAFT},
1510 {{i16T, i64T, i16T}, SpecificationVersion::V_1_1_DRAFT},
1511 {{i32T, i64T, i32T}, SpecificationVersion::V_1_1_DRAFT},
1512 {{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT},
1513 {{fp16T, i64T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
1514 {{fp32T, i64T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
1515 {{boolT, i64T, boolT}, SpecificationVersion::V_1_1_DRAFT}}},
1516 {{Extension::fp8e4m3, Extension::int64},
1517 {{{fp8e4m3T, i64T, fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT}},
1518 allOf},
1519 {{Extension::fp8e5m2, Extension::int64},
1520 {{{fp8e5m2T, i64T, fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT}},
1521 allOf},
1522 {{Extension::bf16, Extension::int64},
1523 {{{bf16T, i64T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
1524 allOf}}},
1525 {"tosa.row_gather",
1526 {{{Extension::int16},
1527 {{{i16T, i32T, i16T}, SpecificationVersion::V_1_1_DRAFT}}},
1528 {{Extension::int64},
1529 {{{i64T, i32T, i64T}, SpecificationVersion::V_1_1_DRAFT},
1530 {{boolT, i64T, boolT}, SpecificationVersion::V_1_1_DRAFT},
1531 {{i8T, i64T, i8T}, SpecificationVersion::V_1_1_DRAFT},
1532 {{i32T, i64T, i32T}, SpecificationVersion::V_1_1_DRAFT},
1533 {{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT},
1534 {{fp16T, i64T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
1535 {{fp32T, i64T, fp32T}, SpecificationVersion::V_1_1_DRAFT}}},
1536 {{Extension::int16, Extension::int64},
1537 {{{i16T, i64T, i16T}, SpecificationVersion::V_1_1_DRAFT}},
1538 allOf},
1539 {{Extension::fp8e4m3},
1540 {{{fp8e4m3T, i32T, fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT}}},
1541 {{Extension::fp8e5m2},
1542 {{{fp8e5m2T, i32T, fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT}}},
1543 {{Extension::bf16},
1544 {{{bf16T, i32T, bf16T}, SpecificationVersion::V_1_1_DRAFT}}},
1545 {{Extension::mx_common, Extension::mx_fp8e4m3},
1546 {{{bs32_fp8ue8m0_fp8e4m3T, i32T, bs32_fp8ue8m0_fp8e4m3T},
1547 SpecificationVersion::V_1_1_DRAFT}},
1548 allOf},
1549 {{Extension::mx_common, Extension::mx_fp8e5m2},
1550 {{{bs32_fp8ue8m0_fp8e5m2T, i32T, bs32_fp8ue8m0_fp8e5m2T},
1551 SpecificationVersion::V_1_1_DRAFT}},
1552 allOf},
1553 {{Extension::mx_common, Extension::mx_fp6e3m2},
1554 {{{bs32_fp8ue8m0_fp6e3m2T, i32T, bs32_fp8ue8m0_fp6e3m2T},
1555 SpecificationVersion::V_1_1_DRAFT}},
1556 allOf},
1557 {{Extension::mx_common, Extension::mx_fp6e2m3},
1558 {{{bs32_fp8ue8m0_fp6e2m3T, i32T, bs32_fp8ue8m0_fp6e2m3T},
1559 SpecificationVersion::V_1_1_DRAFT}},
1560 allOf},
1561 {{Extension::mx_common, Extension::mx_fp4e2m1},
1562 {{{bs32_fp8ue8m0_fp4e2m1T, i32T, bs32_fp8ue8m0_fp4e2m1T},
1563 SpecificationVersion::V_1_1_DRAFT}},
1564 allOf},
1565 {{Extension::mx_common, Extension::mx_int8},
1566 {{{bs32_fp8ue8m0_mxint8T, i32T, bs32_fp8ue8m0_mxint8T},
1567 SpecificationVersion::V_1_1_DRAFT}},
1568 allOf},
1569 {{Extension::fp8e4m3, Extension::int64},
1570 {{{fp8e4m3T, i64T, fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT}},
1571 allOf},
1572 {{Extension::fp8e5m2, Extension::int64},
1573 {{{fp8e5m2T, i64T, fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT}},
1574 allOf},
1575 {{Extension::bf16, Extension::int64},
1576 {{{bf16T, i64T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
1577 allOf},
1578 {{Extension::int64, Extension::mx_common, Extension::mx_fp8e4m3},
1579 {{{bs32_fp8ue8m0_fp8e4m3T, i64T, bs32_fp8ue8m0_fp8e4m3T},
1580 SpecificationVersion::V_1_1_DRAFT}},
1581 allOf},
1582 {{Extension::int64, Extension::mx_common, Extension::mx_fp8e5m2},
1583 {{{bs32_fp8ue8m0_fp8e5m2T, i64T, bs32_fp8ue8m0_fp8e5m2T},
1584 SpecificationVersion::V_1_1_DRAFT}},
1585 allOf},
1586 {{Extension::int64, Extension::mx_common, Extension::mx_fp6e3m2},
1587 {{{bs32_fp8ue8m0_fp6e3m2T, i64T, bs32_fp8ue8m0_fp6e3m2T},
1588 SpecificationVersion::V_1_1_DRAFT}},
1589 allOf},
1590 {{Extension::int64, Extension::mx_common, Extension::mx_fp6e2m3},
1591 {{{bs32_fp8ue8m0_fp6e2m3T, i64T, bs32_fp8ue8m0_fp6e2m3T},
1592 SpecificationVersion::V_1_1_DRAFT}},
1593 allOf},
1594 {{Extension::int64, Extension::mx_common, Extension::mx_fp4e2m1},
1595 {{{bs32_fp8ue8m0_fp4e2m1T, i64T, bs32_fp8ue8m0_fp4e2m1T},
1596 SpecificationVersion::V_1_1_DRAFT}},
1597 allOf},
1598 {{Extension::int64, Extension::mx_common, Extension::mx_int8},
1599 {{{bs32_fp8ue8m0_mxint8T, i64T, bs32_fp8ue8m0_mxint8T},
1600 SpecificationVersion::V_1_1_DRAFT}},
1601 allOf}}},
1602 {"tosa.row_gather_block_scaled",
1603 {{{Extension::fp8e4m3},
1604 {{{fp8e4m3T, i32T, i32T, fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT}}},
1605 {{Extension::fp8e5m2},
1606 {{{fp8e5m2T, i32T, i32T, fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT}}},
1607 {{Extension::bf16},
1608 {{{bf16T, i32T, i32T, bf16T}, SpecificationVersion::V_1_1_DRAFT}}},
1609 {{Extension::int64},
1610 {{{i8T, i64T, i32T, i8T}, SpecificationVersion::V_1_1_DRAFT},
1611 {{i16T, i64T, i32T, i16T}, SpecificationVersion::V_1_1_DRAFT},
1612 {{i32T, i64T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT},
1613 {{i64T, i64T, i32T, i64T}, SpecificationVersion::V_1_1_DRAFT},
1614 {{fp16T, i64T, i32T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
1615 {{fp32T, i64T, i32T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
1616 {{boolT, i64T, i32T, boolT}, SpecificationVersion::V_1_1_DRAFT}}},
1617 {{Extension::fp8e4m3, Extension::int64},
1618 {{{fp8e4m3T, i64T, i32T, fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT}},
1619 allOf},
1620 {{Extension::fp8e5m2, Extension::int64},
1621 {{{fp8e5m2T, i64T, i32T, fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT}},
1622 allOf},
1623 {{Extension::bf16, Extension::int64},
1624 {{{bf16T, i64T, i32T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
1625 allOf},
1626 {{Extension::mxfp},
1627 {{{fp4e2m1T, fp8ue8m0T, i32T, i32T, fp4e2m1T, fp8ue8m0T},
1628 SpecificationVersion::V_1_1_DRAFT},
1629 {{fp6e2m3T, fp8ue8m0T, i32T, i32T, fp6e2m3T, fp8ue8m0T},
1630 SpecificationVersion::V_1_1_DRAFT},
1631 {{fp6e3m2T, fp8ue8m0T, i32T, i32T, fp6e3m2T, fp8ue8m0T},
1632 SpecificationVersion::V_1_1_DRAFT},
1633 {{fp8e4m3T, fp8ue8m0T, i32T, i32T, fp8e4m3T, fp8ue8m0T},
1634 SpecificationVersion::V_1_1_DRAFT},
1635 {{fp8e5m2T, fp8ue8m0T, i32T, i32T, fp8e5m2T, fp8ue8m0T},
1636 SpecificationVersion::V_1_1_DRAFT},
1637 {{mxint8T, fp8ue8m0T, i32T, i32T, mxint8T, fp8ue8m0T},
1638 SpecificationVersion::V_1_1_DRAFT}}},
1639 {{Extension::mxfp, Extension::int64},
1640 {{{fp4e2m1T, fp8ue8m0T, i64T, i32T, fp4e2m1T, fp8ue8m0T},
1641 SpecificationVersion::V_1_1_DRAFT},
1642 {{fp6e2m3T, fp8ue8m0T, i64T, i32T, fp6e2m3T, fp8ue8m0T},
1643 SpecificationVersion::V_1_1_DRAFT},
1644 {{fp6e3m2T, fp8ue8m0T, i64T, i32T, fp6e3m2T, fp8ue8m0T},
1645 SpecificationVersion::V_1_1_DRAFT},
1646 {{fp8e4m3T, fp8ue8m0T, i64T, i32T, fp8e4m3T, fp8ue8m0T},
1647 SpecificationVersion::V_1_1_DRAFT},
1648 {{fp8e5m2T, fp8ue8m0T, i64T, i32T, fp8e5m2T, fp8ue8m0T},
1649 SpecificationVersion::V_1_1_DRAFT},
1650 {{mxint8T, fp8ue8m0T, i64T, i32T, mxint8T, fp8ue8m0T},
1651 SpecificationVersion::V_1_1_DRAFT}},
1652 allOf}}},
1653 {"tosa.scatter",
1654 {{{Extension::fp8e4m3},
1655 {{{fp8e4m3T, i32T, fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
1656 {{Extension::fp8e5m2},
1657 {{{fp8e5m2T, i32T, fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
1658 {{Extension::bf16},
1659 {{{bf16T, i32T, bf16T, bf16T}, SpecificationVersion::V_1_0}}},
1660 {{Extension::int64},
1661 {{{i8T, i64T, i8T, i8T}, SpecificationVersion::V_1_1_DRAFT},
1662 {{i16T, i64T, i16T, i16T}, SpecificationVersion::V_1_1_DRAFT},
1663 {{i32T, i64T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT},
1664 {{i64T, i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT},
1665 {{fp16T, i64T, fp16T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
1666 {{fp32T, i64T, fp32T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
1667 {{boolT, i64T, boolT, boolT}, SpecificationVersion::V_1_1_DRAFT}}},
1668 {{Extension::fp8e4m3, Extension::int64},
1669 {{{fp8e4m3T, i64T, fp8e4m3T, fp8e4m3T},
1670 SpecificationVersion::V_1_1_DRAFT}},
1671 allOf},
1672 {{Extension::fp8e5m2, Extension::int64},
1673 {{{fp8e5m2T, i64T, fp8e5m2T, fp8e5m2T},
1674 SpecificationVersion::V_1_1_DRAFT}},
1675 allOf},
1676 {{Extension::bf16, Extension::int64},
1677 {{{bf16T, i64T, bf16T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
1678 allOf}}},
1679 {"tosa.resize",
1680 {{{Extension::int16},
1681 {{{i16T, i48T}, SpecificationVersion::V_1_0},
1682 {{i16T, i16T}, SpecificationVersion::V_1_0}}},
1683 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1684 {"tosa.cast",
1685 {{{Extension::bf16},
1686 {{{i8T, bf16T}, SpecificationVersion::V_1_0},
1687 {{i16T, bf16T}, SpecificationVersion::V_1_0},
1688 {{i32T, bf16T}, SpecificationVersion::V_1_0},
1689 {{bf16T, i8T}, SpecificationVersion::V_1_0},
1690 {{bf16T, i16T}, SpecificationVersion::V_1_0},
1691 {{bf16T, i32T}, SpecificationVersion::V_1_0},
1692 {{bf16T, fp32T}, SpecificationVersion::V_1_0},
1693 {{fp32T, bf16T}, SpecificationVersion::V_1_0}}},
1694 {{Extension::int64},
1695 {{{i32T, i64T}, SpecificationVersion::V_1_1_DRAFT},
1696 {{i64T, i32T}, SpecificationVersion::V_1_1_DRAFT},
1697 {{boolT, i64T}, SpecificationVersion::V_1_1_DRAFT},
1698 {{i64T, boolT}, SpecificationVersion::V_1_1_DRAFT}}},
1699 {{Extension::bf16, Extension::fp8e4m3},
1700 {{{bf16T, fp8e4m3T}, SpecificationVersion::V_1_0},
1701 {{fp8e4m3T, bf16T}, SpecificationVersion::V_1_0}},
1702 allOf},
1703 {{Extension::bf16, Extension::fp8e5m2},
1704 {{{bf16T, fp8e5m2T}, SpecificationVersion::V_1_0},
1705 {{fp8e5m2T, bf16T}, SpecificationVersion::V_1_0}},
1706 allOf},
1707 {{Extension::fp8e4m3},
1708 {{{fp8e4m3T, fp16T}, SpecificationVersion::V_1_0},
1709 {{fp8e4m3T, fp32T}, SpecificationVersion::V_1_0},
1710 {{fp16T, fp8e4m3T}, SpecificationVersion::V_1_0},
1711 {{fp32T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
1712 {{Extension::fp8e5m2},
1713 {{{fp8e5m2T, fp16T}, SpecificationVersion::V_1_0},
1714 {{fp8e5m2T, fp32T}, SpecificationVersion::V_1_0},
1715 {{fp16T, fp8e5m2T}, SpecificationVersion::V_1_0},
1716 {{fp32T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
1717 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp8e4m3},
1718 {{{fp8e4m3T, bs32_fp8ue8m0_fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT},
1719 {{bs32_fp8ue8m0_fp8e4m3T, fp8e4m3T},
1720 SpecificationVersion::V_1_1_DRAFT}},
1721 allOf},
1722 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp8e5m2},
1723 {{{fp8e4m3T, bs32_fp8ue8m0_fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT},
1724 {{bs32_fp8ue8m0_fp8e5m2T, fp8e4m3T},
1725 SpecificationVersion::V_1_1_DRAFT}},
1726 allOf},
1727 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp6e3m2},
1728 {{{fp8e4m3T, bs32_fp8ue8m0_fp6e3m2T}, SpecificationVersion::V_1_1_DRAFT},
1729 {{bs32_fp8ue8m0_fp6e3m2T, fp8e4m3T},
1730 SpecificationVersion::V_1_1_DRAFT}},
1731 allOf},
1732 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp6e2m3},
1733 {{{fp8e4m3T, bs32_fp8ue8m0_fp6e2m3T}, SpecificationVersion::V_1_1_DRAFT},
1734 {{bs32_fp8ue8m0_fp6e2m3T, fp8e4m3T},
1735 SpecificationVersion::V_1_1_DRAFT}},
1736 allOf},
1737 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp4e2m1},
1738 {{{fp8e4m3T, bs32_fp8ue8m0_fp4e2m1T}, SpecificationVersion::V_1_1_DRAFT},
1739 {{bs32_fp8ue8m0_fp4e2m1T, fp8e4m3T},
1740 SpecificationVersion::V_1_1_DRAFT}},
1741 allOf},
1742 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_int8},
1743 {{{fp8e4m3T, bs32_fp8ue8m0_mxint8T}, SpecificationVersion::V_1_1_DRAFT},
1744 {{bs32_fp8ue8m0_mxint8T, fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT}},
1745 allOf},
1746 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp8e4m3},
1747 {{{fp8e5m2T, bs32_fp8ue8m0_fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT},
1748 {{bs32_fp8ue8m0_fp8e4m3T, fp8e5m2T},
1749 SpecificationVersion::V_1_1_DRAFT}},
1750 allOf},
1751 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp8e5m2},
1752 {{{fp8e5m2T, bs32_fp8ue8m0_fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT},
1753 {{bs32_fp8ue8m0_fp8e5m2T, fp8e5m2T},
1754 SpecificationVersion::V_1_1_DRAFT}},
1755 allOf},
1756 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp6e3m2},
1757 {{{fp8e5m2T, bs32_fp8ue8m0_fp6e3m2T}, SpecificationVersion::V_1_1_DRAFT},
1758 {{bs32_fp8ue8m0_fp6e3m2T, fp8e5m2T},
1759 SpecificationVersion::V_1_1_DRAFT}},
1760 allOf},
1761 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp6e2m3},
1762 {{{fp8e5m2T, bs32_fp8ue8m0_fp6e2m3T}, SpecificationVersion::V_1_1_DRAFT},
1763 {{bs32_fp8ue8m0_fp6e2m3T, fp8e5m2T},
1764 SpecificationVersion::V_1_1_DRAFT}},
1765 allOf},
1766 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp4e2m1},
1767 {{{fp8e5m2T, bs32_fp8ue8m0_fp4e2m1T}, SpecificationVersion::V_1_1_DRAFT},
1768 {{bs32_fp8ue8m0_fp4e2m1T, fp8e5m2T},
1769 SpecificationVersion::V_1_1_DRAFT}},
1770 allOf},
1771 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_int8},
1772 {{{fp8e5m2T, bs32_fp8ue8m0_mxint8T}, SpecificationVersion::V_1_1_DRAFT},
1773 {{bs32_fp8ue8m0_mxint8T, fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT}},
1774 allOf},
1775 {{Extension::mx_common, Extension::mx_fp8e4m3},
1776 {{{fp16T, bs32_fp8ue8m0_fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT},
1777 {{fp32T, bs32_fp8ue8m0_fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT},
1778 {{bs32_fp8ue8m0_fp8e4m3T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
1779 {{bs32_fp8ue8m0_fp8e4m3T, fp32T}, SpecificationVersion::V_1_1_DRAFT}},
1780 allOf},
1781 {{Extension::mx_common, Extension::mx_fp8e5m2},
1782 {{{fp16T, bs32_fp8ue8m0_fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT},
1783 {{fp32T, bs32_fp8ue8m0_fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT},
1784 {{bs32_fp8ue8m0_fp8e5m2T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
1785 {{bs32_fp8ue8m0_fp8e5m2T, fp32T}, SpecificationVersion::V_1_1_DRAFT}},
1786 allOf},
1787 {{Extension::mx_common, Extension::mx_fp6e3m2},
1788 {{{fp16T, bs32_fp8ue8m0_fp6e3m2T}, SpecificationVersion::V_1_1_DRAFT},
1789 {{fp32T, bs32_fp8ue8m0_fp6e3m2T}, SpecificationVersion::V_1_1_DRAFT},
1790 {{bs32_fp8ue8m0_fp6e3m2T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
1791 {{bs32_fp8ue8m0_fp6e3m2T, fp32T}, SpecificationVersion::V_1_1_DRAFT}},
1792 allOf},
1793 {{Extension::mx_common, Extension::mx_fp6e2m3},
1794 {{{fp16T, bs32_fp8ue8m0_fp6e2m3T}, SpecificationVersion::V_1_1_DRAFT},
1795 {{fp32T, bs32_fp8ue8m0_fp6e2m3T}, SpecificationVersion::V_1_1_DRAFT},
1796 {{bs32_fp8ue8m0_fp6e2m3T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
1797 {{bs32_fp8ue8m0_fp6e2m3T, fp32T}, SpecificationVersion::V_1_1_DRAFT}},
1798 allOf},
1799 {{Extension::mx_common, Extension::mx_fp4e2m1},
1800 {{{fp16T, bs32_fp8ue8m0_fp4e2m1T}, SpecificationVersion::V_1_1_DRAFT},
1801 {{fp32T, bs32_fp8ue8m0_fp4e2m1T}, SpecificationVersion::V_1_1_DRAFT},
1802 {{bs32_fp8ue8m0_fp4e2m1T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
1803 {{bs32_fp8ue8m0_fp4e2m1T, fp32T}, SpecificationVersion::V_1_1_DRAFT}},
1804 allOf},
1805 {{Extension::mx_common, Extension::mx_int8},
1806 {{{fp16T, bs32_fp8ue8m0_mxint8T}, SpecificationVersion::V_1_1_DRAFT},
1807 {{fp32T, bs32_fp8ue8m0_mxint8T}, SpecificationVersion::V_1_1_DRAFT},
1808 {{bs32_fp8ue8m0_mxint8T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
1809 {{bs32_fp8ue8m0_mxint8T, fp32T}, SpecificationVersion::V_1_1_DRAFT}},
1810 allOf},
1811 {{Extension::bf16, Extension::mx_common, Extension::mx_fp8e4m3},
1812 {{{bf16T, bs32_fp8ue8m0_fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT},
1813 {{bs32_fp8ue8m0_fp8e4m3T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
1814 allOf},
1815 {{Extension::bf16, Extension::mx_common, Extension::mx_fp8e5m2},
1816 {{{bf16T, bs32_fp8ue8m0_fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT},
1817 {{bs32_fp8ue8m0_fp8e5m2T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
1818 allOf},
1819 {{Extension::bf16, Extension::mx_common, Extension::mx_fp6e3m2},
1820 {{{bf16T, bs32_fp8ue8m0_fp6e3m2T}, SpecificationVersion::V_1_1_DRAFT},
1821 {{bs32_fp8ue8m0_fp6e3m2T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
1822 allOf},
1823 {{Extension::bf16, Extension::mx_common, Extension::mx_fp6e2m3},
1824 {{{bf16T, bs32_fp8ue8m0_fp6e2m3T}, SpecificationVersion::V_1_1_DRAFT},
1825 {{bs32_fp8ue8m0_fp6e2m3T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
1826 allOf},
1827 {{Extension::bf16, Extension::mx_common, Extension::mx_fp4e2m1},
1828 {{{bf16T, bs32_fp8ue8m0_fp4e2m1T}, SpecificationVersion::V_1_1_DRAFT},
1829 {{bs32_fp8ue8m0_fp4e2m1T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
1830 allOf},
1831 {{Extension::bf16, Extension::mx_common, Extension::mx_int8},
1832 {{{bf16T, bs32_fp8ue8m0_mxint8T}, SpecificationVersion::V_1_1_DRAFT},
1833 {{bs32_fp8ue8m0_mxint8T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
1834 allOf}}},
1835 {"tosa.cast_from_block_scaled",
1836 {{{Extension::bf16, Extension::mxfp},
1837 {{{fp4e2m1T, fp8ue8m0T, bf16T}, SpecificationVersion::V_1_1_DRAFT},
1838 {{fp6e2m3T, fp8ue8m0T, bf16T}, SpecificationVersion::V_1_1_DRAFT},
1839 {{fp6e3m2T, fp8ue8m0T, bf16T}, SpecificationVersion::V_1_1_DRAFT},
1840 {{fp8e4m3T, fp8ue8m0T, bf16T}, SpecificationVersion::V_1_1_DRAFT},
1841 {{fp8e5m2T, fp8ue8m0T, bf16T}, SpecificationVersion::V_1_1_DRAFT},
1842 {{mxint8T, fp8ue8m0T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
1843 allOf},
1844 {{Extension::mxfp},
1845 {{{fp4e2m1T, fp8ue8m0T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
1846 {{fp6e2m3T, fp8ue8m0T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
1847 {{fp6e3m2T, fp8ue8m0T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
1848 {{fp8e4m3T, fp8ue8m0T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
1849 {{fp8e5m2T, fp8ue8m0T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
1850 {{mxint8T, fp8ue8m0T, fp32T}, SpecificationVersion::V_1_1_DRAFT}}}}},
1851 {"tosa.cast_to_block_scaled",
1852 {{{Extension::mxfp},
1853 {{{bf16T, fp4e2m1T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
1854 {{fp32T, fp4e2m1T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
1855 {{fp32T, fp6e2m3T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
1856 {{fp32T, fp6e3m2T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
1857 {{fp32T, fp8e4m3T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
1858 {{fp32T, fp8e5m2T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
1859 {{fp32T, mxint8T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT}}},
1860 {{Extension::bf16, Extension::mxfp},
1861 {{{bf16T, fp6e2m3T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
1862 {{bf16T, fp6e3m2T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
1863 {{bf16T, fp8e4m3T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
1864 {{bf16T, fp8e5m2T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
1865 {{bf16T, mxint8T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT}},
1866 allOf}}},
1867 {"tosa.rescale",
1868 {{{Extension::int16},
1869 {{{i48T, i48T, i8T, i8T}, SpecificationVersion::V_1_0},
1870 {{i48T, i48T, i16T, i16T}, SpecificationVersion::V_1_0},
1871 {{i48T, i48T, i32T, i32T}, SpecificationVersion::V_1_0}}}}},
1872 {"tosa.const",
1873 {{{Extension::int4}, {{{i4T}, SpecificationVersion::V_1_0}}},
1874 {{Extension::int16}, {{{i48T}, SpecificationVersion::V_1_0}}},
1875 {{Extension::int64}, {{{i64T}, SpecificationVersion::V_1_1_DRAFT}}},
1876 {{Extension::fp8e4m3}, {{{fp8e4m3T}, SpecificationVersion::V_1_0}}},
1877 {{Extension::fp8e5m2}, {{{fp8e5m2T}, SpecificationVersion::V_1_0}}},
1878 {{Extension::bf16}, {{{bf16T}, SpecificationVersion::V_1_0}}},
1879 {{Extension::mxfp},
1880 {{{fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
1881 {{fp6e3m2T}, SpecificationVersion::V_1_1_DRAFT},
1882 {{fp6e2m3T}, SpecificationVersion::V_1_1_DRAFT},
1883 {{fp4e2m1T}, SpecificationVersion::V_1_1_DRAFT},
1884 {{mxint8T}, SpecificationVersion::V_1_1_DRAFT}}},
1885 {{Extension::mx_common, Extension::mx_fp8e4m3},
1886 {{{bs32_fp8ue8m0_fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT}},
1887 allOf},
1888 {{Extension::mx_common, Extension::mx_fp8e5m2},
1889 {{{bs32_fp8ue8m0_fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT}},
1890 allOf},
1891 {{Extension::mx_common, Extension::mx_fp6e3m2},
1892 {{{bs32_fp8ue8m0_fp6e3m2T}, SpecificationVersion::V_1_1_DRAFT}},
1893 allOf},
1894 {{Extension::mx_common, Extension::mx_fp6e2m3},
1895 {{{bs32_fp8ue8m0_fp6e2m3T}, SpecificationVersion::V_1_1_DRAFT}},
1896 allOf},
1897 {{Extension::mx_common, Extension::mx_fp4e2m1},
1898 {{{bs32_fp8ue8m0_fp4e2m1T}, SpecificationVersion::V_1_1_DRAFT}},
1899 allOf},
1900 {{Extension::mx_common, Extension::mx_int8},
1901 {{{bs32_fp8ue8m0_mxint8T}, SpecificationVersion::V_1_1_DRAFT}},
1902 allOf}}},
1903 {"tosa.identity",
1904 {{{Extension::int4}, {{{i4T, i4T}, SpecificationVersion::V_1_0}}},
1905 {{Extension::int16}, {{{i48T, i48T}, SpecificationVersion::V_1_0}}},
1906 {{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
1907 {{Extension::fp8e4m3},
1908 {{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
1909 {{Extension::fp8e5m2},
1910 {{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
1911 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
1912 {"tosa.variable",
1913 {{{Extension::variable},
1914 {{{i8T}, SpecificationVersion::V_1_0},
1915 {{fp16T}, SpecificationVersion::V_1_0},
1916 {{fp32T}, SpecificationVersion::V_1_0}}}}},
1917 {"tosa.variable_write",
1918 {{{Extension::variable},
1919 {{{i8T}, SpecificationVersion::V_1_0},
1920 {{fp16T}, SpecificationVersion::V_1_0},
1921 {{fp32T}, SpecificationVersion::V_1_0}}}}},
1922 {"tosa.variable_read",
1923 {{{Extension::variable},
1924 {{{i8T}, SpecificationVersion::V_1_0},
1925 {{fp16T}, SpecificationVersion::V_1_0},
1926 {{fp32T}, SpecificationVersion::V_1_0}}}}},
1927 {"tosa.dim",
1928 {{{Extension::shape},
1929 {{{boolT}, SpecificationVersion::V_1_1_DRAFT},
1930 {{i8T}, SpecificationVersion::V_1_1_DRAFT},
1931 {{i16T}, SpecificationVersion::V_1_1_DRAFT},
1932 {{i32T}, SpecificationVersion::V_1_1_DRAFT},
1933 {{fp16T}, SpecificationVersion::V_1_1_DRAFT},
1934 {{fp32T}, SpecificationVersion::V_1_1_DRAFT}}},
1935 {{Extension::fp8e4m3, Extension::shape},
1936 {{{fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT}},
1937 allOf},
1938 {{Extension::fp8e5m2, Extension::shape},
1939 {{{fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT}},
1940 allOf},
1941 {{Extension::bf16, Extension::shape},
1942 {{{bf16T}, SpecificationVersion::V_1_1_DRAFT}},
1943 allOf},
1944 {{Extension::mxfp, Extension::shape},
1945 {{{fp6e3m2T}, SpecificationVersion::V_1_1_DRAFT},
1946 {{fp6e2m3T}, SpecificationVersion::V_1_1_DRAFT},
1947 {{fp4e2m1T}, SpecificationVersion::V_1_1_DRAFT},
1948 {{mxint8T}, SpecificationVersion::V_1_1_DRAFT},
1949 {{fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT}},
1950 allOf},
1951 {{Extension::int64, Extension::shape},
1952 {{{i64T}, SpecificationVersion::V_1_1_DRAFT}},
1953 allOf}}}};
1954
1955// End of auto-generated metadata
Generated on for MLIR by doxygen 1.14.0