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 {{fp32T, fp32T, fp32T, fp32T, fp32T},
76 SpecificationVersion::V_1_1_DRAFT}}}}},
77 {"tosa.max_pool2d",
78 {{{Profile::pro_int}, {{{i8T, i8T}, SpecificationVersion::V_1_0}}},
79 {{Profile::pro_fp},
80 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
81 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
82 {"tosa.max_pool2d_adaptive",
83 {{{Profile::pro_int}, {{{i8T, i8T}, SpecificationVersion::V_1_1_DRAFT}}},
84 {{Profile::pro_fp},
85 {{{fp16T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
86 {{fp32T, fp32T}, SpecificationVersion::V_1_1_DRAFT}}}}},
87 {"tosa.transpose_conv2d",
88 {{{Profile::pro_int},
89 {{{i8T, i8T, i32T, i8T, i8T, i32T, i32T}, SpecificationVersion::V_1_0}}},
90 {{Profile::pro_fp},
91 {{{fp16T, fp16T, fp16T, fp16T, fp16T, fp16T, fp16T},
92 SpecificationVersion::V_1_0},
93 {{fp16T, fp16T, fp16T, fp16T, fp16T, fp32T, fp16T},
94 SpecificationVersion::V_1_0},
95 {{fp32T, fp32T, fp32T, fp32T, fp32T, fp32T, fp32T},
96 SpecificationVersion::V_1_0}}}}},
97 {"tosa.clamp",
98 {{{Profile::pro_int}, {{{i8T, i8T}, SpecificationVersion::V_1_0}}},
99 {{Profile::pro_fp},
100 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
101 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
102 {"tosa.erf",
103 {{{Profile::pro_fp},
104 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
105 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
106 {"tosa.sigmoid",
107 {{{Profile::pro_fp},
108 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
109 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
110 {"tosa.tanh",
111 {{{Profile::pro_fp},
112 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
113 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
114 {"tosa.add",
115 {{{Profile::pro_int, Profile::pro_fp},
116 {{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}},
117 anyOf},
118 {{Profile::pro_fp},
119 {{{fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
120 {{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
121 {"tosa.arithmetic_right_shift",
122 {{{Profile::pro_int},
123 {{{i8T, i8T, i8T}, SpecificationVersion::V_1_0},
124 {{i16T, i16T, i16T}, SpecificationVersion::V_1_0},
125 {{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}}}},
126 {"tosa.bitwise_and",
127 {{{Profile::pro_int},
128 {{{i8T, i8T, i8T}, SpecificationVersion::V_1_0},
129 {{i16T, i16T, i16T}, SpecificationVersion::V_1_0},
130 {{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}}}},
131 {"tosa.bitwise_or",
132 {{{Profile::pro_int},
133 {{{i8T, i8T, i8T}, SpecificationVersion::V_1_0},
134 {{i16T, i16T, i16T}, SpecificationVersion::V_1_0},
135 {{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}}}},
136 {"tosa.bitwise_xor",
137 {{{Profile::pro_int},
138 {{{i8T, i8T, i8T}, SpecificationVersion::V_1_0},
139 {{i16T, i16T, i16T}, SpecificationVersion::V_1_0},
140 {{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}}}},
141 {"tosa.intdiv",
142 {{{Profile::pro_int, Profile::pro_fp},
143 {{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}},
144 anyOf}}},
145 {"tosa.logical_and",
146 {{{Profile::pro_int, Profile::pro_fp},
147 {{{boolT, boolT, boolT}, SpecificationVersion::V_1_0}},
148 anyOf}}},
149 {"tosa.logical_left_shift",
150 {{{Profile::pro_int, Profile::pro_fp},
151 {{{i8T, i8T, i8T}, SpecificationVersion::V_1_0},
152 {{i16T, i16T, i16T}, SpecificationVersion::V_1_0},
153 {{i32T, i32T, i32T}, SpecificationVersion::V_1_0}},
154 anyOf}}},
155 {"tosa.logical_right_shift",
156 {{{Profile::pro_int, Profile::pro_fp},
157 {{{i8T, i8T, i8T}, SpecificationVersion::V_1_0},
158 {{i16T, i16T, i16T}, SpecificationVersion::V_1_0},
159 {{i32T, i32T, i32T}, SpecificationVersion::V_1_0}},
160 anyOf}}},
161 {"tosa.logical_or",
162 {{{Profile::pro_int, Profile::pro_fp},
163 {{{boolT, boolT, boolT}, SpecificationVersion::V_1_0}},
164 anyOf}}},
165 {"tosa.logical_xor",
166 {{{Profile::pro_int, Profile::pro_fp},
167 {{{boolT, boolT, boolT}, SpecificationVersion::V_1_0}},
168 anyOf}}},
169 {"tosa.maximum",
170 {{{Profile::pro_int}, {{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}},
171 {{Profile::pro_fp},
172 {{{fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
173 {{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
174 {"tosa.minimum",
175 {{{Profile::pro_int}, {{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}},
176 {{Profile::pro_fp},
177 {{{fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
178 {{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
179 {"tosa.mul",
180 {{{Profile::pro_int},
181 {{{i8T, i8T, i32T}, SpecificationVersion::V_1_0},
182 {{i16T, i16T, i32T}, SpecificationVersion::V_1_0}}},
183 {{Profile::pro_int, Profile::pro_fp},
184 {{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}},
185 anyOf},
186 {{Profile::pro_fp},
187 {{{fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
188 {{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
189 {"tosa.pow",
190 {{{Profile::pro_fp},
191 {{{fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
192 {{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
193 {"tosa.sub",
194 {{{Profile::pro_int, Profile::pro_fp},
195 {{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}},
196 anyOf},
197 {{Profile::pro_fp},
198 {{{fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
199 {{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
200 {"tosa.table",
201 {{{Profile::pro_int}, {{{i8T, i8T, i8T}, SpecificationVersion::V_1_0}}}}},
202 {"tosa.abs",
203 {{{Profile::pro_int}, {{{i32T, i32T}, SpecificationVersion::V_1_0}}},
204 {{Profile::pro_fp},
205 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
206 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
207 {"tosa.bitwise_not",
208 {{{Profile::pro_int},
209 {{{i8T, i8T}, SpecificationVersion::V_1_0},
210 {{i16T, i16T}, SpecificationVersion::V_1_0},
211 {{i32T, i32T}, SpecificationVersion::V_1_0}}}}},
212 {"tosa.ceil",
213 {{{Profile::pro_fp},
214 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
215 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
216 {"tosa.clz",
217 {{{Profile::pro_int}, {{{i32T, i32T}, SpecificationVersion::V_1_0}}}}},
218 {"tosa.cos",
219 {{{Profile::pro_fp},
220 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
221 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
222 {"tosa.exp",
223 {{{Profile::pro_fp},
224 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
225 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
226 {"tosa.floor",
227 {{{Profile::pro_fp},
228 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
229 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
230 {"tosa.log",
231 {{{Profile::pro_fp},
232 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
233 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
234 {"tosa.logical_not",
235 {{{Profile::pro_int, Profile::pro_fp},
236 {{{boolT, boolT}, SpecificationVersion::V_1_0}},
237 anyOf}}},
238 {"tosa.negate",
239 {{{Profile::pro_int},
240 {{{i8T, i8T, i8T, i8T}, SpecificationVersion::V_1_0},
241 {{i16T, i16T, i16T, i16T}, SpecificationVersion::V_1_0},
242 {{i32T, i32T, i32T, i32T}, SpecificationVersion::V_1_0}}},
243 {{Profile::pro_fp},
244 {{{fp16T, fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
245 {{fp32T, fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
246 {"tosa.reciprocal",
247 {{{Profile::pro_fp},
248 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
249 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
250 {"tosa.rsqrt",
251 {{{Profile::pro_fp},
252 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
253 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
254 {"tosa.sin",
255 {{{Profile::pro_fp},
256 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
257 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
258 {"tosa.select",
259 {{{Profile::pro_int, Profile::pro_fp},
260 {{{boolT, boolT, boolT}, SpecificationVersion::V_1_0}},
261 anyOf},
262 {{Profile::pro_int},
263 {{{i8T, i8T, i8T}, SpecificationVersion::V_1_0},
264 {{i16T, i16T, i16T}, SpecificationVersion::V_1_0},
265 {{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}},
266 {{Profile::pro_fp},
267 {{{fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
268 {{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
269 {"tosa.equal",
270 {{{Profile::pro_int},
271 {{{i32T, i32T, boolT}, SpecificationVersion::V_1_0}}},
272 {{Profile::pro_fp},
273 {{{fp16T, fp16T, boolT}, SpecificationVersion::V_1_0},
274 {{fp32T, fp32T, boolT}, SpecificationVersion::V_1_0}}}}},
275 {"tosa.greater",
276 {{{Profile::pro_int},
277 {{{i32T, i32T, boolT}, SpecificationVersion::V_1_0}}},
278 {{Profile::pro_fp},
279 {{{fp16T, fp16T, boolT}, SpecificationVersion::V_1_0},
280 {{fp32T, fp32T, boolT}, SpecificationVersion::V_1_0}}}}},
281 {"tosa.greater_equal",
282 {{{Profile::pro_int},
283 {{{i32T, i32T, boolT}, SpecificationVersion::V_1_0}}},
284 {{Profile::pro_fp},
285 {{{fp16T, fp16T, boolT}, SpecificationVersion::V_1_0},
286 {{fp32T, fp32T, boolT}, SpecificationVersion::V_1_0}}}}},
287 {"tosa.reduce_all",
288 {{{Profile::pro_int, Profile::pro_fp},
289 {{{boolT, boolT}, SpecificationVersion::V_1_0}},
290 anyOf}}},
291 {"tosa.reduce_any",
292 {{{Profile::pro_int, Profile::pro_fp},
293 {{{boolT, boolT}, SpecificationVersion::V_1_0}},
294 anyOf}}},
295 {"tosa.reduce_max",
296 {{{Profile::pro_int},
297 {{{i8T, i8T}, SpecificationVersion::V_1_0},
298 {{i16T, i16T}, SpecificationVersion::V_1_0},
299 {{i32T, i32T}, SpecificationVersion::V_1_0}}},
300 {{Profile::pro_fp},
301 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
302 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
303 {"tosa.reduce_min",
304 {{{Profile::pro_int},
305 {{{i8T, i8T}, SpecificationVersion::V_1_0},
306 {{i16T, i16T}, SpecificationVersion::V_1_0},
307 {{i32T, i32T}, SpecificationVersion::V_1_0}}},
308 {{Profile::pro_fp},
309 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
310 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
311 {"tosa.reduce_product",
312 {{{Profile::pro_fp},
313 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
314 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
315 {"tosa.reduce_sum",
316 {{{Profile::pro_int}, {{{i32T, i32T}, SpecificationVersion::V_1_0}}},
317 {{Profile::pro_fp},
318 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
319 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
320 {"tosa.concat",
321 {{{Profile::pro_int, Profile::pro_fp},
322 {{{boolT, boolT}, SpecificationVersion::V_1_0}},
323 anyOf},
324 {{Profile::pro_int},
325 {{{i8T, i8T}, SpecificationVersion::V_1_0},
326 {{i32T, i32T}, SpecificationVersion::V_1_0}}},
327 {{Profile::pro_fp},
328 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
329 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
330 {"tosa.pad",
331 {{{Profile::pro_int, Profile::pro_fp},
332 {{{boolT, boolT, boolT}, SpecificationVersion::V_1_0}},
333 anyOf},
334 {{Profile::pro_int},
335 {{{i8T, i8T, i8T}, SpecificationVersion::V_1_0},
336 {{i16T, i16T, i16T}, SpecificationVersion::V_1_0},
337 {{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}},
338 {{Profile::pro_fp},
339 {{{fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
340 {{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
341 {"tosa.reshape",
342 {{{Profile::pro_int, Profile::pro_fp},
343 {{{boolT, boolT}, SpecificationVersion::V_1_0}},
344 anyOf},
345 {{Profile::pro_int},
346 {{{i8T, i8T}, SpecificationVersion::V_1_0},
347 {{i16T, i16T}, SpecificationVersion::V_1_0},
348 {{i32T, i32T}, SpecificationVersion::V_1_0}}},
349 {{Profile::pro_fp},
350 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
351 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
352 {"tosa.reverse",
353 {{{Profile::pro_int, Profile::pro_fp},
354 {{{boolT, boolT}, SpecificationVersion::V_1_0}},
355 anyOf},
356 {{Profile::pro_int},
357 {{{i8T, i8T}, SpecificationVersion::V_1_0},
358 {{i16T, i16T}, SpecificationVersion::V_1_0},
359 {{i32T, i32T}, SpecificationVersion::V_1_0}}},
360 {{Profile::pro_fp},
361 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
362 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
363 {"tosa.slice",
364 {{{Profile::pro_int, Profile::pro_fp},
365 {{{boolT, boolT}, SpecificationVersion::V_1_0}},
366 anyOf},
367 {{Profile::pro_int},
368 {{{i8T, i8T}, SpecificationVersion::V_1_0},
369 {{i16T, i16T}, SpecificationVersion::V_1_0},
370 {{i32T, i32T}, SpecificationVersion::V_1_0}}},
371 {{Profile::pro_fp},
372 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
373 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
374 {"tosa.tile",
375 {{{Profile::pro_int, Profile::pro_fp},
376 {{{boolT, boolT}, SpecificationVersion::V_1_0}},
377 anyOf},
378 {{Profile::pro_int},
379 {{{i8T, i8T}, SpecificationVersion::V_1_0},
380 {{i16T, i16T}, SpecificationVersion::V_1_0},
381 {{i32T, i32T}, SpecificationVersion::V_1_0}}},
382 {{Profile::pro_fp},
383 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
384 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
385 {"tosa.transpose",
386 {{{Profile::pro_int, Profile::pro_fp},
387 {{{boolT, boolT}, SpecificationVersion::V_1_0}},
388 anyOf},
389 {{Profile::pro_int},
390 {{{i8T, i8T}, SpecificationVersion::V_1_0},
391 {{i16T, i16T}, SpecificationVersion::V_1_0},
392 {{i32T, i32T}, SpecificationVersion::V_1_0}}},
393 {{Profile::pro_fp},
394 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
395 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
396 {"tosa.gather",
397 {{{Profile::pro_int},
398 {{{i8T, i32T, i8T}, SpecificationVersion::V_1_0},
399 {{i16T, i32T, i16T}, SpecificationVersion::V_1_0},
400 {{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}},
401 {{Profile::pro_fp},
402 {{{i8T, i32T, i8T}, SpecificationVersion::V_1_1_DRAFT},
403 {{i16T, i32T, i16T}, SpecificationVersion::V_1_1_DRAFT},
404 {{i32T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT},
405 {{fp16T, i32T, fp16T}, SpecificationVersion::V_1_0},
406 {{fp32T, i32T, fp32T}, SpecificationVersion::V_1_0}}},
407 {{Profile::pro_fp, Profile::pro_int},
408 {{{boolT, i32T, boolT}, SpecificationVersion::V_1_1_DRAFT},
409 {{i8T, i64T, i8T}, SpecificationVersion::V_1_1_DRAFT},
410 {{i16T, i64T, i16T}, SpecificationVersion::V_1_1_DRAFT},
411 {{i32T, i64T, i32T}, SpecificationVersion::V_1_1_DRAFT}},
412 anyOf}}},
413 {"tosa.row_gather",
414 {{{Profile::pro_int},
415 {{{boolT, i32T, boolT}, SpecificationVersion::V_1_1_DRAFT},
416 {{i8T, i32T, i8T}, SpecificationVersion::V_1_1_DRAFT},
417 {{i16T, i32T, i16T}, SpecificationVersion::V_1_1_DRAFT},
418 {{i32T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT},
419 {{i64T, i32T, i64T}, SpecificationVersion::V_1_1_DRAFT},
420 {{boolT, i64T, boolT}, SpecificationVersion::V_1_1_DRAFT},
421 {{i8T, i64T, i8T}, SpecificationVersion::V_1_1_DRAFT},
422 {{i16T, i64T, i16T}, SpecificationVersion::V_1_1_DRAFT},
423 {{i32T, i64T, i32T}, SpecificationVersion::V_1_1_DRAFT},
424 {{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
425 {{Profile::pro_fp},
426 {{{boolT, i32T, boolT}, SpecificationVersion::V_1_1_DRAFT},
427 {{i8T, i32T, i8T}, SpecificationVersion::V_1_1_DRAFT},
428 {{i16T, i32T, i16T}, SpecificationVersion::V_1_1_DRAFT},
429 {{i32T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT},
430 {{i64T, i32T, i64T}, SpecificationVersion::V_1_1_DRAFT},
431 {{fp16T, i32T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
432 {{fp32T, i32T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
433 {{boolT, i64T, boolT}, SpecificationVersion::V_1_1_DRAFT},
434 {{i8T, i64T, i8T}, SpecificationVersion::V_1_1_DRAFT},
435 {{i16T, i64T, i16T}, SpecificationVersion::V_1_1_DRAFT},
436 {{i32T, i64T, i32T}, SpecificationVersion::V_1_1_DRAFT},
437 {{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT},
438 {{fp16T, i64T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
439 {{fp32T, i64T, fp32T}, SpecificationVersion::V_1_1_DRAFT}}}}},
440 {"tosa.row_gather_block_scaled",
441 {{{Profile::pro_int},
442 {{{i8T, i32T, i32T, i8T}, SpecificationVersion::V_1_1_DRAFT},
443 {{i16T, i32T, i32T, i16T}, SpecificationVersion::V_1_1_DRAFT},
444 {{i32T, i32T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT}}},
445 {{Profile::pro_fp},
446 {{{i8T, i32T, i32T, i8T}, SpecificationVersion::V_1_1_DRAFT},
447 {{i16T, i32T, i32T, i16T}, SpecificationVersion::V_1_1_DRAFT},
448 {{i32T, i32T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT},
449 {{fp16T, i32T, i32T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
450 {{fp32T, i32T, i32T, fp32T}, SpecificationVersion::V_1_1_DRAFT}}},
451 {{Profile::pro_fp, Profile::pro_int},
452 {{{boolT, i32T, i32T, boolT}, SpecificationVersion::V_1_1_DRAFT},
453 {{i8T, i64T, i32T, i8T}, SpecificationVersion::V_1_1_DRAFT},
454 {{i16T, i64T, i32T, i16T}, SpecificationVersion::V_1_1_DRAFT},
455 {{i32T, i64T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT},
456 {{i64T, i64T, i32T, i64T}, SpecificationVersion::V_1_1_DRAFT},
457 {{boolT, i64T, i32T, boolT}, SpecificationVersion::V_1_1_DRAFT}},
458 anyOf}}},
459 {"tosa.scatter",
460 {{{Profile::pro_int},
461 {{{i8T, i32T, i8T, i8T}, SpecificationVersion::V_1_0},
462 {{i16T, i32T, i16T, i16T}, SpecificationVersion::V_1_0},
463 {{i32T, i32T, i32T, i32T}, SpecificationVersion::V_1_0}}},
464 {{Profile::pro_fp},
465 {{{i8T, i32T, i8T, i8T}, SpecificationVersion::V_1_1_DRAFT},
466 {{i16T, i32T, i16T, i16T}, SpecificationVersion::V_1_1_DRAFT},
467 {{i32T, i32T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT},
468 {{fp16T, i32T, fp16T, fp16T}, SpecificationVersion::V_1_0},
469 {{fp32T, i32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}},
470 {{Profile::pro_fp, Profile::pro_int},
471 {{{boolT, i32T, boolT, boolT}, SpecificationVersion::V_1_1_DRAFT},
472 {{i8T, i64T, i8T, i8T}, SpecificationVersion::V_1_1_DRAFT},
473 {{i16T, i64T, i16T, i16T}, SpecificationVersion::V_1_1_DRAFT},
474 {{i32T, i64T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT}},
475 anyOf}}},
476 {"tosa.resize",
477 {{{Profile::pro_int},
478 {{{i8T, i32T}, SpecificationVersion::V_1_0},
479 {{i8T, i8T}, SpecificationVersion::V_1_0}}},
480 {{Profile::pro_fp},
481 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
482 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
483 {"tosa.cast",
484 {{{Profile::pro_int},
485 {{{boolT, i8T}, SpecificationVersion::V_1_0},
486 {{boolT, i16T}, SpecificationVersion::V_1_0},
487 {{boolT, i32T}, SpecificationVersion::V_1_0},
488 {{i8T, boolT}, SpecificationVersion::V_1_0},
489 {{i8T, i16T}, SpecificationVersion::V_1_0},
490 {{i8T, i32T}, SpecificationVersion::V_1_0},
491 {{i16T, boolT}, SpecificationVersion::V_1_0},
492 {{i16T, i8T}, SpecificationVersion::V_1_0},
493 {{i16T, i32T}, SpecificationVersion::V_1_0},
494 {{i32T, boolT}, SpecificationVersion::V_1_0},
495 {{i32T, i8T}, SpecificationVersion::V_1_0},
496 {{i32T, i16T}, SpecificationVersion::V_1_0}}},
497 {{Profile::pro_fp},
498 {{{boolT, fp32T}, SpecificationVersion::V_1_1_DRAFT},
499 {{i8T, fp16T}, SpecificationVersion::V_1_0},
500 {{i8T, fp32T}, SpecificationVersion::V_1_0},
501 {{i16T, fp16T}, SpecificationVersion::V_1_0},
502 {{i16T, fp32T}, SpecificationVersion::V_1_0},
503 {{i32T, fp16T}, SpecificationVersion::V_1_0},
504 {{i32T, fp32T}, SpecificationVersion::V_1_0},
505 {{fp16T, i8T}, SpecificationVersion::V_1_0},
506 {{fp16T, i16T}, SpecificationVersion::V_1_0},
507 {{fp16T, i32T}, SpecificationVersion::V_1_0},
508 {{fp16T, fp32T}, SpecificationVersion::V_1_0},
509 {{fp32T, i8T}, SpecificationVersion::V_1_0},
510 {{fp32T, i16T}, SpecificationVersion::V_1_0},
511 {{fp32T, i32T}, SpecificationVersion::V_1_0},
512 {{fp32T, fp16T}, SpecificationVersion::V_1_0},
513 {{fp32T, boolT}, SpecificationVersion::V_1_1_DRAFT}}}}},
514 {"tosa.rescale",
515 {{{Profile::pro_int},
516 {{{i8T, i8T, i8T, i8T}, SpecificationVersion::V_1_0},
517 {{i8T, i8T, i16T, i16T}, SpecificationVersion::V_1_0},
518 {{i8T, i8T, i32T, i32T}, SpecificationVersion::V_1_0},
519 {{i16T, i16T, i8T, i8T}, SpecificationVersion::V_1_0},
520 {{i16T, i16T, i16T, i16T}, SpecificationVersion::V_1_0},
521 {{i16T, i16T, i32T, i32T}, SpecificationVersion::V_1_0},
522 {{i32T, i32T, i8T, i8T}, SpecificationVersion::V_1_0},
523 {{i32T, i32T, i16T, i16T}, SpecificationVersion::V_1_0},
524 {{i32T, i32T, i32T, i32T}, SpecificationVersion::V_1_0}}}}},
525 {"tosa.const",
526 {{{Profile::pro_int, Profile::pro_fp},
527 {{{boolT}, SpecificationVersion::V_1_0},
528 {{i8T}, SpecificationVersion::V_1_0},
529 {{i16T}, SpecificationVersion::V_1_0},
530 {{i32T}, SpecificationVersion::V_1_0}},
531 anyOf},
532 {{Profile::pro_fp},
533 {{{fp16T}, SpecificationVersion::V_1_0},
534 {{fp32T}, SpecificationVersion::V_1_0}}}}},
535 {"tosa.identity",
536 {{{Profile::pro_int, Profile::pro_fp},
537 {{{boolT, boolT}, SpecificationVersion::V_1_0},
538 {{i8T, i8T}, SpecificationVersion::V_1_0},
539 {{i16T, i16T}, SpecificationVersion::V_1_0},
540 {{i32T, i32T}, SpecificationVersion::V_1_0}},
541 anyOf},
542 {{Profile::pro_fp},
543 {{{fp16T, fp16T}, SpecificationVersion::V_1_0},
544 {{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
545 {"tosa.variable",
546 {{{Profile::pro_int}, {{{i8T}, SpecificationVersion::V_1_0}}},
547 {{Profile::pro_fp},
548 {{{fp16T}, SpecificationVersion::V_1_0},
549 {{fp32T}, SpecificationVersion::V_1_0}}}}},
550 {"tosa.variable_write",
551 {{{Profile::pro_int}, {{{i8T}, SpecificationVersion::V_1_0}}},
552 {{Profile::pro_fp},
553 {{{fp16T}, SpecificationVersion::V_1_0},
554 {{fp32T}, SpecificationVersion::V_1_0}}}}},
555 {"tosa.variable_read",
556 {{{Profile::pro_int}, {{{i8T}, SpecificationVersion::V_1_0}}},
557 {{Profile::pro_fp},
558 {{{fp16T}, SpecificationVersion::V_1_0},
559 {{fp32T}, SpecificationVersion::V_1_0}}}}},
560 {"tosa.custom", {{{Profile::pro_int, Profile::pro_fp}, {}, anyOf}}},
561 {"tosa.const_shape", {{{Profile::pro_int, Profile::pro_fp}, {}, anyOf}}},
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, fp16T, fp8e4m3T, fp8e4m3T, fp32T, fp16T},
619 SpecificationVersion::V_1_1_DRAFT},
620 {{fp8e4m3T, fp16T, fp16T, fp8e4m3T, fp16T, fp32T, fp16T},
621 SpecificationVersion::V_1_1_DRAFT},
622 {{fp16T, fp8e4m3T, fp16T, fp16T, fp8e4m3T, fp32T, fp16T},
623 SpecificationVersion::V_1_1_DRAFT},
624 {{fp8e4m3T, fp8e4m3T, fp32T, fp8e4m3T, fp8e4m3T, fp32T, fp32T},
625 SpecificationVersion::V_1_1_DRAFT},
626 {{fp8e4m3T, fp16T, fp32T, fp8e4m3T, fp16T, fp32T, fp32T},
627 SpecificationVersion::V_1_1_DRAFT},
628 {{fp8e4m3T, fp32T, fp32T, fp8e4m3T, fp32T, fp32T, fp32T},
629 SpecificationVersion::V_1_1_DRAFT},
630 {{fp16T, fp8e4m3T, fp32T, fp16T, fp8e4m3T, fp32T, fp32T},
631 SpecificationVersion::V_1_1_DRAFT},
632 {{fp32T, fp8e4m3T, fp32T, fp32T, fp8e4m3T, fp32T, fp32T},
633 SpecificationVersion::V_1_1_DRAFT}}},
634 {{Extension::fp8e5m2},
635 {{{fp8e5m2T, fp8e5m2T, fp16T, fp8e5m2T, fp8e5m2T, fp16T, fp16T},
636 SpecificationVersion::V_1_0},
637 {{fp8e5m2T, fp8e5m2T, fp16T, fp8e5m2T, fp8e5m2T, fp32T, fp16T},
638 SpecificationVersion::V_1_1_DRAFT},
639 {{fp8e5m2T, fp16T, fp16T, fp8e5m2T, fp16T, fp32T, fp16T},
640 SpecificationVersion::V_1_1_DRAFT},
641 {{fp16T, fp8e5m2T, fp16T, fp16T, fp8e5m2T, fp32T, fp16T},
642 SpecificationVersion::V_1_1_DRAFT},
643 {{fp8e5m2T, fp8e5m2T, fp32T, fp8e5m2T, fp8e5m2T, fp32T, fp32T},
644 SpecificationVersion::V_1_1_DRAFT},
645 {{fp8e5m2T, fp16T, fp32T, fp8e5m2T, fp16T, fp32T, fp32T},
646 SpecificationVersion::V_1_1_DRAFT},
647 {{fp8e5m2T, fp32T, fp32T, fp8e5m2T, fp32T, fp32T, fp32T},
648 SpecificationVersion::V_1_1_DRAFT},
649 {{fp16T, fp8e5m2T, fp32T, fp16T, fp8e5m2T, fp32T, fp32T},
650 SpecificationVersion::V_1_1_DRAFT},
651 {{fp32T, fp8e5m2T, fp32T, fp32T, fp8e5m2T, fp32T, fp32T},
652 SpecificationVersion::V_1_1_DRAFT}}},
653 {{Extension::bf16},
654 {{{bf16T, bf16T, bf16T, bf16T, bf16T, fp32T, bf16T},
655 SpecificationVersion::V_1_0},
656 {{fp16T, fp16T, fp16T, fp16T, fp16T, bf16T, fp16T},
657 SpecificationVersion::V_1_1_DRAFT},
658 {{fp16T, bf16T, fp16T, fp16T, bf16T, bf16T, fp16T},
659 SpecificationVersion::V_1_1_DRAFT},
660 {{fp16T, bf16T, fp16T, fp16T, bf16T, fp32T, fp16T},
661 SpecificationVersion::V_1_1_DRAFT},
662 {{bf16T, fp16T, fp16T, bf16T, fp16T, bf16T, fp16T},
663 SpecificationVersion::V_1_1_DRAFT},
664 {{bf16T, fp16T, fp16T, bf16T, fp16T, fp32T, fp16T},
665 SpecificationVersion::V_1_1_DRAFT},
666 {{bf16T, bf16T, fp16T, bf16T, bf16T, bf16T, fp16T},
667 SpecificationVersion::V_1_1_DRAFT},
668 {{bf16T, bf16T, fp16T, bf16T, bf16T, fp32T, fp16T},
669 SpecificationVersion::V_1_1_DRAFT},
670 {{fp16T, bf16T, fp32T, fp16T, bf16T, fp32T, fp32T},
671 SpecificationVersion::V_1_1_DRAFT},
672 {{bf16T, fp16T, fp32T, bf16T, fp16T, fp32T, fp32T},
673 SpecificationVersion::V_1_1_DRAFT},
674 {{bf16T, bf16T, fp32T, bf16T, bf16T, fp32T, fp32T},
675 SpecificationVersion::V_1_1_DRAFT},
676 {{bf16T, fp32T, fp32T, bf16T, fp32T, fp32T, fp32T},
677 SpecificationVersion::V_1_1_DRAFT},
678 {{fp32T, bf16T, fp32T, fp32T, bf16T, fp32T, fp32T},
679 SpecificationVersion::V_1_1_DRAFT}}},
680 {{Extension::bf16, Extension::fp8e4m3},
681 {{{fp8e4m3T, fp8e4m3T, fp16T, fp8e4m3T, fp8e4m3T, bf16T, fp16T},
682 SpecificationVersion::V_1_1_DRAFT},
683 {{fp8e4m3T, fp16T, fp16T, fp8e4m3T, fp16T, bf16T, fp16T},
684 SpecificationVersion::V_1_1_DRAFT},
685 {{fp8e4m3T, bf16T, fp16T, fp8e4m3T, bf16T, bf16T, fp16T},
686 SpecificationVersion::V_1_1_DRAFT},
687 {{fp8e4m3T, bf16T, fp16T, fp8e4m3T, bf16T, fp32T, fp16T},
688 SpecificationVersion::V_1_1_DRAFT},
689 {{fp16T, fp8e4m3T, fp16T, fp16T, fp8e4m3T, bf16T, fp16T},
690 SpecificationVersion::V_1_1_DRAFT},
691 {{bf16T, fp8e4m3T, fp16T, bf16T, fp8e4m3T, bf16T, fp16T},
692 SpecificationVersion::V_1_1_DRAFT},
693 {{bf16T, fp8e4m3T, fp16T, bf16T, fp8e4m3T, fp32T, fp16T},
694 SpecificationVersion::V_1_1_DRAFT},
695 {{fp8e4m3T, bf16T, fp32T, fp8e4m3T, bf16T, fp32T, fp32T},
696 SpecificationVersion::V_1_1_DRAFT},
697 {{bf16T, fp8e4m3T, fp32T, bf16T, fp8e4m3T, fp32T, fp32T},
698 SpecificationVersion::V_1_1_DRAFT}},
699 allOf},
700 {{Extension::bf16, Extension::fp8e4m3, Extension::fp8e5m2},
701 {{{fp8e4m3T, fp8e5m2T, fp16T, fp8e4m3T, fp8e5m2T, bf16T, fp16T},
702 SpecificationVersion::V_1_1_DRAFT},
703 {{fp8e5m2T, fp8e4m3T, fp16T, fp8e5m2T, fp8e4m3T, bf16T, fp16T},
704 SpecificationVersion::V_1_1_DRAFT}},
705 allOf},
706 {{Extension::fp8e4m3, Extension::fp8e5m2},
707 {{{fp8e4m3T, fp8e5m2T, fp16T, fp8e4m3T, fp8e5m2T, fp32T, fp16T},
708 SpecificationVersion::V_1_1_DRAFT},
709 {{fp8e5m2T, fp8e4m3T, fp16T, fp8e5m2T, fp8e4m3T, fp32T, fp16T},
710 SpecificationVersion::V_1_1_DRAFT},
711 {{fp8e4m3T, fp8e5m2T, fp32T, fp8e4m3T, fp8e5m2T, fp32T, fp32T},
712 SpecificationVersion::V_1_1_DRAFT},
713 {{fp8e5m2T, fp8e4m3T, fp32T, fp8e5m2T, fp8e4m3T, fp32T, fp32T},
714 SpecificationVersion::V_1_1_DRAFT}},
715 allOf},
716 {{Extension::bf16, Extension::fp8e4m3, Extension::mx_common,
717 Extension::mx_fp8e4m3},
718 {{{fp8e4m3T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp8e4m3T, fp32T, bf16T,
719 fp16T},
720 SpecificationVersion::V_1_1_DRAFT},
721 {{bs32_fp8ue8m0_fp8e4m3T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, bf16T,
722 fp16T},
723 SpecificationVersion::V_1_1_DRAFT}},
724 allOf},
725 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp8e4m3},
726 {{{fp8e4m3T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp8e4m3T, fp32T, fp32T,
727 fp16T},
728 SpecificationVersion::V_1_1_DRAFT},
729 {{bs32_fp8ue8m0_fp8e4m3T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, fp32T,
730 fp16T},
731 SpecificationVersion::V_1_1_DRAFT},
732 {{fp8e4m3T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp8e4m3T, fp32T, fp32T,
733 fp32T},
734 SpecificationVersion::V_1_1_DRAFT},
735 {{bs32_fp8ue8m0_fp8e4m3T, fp8e4m3T, fp32T, fp32T, fp8e4m3T, fp32T,
736 fp32T},
737 SpecificationVersion::V_1_1_DRAFT}},
738 allOf},
739 {{Extension::bf16, Extension::fp8e4m3, Extension::mx_common,
740 Extension::mx_fp8e5m2},
741 {{{fp8e4m3T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp8e4m3T, fp32T, bf16T,
742 fp16T},
743 SpecificationVersion::V_1_1_DRAFT},
744 {{bs32_fp8ue8m0_fp8e5m2T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, bf16T,
745 fp16T},
746 SpecificationVersion::V_1_1_DRAFT}},
747 allOf},
748 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp8e5m2},
749 {{{fp8e4m3T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp8e4m3T, fp32T, fp32T,
750 fp16T},
751 SpecificationVersion::V_1_1_DRAFT},
752 {{bs32_fp8ue8m0_fp8e5m2T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, fp32T,
753 fp16T},
754 SpecificationVersion::V_1_1_DRAFT},
755 {{fp8e4m3T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp8e4m3T, fp32T, fp32T,
756 fp32T},
757 SpecificationVersion::V_1_1_DRAFT},
758 {{bs32_fp8ue8m0_fp8e5m2T, fp8e4m3T, fp32T, fp32T, fp8e4m3T, fp32T,
759 fp32T},
760 SpecificationVersion::V_1_1_DRAFT}},
761 allOf},
762 {{Extension::bf16, Extension::fp8e4m3, Extension::mx_common,
763 Extension::mx_fp6e3m2},
764 {{{fp8e4m3T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp8e4m3T, fp32T, bf16T,
765 fp16T},
766 SpecificationVersion::V_1_1_DRAFT},
767 {{bs32_fp8ue8m0_fp6e3m2T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, bf16T,
768 fp16T},
769 SpecificationVersion::V_1_1_DRAFT}},
770 allOf},
771 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp6e3m2},
772 {{{fp8e4m3T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp8e4m3T, fp32T, fp32T,
773 fp16T},
774 SpecificationVersion::V_1_1_DRAFT},
775 {{bs32_fp8ue8m0_fp6e3m2T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, fp32T,
776 fp16T},
777 SpecificationVersion::V_1_1_DRAFT},
778 {{fp8e4m3T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp8e4m3T, fp32T, fp32T,
779 fp32T},
780 SpecificationVersion::V_1_1_DRAFT},
781 {{bs32_fp8ue8m0_fp6e3m2T, fp8e4m3T, fp32T, fp32T, fp8e4m3T, fp32T,
782 fp32T},
783 SpecificationVersion::V_1_1_DRAFT}},
784 allOf},
785 {{Extension::bf16, Extension::fp8e4m3, Extension::mx_common,
786 Extension::mx_fp6e2m3},
787 {{{fp8e4m3T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp8e4m3T, fp32T, bf16T,
788 fp16T},
789 SpecificationVersion::V_1_1_DRAFT},
790 {{bs32_fp8ue8m0_fp6e2m3T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, bf16T,
791 fp16T},
792 SpecificationVersion::V_1_1_DRAFT}},
793 allOf},
794 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp6e2m3},
795 {{{fp8e4m3T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp8e4m3T, fp32T, fp32T,
796 fp16T},
797 SpecificationVersion::V_1_1_DRAFT},
798 {{bs32_fp8ue8m0_fp6e2m3T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, fp32T,
799 fp16T},
800 SpecificationVersion::V_1_1_DRAFT},
801 {{fp8e4m3T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp8e4m3T, fp32T, fp32T,
802 fp32T},
803 SpecificationVersion::V_1_1_DRAFT},
804 {{bs32_fp8ue8m0_fp6e2m3T, fp8e4m3T, fp32T, fp32T, fp8e4m3T, fp32T,
805 fp32T},
806 SpecificationVersion::V_1_1_DRAFT}},
807 allOf},
808 {{Extension::bf16, Extension::fp8e4m3, Extension::mx_common,
809 Extension::mx_fp4e2m1},
810 {{{fp8e4m3T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp8e4m3T, fp32T, bf16T,
811 fp16T},
812 SpecificationVersion::V_1_1_DRAFT},
813 {{bs32_fp8ue8m0_fp4e2m1T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, bf16T,
814 fp16T},
815 SpecificationVersion::V_1_1_DRAFT}},
816 allOf},
817 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp4e2m1},
818 {{{fp8e4m3T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp8e4m3T, fp32T, fp32T,
819 fp16T},
820 SpecificationVersion::V_1_1_DRAFT},
821 {{bs32_fp8ue8m0_fp4e2m1T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, fp32T,
822 fp16T},
823 SpecificationVersion::V_1_1_DRAFT},
824 {{fp8e4m3T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp8e4m3T, fp32T, fp32T,
825 fp32T},
826 SpecificationVersion::V_1_1_DRAFT},
827 {{bs32_fp8ue8m0_fp4e2m1T, fp8e4m3T, fp32T, fp32T, fp8e4m3T, fp32T,
828 fp32T},
829 SpecificationVersion::V_1_1_DRAFT}},
830 allOf},
831 {{Extension::bf16, Extension::fp8e4m3, Extension::mx_common,
832 Extension::mx_int8},
833 {{{fp8e4m3T, bs32_fp8ue8m0_mxint8T, fp16T, fp8e4m3T, fp32T, bf16T,
834 fp16T},
835 SpecificationVersion::V_1_1_DRAFT},
836 {{bs32_fp8ue8m0_mxint8T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, bf16T,
837 fp16T},
838 SpecificationVersion::V_1_1_DRAFT}},
839 allOf},
840 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_int8},
841 {{{fp8e4m3T, bs32_fp8ue8m0_mxint8T, fp16T, fp8e4m3T, fp32T, fp32T,
842 fp16T},
843 SpecificationVersion::V_1_1_DRAFT},
844 {{bs32_fp8ue8m0_mxint8T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, fp32T,
845 fp16T},
846 SpecificationVersion::V_1_1_DRAFT},
847 {{fp8e4m3T, bs32_fp8ue8m0_mxint8T, fp32T, fp8e4m3T, fp32T, fp32T,
848 fp32T},
849 SpecificationVersion::V_1_1_DRAFT},
850 {{bs32_fp8ue8m0_mxint8T, fp8e4m3T, fp32T, fp32T, fp8e4m3T, fp32T,
851 fp32T},
852 SpecificationVersion::V_1_1_DRAFT}},
853 allOf},
854 {{Extension::bf16, Extension::fp8e5m2},
855 {{{fp8e5m2T, fp8e5m2T, fp16T, fp8e5m2T, fp8e5m2T, bf16T, fp16T},
856 SpecificationVersion::V_1_1_DRAFT},
857 {{fp8e5m2T, fp16T, fp16T, fp8e5m2T, fp16T, bf16T, fp16T},
858 SpecificationVersion::V_1_1_DRAFT},
859 {{fp8e5m2T, bf16T, fp16T, fp8e5m2T, bf16T, bf16T, fp16T},
860 SpecificationVersion::V_1_1_DRAFT},
861 {{fp8e5m2T, bf16T, fp16T, fp8e5m2T, bf16T, fp32T, fp16T},
862 SpecificationVersion::V_1_1_DRAFT},
863 {{fp16T, fp8e5m2T, fp16T, fp16T, fp8e5m2T, bf16T, fp16T},
864 SpecificationVersion::V_1_1_DRAFT},
865 {{bf16T, fp8e5m2T, fp16T, bf16T, fp8e5m2T, bf16T, fp16T},
866 SpecificationVersion::V_1_1_DRAFT},
867 {{bf16T, fp8e5m2T, fp16T, bf16T, fp8e5m2T, fp32T, fp16T},
868 SpecificationVersion::V_1_1_DRAFT},
869 {{fp8e5m2T, bf16T, fp32T, fp8e5m2T, bf16T, fp32T, fp32T},
870 SpecificationVersion::V_1_1_DRAFT},
871 {{bf16T, fp8e5m2T, fp32T, bf16T, fp8e5m2T, fp32T, fp32T},
872 SpecificationVersion::V_1_1_DRAFT}},
873 allOf},
874 {{Extension::bf16, Extension::fp8e5m2, Extension::mx_common,
875 Extension::mx_fp8e4m3},
876 {{{fp8e5m2T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp8e5m2T, fp32T, bf16T,
877 fp16T},
878 SpecificationVersion::V_1_1_DRAFT},
879 {{bs32_fp8ue8m0_fp8e4m3T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, bf16T,
880 fp16T},
881 SpecificationVersion::V_1_1_DRAFT}},
882 allOf},
883 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp8e4m3},
884 {{{fp8e5m2T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp8e5m2T, fp32T, fp32T,
885 fp16T},
886 SpecificationVersion::V_1_1_DRAFT},
887 {{bs32_fp8ue8m0_fp8e4m3T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, fp32T,
888 fp16T},
889 SpecificationVersion::V_1_1_DRAFT},
890 {{fp8e5m2T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp8e5m2T, fp32T, fp32T,
891 fp32T},
892 SpecificationVersion::V_1_1_DRAFT},
893 {{bs32_fp8ue8m0_fp8e4m3T, fp8e5m2T, fp32T, fp32T, fp8e5m2T, fp32T,
894 fp32T},
895 SpecificationVersion::V_1_1_DRAFT}},
896 allOf},
897 {{Extension::bf16, Extension::fp8e5m2, Extension::mx_common,
898 Extension::mx_fp8e5m2},
899 {{{fp8e5m2T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp8e5m2T, fp32T, bf16T,
900 fp16T},
901 SpecificationVersion::V_1_1_DRAFT},
902 {{bs32_fp8ue8m0_fp8e5m2T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, bf16T,
903 fp16T},
904 SpecificationVersion::V_1_1_DRAFT}},
905 allOf},
906 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp8e5m2},
907 {{{fp8e5m2T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp8e5m2T, fp32T, fp32T,
908 fp16T},
909 SpecificationVersion::V_1_1_DRAFT},
910 {{bs32_fp8ue8m0_fp8e5m2T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, fp32T,
911 fp16T},
912 SpecificationVersion::V_1_1_DRAFT},
913 {{fp8e5m2T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp8e5m2T, fp32T, fp32T,
914 fp32T},
915 SpecificationVersion::V_1_1_DRAFT},
916 {{bs32_fp8ue8m0_fp8e5m2T, fp8e5m2T, fp32T, fp32T, fp8e5m2T, fp32T,
917 fp32T},
918 SpecificationVersion::V_1_1_DRAFT}},
919 allOf},
920 {{Extension::bf16, Extension::fp8e5m2, Extension::mx_common,
921 Extension::mx_fp6e3m2},
922 {{{fp8e5m2T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp8e5m2T, fp32T, bf16T,
923 fp16T},
924 SpecificationVersion::V_1_1_DRAFT},
925 {{bs32_fp8ue8m0_fp6e3m2T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, bf16T,
926 fp16T},
927 SpecificationVersion::V_1_1_DRAFT}},
928 allOf},
929 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp6e3m2},
930 {{{fp8e5m2T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp8e5m2T, fp32T, fp32T,
931 fp16T},
932 SpecificationVersion::V_1_1_DRAFT},
933 {{bs32_fp8ue8m0_fp6e3m2T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, fp32T,
934 fp16T},
935 SpecificationVersion::V_1_1_DRAFT},
936 {{fp8e5m2T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp8e5m2T, fp32T, fp32T,
937 fp32T},
938 SpecificationVersion::V_1_1_DRAFT},
939 {{bs32_fp8ue8m0_fp6e3m2T, fp8e5m2T, fp32T, fp32T, fp8e5m2T, fp32T,
940 fp32T},
941 SpecificationVersion::V_1_1_DRAFT}},
942 allOf},
943 {{Extension::bf16, Extension::fp8e5m2, Extension::mx_common,
944 Extension::mx_fp6e2m3},
945 {{{fp8e5m2T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp8e5m2T, fp32T, bf16T,
946 fp16T},
947 SpecificationVersion::V_1_1_DRAFT},
948 {{bs32_fp8ue8m0_fp6e2m3T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, bf16T,
949 fp16T},
950 SpecificationVersion::V_1_1_DRAFT}},
951 allOf},
952 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp6e2m3},
953 {{{fp8e5m2T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp8e5m2T, fp32T, fp32T,
954 fp16T},
955 SpecificationVersion::V_1_1_DRAFT},
956 {{bs32_fp8ue8m0_fp6e2m3T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, fp32T,
957 fp16T},
958 SpecificationVersion::V_1_1_DRAFT},
959 {{fp8e5m2T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp8e5m2T, fp32T, fp32T,
960 fp32T},
961 SpecificationVersion::V_1_1_DRAFT},
962 {{bs32_fp8ue8m0_fp6e2m3T, fp8e5m2T, fp32T, fp32T, fp8e5m2T, fp32T,
963 fp32T},
964 SpecificationVersion::V_1_1_DRAFT}},
965 allOf},
966 {{Extension::bf16, Extension::fp8e5m2, Extension::mx_common,
967 Extension::mx_fp4e2m1},
968 {{{fp8e5m2T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp8e5m2T, fp32T, bf16T,
969 fp16T},
970 SpecificationVersion::V_1_1_DRAFT},
971 {{bs32_fp8ue8m0_fp4e2m1T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, bf16T,
972 fp16T},
973 SpecificationVersion::V_1_1_DRAFT}},
974 allOf},
975 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp4e2m1},
976 {{{fp8e5m2T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp8e5m2T, fp32T, fp32T,
977 fp16T},
978 SpecificationVersion::V_1_1_DRAFT},
979 {{bs32_fp8ue8m0_fp4e2m1T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, fp32T,
980 fp16T},
981 SpecificationVersion::V_1_1_DRAFT},
982 {{fp8e5m2T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp8e5m2T, fp32T, fp32T,
983 fp32T},
984 SpecificationVersion::V_1_1_DRAFT},
985 {{bs32_fp8ue8m0_fp4e2m1T, fp8e5m2T, fp32T, fp32T, fp8e5m2T, fp32T,
986 fp32T},
987 SpecificationVersion::V_1_1_DRAFT}},
988 allOf},
989 {{Extension::bf16, Extension::fp8e5m2, Extension::mx_common,
990 Extension::mx_int8},
991 {{{fp8e5m2T, bs32_fp8ue8m0_mxint8T, fp16T, fp8e5m2T, fp32T, bf16T,
992 fp16T},
993 SpecificationVersion::V_1_1_DRAFT},
994 {{bs32_fp8ue8m0_mxint8T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, bf16T,
995 fp16T},
996 SpecificationVersion::V_1_1_DRAFT}},
997 allOf},
998 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_int8},
999 {{{fp8e5m2T, bs32_fp8ue8m0_mxint8T, fp16T, fp8e5m2T, fp32T, fp32T,
1000 fp16T},
1001 SpecificationVersion::V_1_1_DRAFT},
1002 {{bs32_fp8ue8m0_mxint8T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, fp32T,
1003 fp16T},
1004 SpecificationVersion::V_1_1_DRAFT},
1005 {{fp8e5m2T, bs32_fp8ue8m0_mxint8T, fp32T, fp8e5m2T, fp32T, fp32T,
1006 fp32T},
1007 SpecificationVersion::V_1_1_DRAFT},
1008 {{bs32_fp8ue8m0_mxint8T, fp8e5m2T, fp32T, fp32T, fp8e5m2T, fp32T,
1009 fp32T},
1010 SpecificationVersion::V_1_1_DRAFT}},
1011 allOf},
1012 {{Extension::bf16, Extension::mx_common, Extension::mx_fp8e4m3},
1013 {{{fp16T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp16T, fp32T, bf16T, fp16T},
1014 SpecificationVersion::V_1_1_DRAFT},
1015 {{bf16T, bs32_fp8ue8m0_fp8e4m3T, fp16T, bf16T, fp32T, bf16T, fp16T},
1016 SpecificationVersion::V_1_1_DRAFT},
1017 {{bf16T, bs32_fp8ue8m0_fp8e4m3T, fp16T, bf16T, fp32T, fp32T, fp16T},
1018 SpecificationVersion::V_1_1_DRAFT},
1019 {{bs32_fp8ue8m0_fp8e4m3T, fp16T, fp16T, fp32T, fp16T, bf16T, fp16T},
1020 SpecificationVersion::V_1_1_DRAFT},
1021 {{bs32_fp8ue8m0_fp8e4m3T, bf16T, fp16T, fp32T, bf16T, bf16T, fp16T},
1022 SpecificationVersion::V_1_1_DRAFT},
1023 {{bs32_fp8ue8m0_fp8e4m3T, bf16T, fp16T, fp32T, bf16T, fp32T, fp16T},
1024 SpecificationVersion::V_1_1_DRAFT},
1025 {{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp32T, fp32T,
1026 bf16T, fp16T},
1027 SpecificationVersion::V_1_1_DRAFT},
1028 {{bf16T, bs32_fp8ue8m0_fp8e4m3T, fp32T, bf16T, fp32T, fp32T, fp32T},
1029 SpecificationVersion::V_1_1_DRAFT},
1030 {{bs32_fp8ue8m0_fp8e4m3T, bf16T, fp32T, fp32T, bf16T, fp32T, fp32T},
1031 SpecificationVersion::V_1_1_DRAFT}},
1032 allOf},
1033 {{Extension::mx_common, Extension::mx_fp8e4m3},
1034 {{{fp16T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp16T, fp32T, fp32T, fp16T},
1035 SpecificationVersion::V_1_1_DRAFT},
1036 {{bs32_fp8ue8m0_fp8e4m3T, fp16T, fp16T, fp32T, fp16T, fp32T, fp16T},
1037 SpecificationVersion::V_1_1_DRAFT},
1038 {{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp32T, fp32T,
1039 fp32T, fp16T},
1040 SpecificationVersion::V_1_1_DRAFT},
1041 {{fp16T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp16T, fp32T, fp32T, fp32T},
1042 SpecificationVersion::V_1_1_DRAFT},
1043 {{fp32T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp32T, fp32T, fp32T},
1044 SpecificationVersion::V_1_1_DRAFT},
1045 {{bs32_fp8ue8m0_fp8e4m3T, fp16T, fp32T, fp32T, fp16T, fp32T, fp32T},
1046 SpecificationVersion::V_1_1_DRAFT},
1047 {{bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp32T, fp32T, fp32T, fp32T},
1048 SpecificationVersion::V_1_1_DRAFT},
1049 {{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp32T,
1050 fp32T, fp32T},
1051 SpecificationVersion::V_1_1_DRAFT}},
1052 allOf},
1053 {{Extension::bf16, Extension::mx_common, Extension::mx_fp8e5m2},
1054 {{{fp16T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp16T, fp32T, bf16T, fp16T},
1055 SpecificationVersion::V_1_1_DRAFT},
1056 {{bf16T, bs32_fp8ue8m0_fp8e5m2T, fp16T, bf16T, fp32T, bf16T, fp16T},
1057 SpecificationVersion::V_1_1_DRAFT},
1058 {{bf16T, bs32_fp8ue8m0_fp8e5m2T, fp16T, bf16T, fp32T, fp32T, fp16T},
1059 SpecificationVersion::V_1_1_DRAFT},
1060 {{bs32_fp8ue8m0_fp8e5m2T, fp16T, fp16T, fp32T, fp16T, bf16T, fp16T},
1061 SpecificationVersion::V_1_1_DRAFT},
1062 {{bs32_fp8ue8m0_fp8e5m2T, bf16T, fp16T, fp32T, bf16T, bf16T, fp16T},
1063 SpecificationVersion::V_1_1_DRAFT},
1064 {{bs32_fp8ue8m0_fp8e5m2T, bf16T, fp16T, fp32T, bf16T, fp32T, fp16T},
1065 SpecificationVersion::V_1_1_DRAFT},
1066 {{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp32T, fp32T,
1067 bf16T, fp16T},
1068 SpecificationVersion::V_1_1_DRAFT},
1069 {{bf16T, bs32_fp8ue8m0_fp8e5m2T, fp32T, bf16T, fp32T, fp32T, fp32T},
1070 SpecificationVersion::V_1_1_DRAFT},
1071 {{bs32_fp8ue8m0_fp8e5m2T, bf16T, fp32T, fp32T, bf16T, fp32T, fp32T},
1072 SpecificationVersion::V_1_1_DRAFT}},
1073 allOf},
1074 {{Extension::mx_common, Extension::mx_fp8e5m2},
1075 {{{fp16T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp16T, fp32T, fp32T, fp16T},
1076 SpecificationVersion::V_1_1_DRAFT},
1077 {{bs32_fp8ue8m0_fp8e5m2T, fp16T, fp16T, fp32T, fp16T, fp32T, fp16T},
1078 SpecificationVersion::V_1_1_DRAFT},
1079 {{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp32T, fp32T,
1080 fp32T, fp16T},
1081 SpecificationVersion::V_1_1_DRAFT},
1082 {{fp16T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp16T, fp32T, fp32T, fp32T},
1083 SpecificationVersion::V_1_1_DRAFT},
1084 {{fp32T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp32T, fp32T, fp32T},
1085 SpecificationVersion::V_1_1_DRAFT},
1086 {{bs32_fp8ue8m0_fp8e5m2T, fp16T, fp32T, fp32T, fp16T, fp32T, fp32T},
1087 SpecificationVersion::V_1_1_DRAFT},
1088 {{bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp32T, fp32T, fp32T, fp32T},
1089 SpecificationVersion::V_1_1_DRAFT},
1090 {{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp32T,
1091 fp32T, fp32T},
1092 SpecificationVersion::V_1_1_DRAFT}},
1093 allOf},
1094 {{Extension::bf16, Extension::mx_common, Extension::mx_fp6e3m2},
1095 {{{fp16T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp16T, fp32T, bf16T, fp16T},
1096 SpecificationVersion::V_1_1_DRAFT},
1097 {{bf16T, bs32_fp8ue8m0_fp6e3m2T, fp16T, bf16T, fp32T, bf16T, fp16T},
1098 SpecificationVersion::V_1_1_DRAFT},
1099 {{bf16T, bs32_fp8ue8m0_fp6e3m2T, fp16T, bf16T, fp32T, fp32T, fp16T},
1100 SpecificationVersion::V_1_1_DRAFT},
1101 {{bs32_fp8ue8m0_fp6e3m2T, fp16T, fp16T, fp32T, fp16T, bf16T, fp16T},
1102 SpecificationVersion::V_1_1_DRAFT},
1103 {{bs32_fp8ue8m0_fp6e3m2T, bf16T, fp16T, fp32T, bf16T, bf16T, fp16T},
1104 SpecificationVersion::V_1_1_DRAFT},
1105 {{bs32_fp8ue8m0_fp6e3m2T, bf16T, fp16T, fp32T, bf16T, fp32T, fp16T},
1106 SpecificationVersion::V_1_1_DRAFT},
1107 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp32T, fp32T,
1108 bf16T, fp16T},
1109 SpecificationVersion::V_1_1_DRAFT},
1110 {{bf16T, bs32_fp8ue8m0_fp6e3m2T, fp32T, bf16T, fp32T, fp32T, fp32T},
1111 SpecificationVersion::V_1_1_DRAFT},
1112 {{bs32_fp8ue8m0_fp6e3m2T, bf16T, fp32T, fp32T, bf16T, fp32T, fp32T},
1113 SpecificationVersion::V_1_1_DRAFT}},
1114 allOf},
1115 {{Extension::mx_common, Extension::mx_fp6e3m2},
1116 {{{fp16T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp16T, fp32T, fp32T, fp16T},
1117 SpecificationVersion::V_1_1_DRAFT},
1118 {{bs32_fp8ue8m0_fp6e3m2T, fp16T, fp16T, fp32T, fp16T, fp32T, fp16T},
1119 SpecificationVersion::V_1_1_DRAFT},
1120 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp32T, fp32T,
1121 fp32T, fp16T},
1122 SpecificationVersion::V_1_1_DRAFT},
1123 {{fp16T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp16T, fp32T, fp32T, fp32T},
1124 SpecificationVersion::V_1_1_DRAFT},
1125 {{fp32T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp32T, fp32T, fp32T},
1126 SpecificationVersion::V_1_1_DRAFT},
1127 {{bs32_fp8ue8m0_fp6e3m2T, fp16T, fp32T, fp32T, fp16T, fp32T, fp32T},
1128 SpecificationVersion::V_1_1_DRAFT},
1129 {{bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp32T, fp32T, fp32T, fp32T},
1130 SpecificationVersion::V_1_1_DRAFT},
1131 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp32T,
1132 fp32T, fp32T},
1133 SpecificationVersion::V_1_1_DRAFT}},
1134 allOf},
1135 {{Extension::bf16, Extension::mx_common, Extension::mx_fp6e2m3},
1136 {{{fp16T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp16T, fp32T, bf16T, fp16T},
1137 SpecificationVersion::V_1_1_DRAFT},
1138 {{bf16T, bs32_fp8ue8m0_fp6e2m3T, fp16T, bf16T, fp32T, bf16T, fp16T},
1139 SpecificationVersion::V_1_1_DRAFT},
1140 {{bf16T, bs32_fp8ue8m0_fp6e2m3T, fp16T, bf16T, fp32T, fp32T, fp16T},
1141 SpecificationVersion::V_1_1_DRAFT},
1142 {{bs32_fp8ue8m0_fp6e2m3T, fp16T, fp16T, fp32T, fp16T, bf16T, fp16T},
1143 SpecificationVersion::V_1_1_DRAFT},
1144 {{bs32_fp8ue8m0_fp6e2m3T, bf16T, fp16T, fp32T, bf16T, bf16T, fp16T},
1145 SpecificationVersion::V_1_1_DRAFT},
1146 {{bs32_fp8ue8m0_fp6e2m3T, bf16T, fp16T, fp32T, bf16T, fp32T, fp16T},
1147 SpecificationVersion::V_1_1_DRAFT},
1148 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp32T, fp32T,
1149 bf16T, fp16T},
1150 SpecificationVersion::V_1_1_DRAFT},
1151 {{bf16T, bs32_fp8ue8m0_fp6e2m3T, fp32T, bf16T, fp32T, fp32T, fp32T},
1152 SpecificationVersion::V_1_1_DRAFT},
1153 {{bs32_fp8ue8m0_fp6e2m3T, bf16T, fp32T, fp32T, bf16T, fp32T, fp32T},
1154 SpecificationVersion::V_1_1_DRAFT}},
1155 allOf},
1156 {{Extension::mx_common, Extension::mx_fp6e2m3},
1157 {{{fp16T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp16T, fp32T, fp32T, fp16T},
1158 SpecificationVersion::V_1_1_DRAFT},
1159 {{bs32_fp8ue8m0_fp6e2m3T, fp16T, fp16T, fp32T, fp16T, fp32T, fp16T},
1160 SpecificationVersion::V_1_1_DRAFT},
1161 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp32T, fp32T,
1162 fp32T, fp16T},
1163 SpecificationVersion::V_1_1_DRAFT},
1164 {{fp16T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp16T, fp32T, fp32T, fp32T},
1165 SpecificationVersion::V_1_1_DRAFT},
1166 {{fp32T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp32T, fp32T, fp32T},
1167 SpecificationVersion::V_1_1_DRAFT},
1168 {{bs32_fp8ue8m0_fp6e2m3T, fp16T, fp32T, fp32T, fp16T, fp32T, fp32T},
1169 SpecificationVersion::V_1_1_DRAFT},
1170 {{bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp32T, fp32T, fp32T, fp32T},
1171 SpecificationVersion::V_1_1_DRAFT},
1172 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp32T,
1173 fp32T, fp32T},
1174 SpecificationVersion::V_1_1_DRAFT}},
1175 allOf},
1176 {{Extension::bf16, Extension::mx_common, Extension::mx_fp4e2m1},
1177 {{{fp16T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp16T, fp32T, bf16T, fp16T},
1178 SpecificationVersion::V_1_1_DRAFT},
1179 {{bf16T, bs32_fp8ue8m0_fp4e2m1T, fp16T, bf16T, fp32T, bf16T, fp16T},
1180 SpecificationVersion::V_1_1_DRAFT},
1181 {{bf16T, bs32_fp8ue8m0_fp4e2m1T, fp16T, bf16T, fp32T, fp32T, fp16T},
1182 SpecificationVersion::V_1_1_DRAFT},
1183 {{bs32_fp8ue8m0_fp4e2m1T, fp16T, fp16T, fp32T, fp16T, bf16T, fp16T},
1184 SpecificationVersion::V_1_1_DRAFT},
1185 {{bs32_fp8ue8m0_fp4e2m1T, bf16T, fp16T, fp32T, bf16T, bf16T, fp16T},
1186 SpecificationVersion::V_1_1_DRAFT},
1187 {{bs32_fp8ue8m0_fp4e2m1T, bf16T, fp16T, fp32T, bf16T, fp32T, fp16T},
1188 SpecificationVersion::V_1_1_DRAFT},
1189 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp32T, fp32T,
1190 bf16T, fp16T},
1191 SpecificationVersion::V_1_1_DRAFT},
1192 {{bf16T, bs32_fp8ue8m0_fp4e2m1T, fp32T, bf16T, fp32T, fp32T, fp32T},
1193 SpecificationVersion::V_1_1_DRAFT},
1194 {{bs32_fp8ue8m0_fp4e2m1T, bf16T, fp32T, fp32T, bf16T, fp32T, fp32T},
1195 SpecificationVersion::V_1_1_DRAFT}},
1196 allOf},
1197 {{Extension::mx_common, Extension::mx_fp4e2m1},
1198 {{{fp16T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp16T, fp32T, fp32T, fp16T},
1199 SpecificationVersion::V_1_1_DRAFT},
1200 {{bs32_fp8ue8m0_fp4e2m1T, fp16T, fp16T, fp32T, fp16T, fp32T, fp16T},
1201 SpecificationVersion::V_1_1_DRAFT},
1202 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp32T, fp32T,
1203 fp32T, fp16T},
1204 SpecificationVersion::V_1_1_DRAFT},
1205 {{fp16T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp16T, fp32T, fp32T, fp32T},
1206 SpecificationVersion::V_1_1_DRAFT},
1207 {{fp32T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp32T, fp32T, fp32T},
1208 SpecificationVersion::V_1_1_DRAFT},
1209 {{bs32_fp8ue8m0_fp4e2m1T, fp16T, fp32T, fp32T, fp16T, fp32T, fp32T},
1210 SpecificationVersion::V_1_1_DRAFT},
1211 {{bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp32T, fp32T, fp32T, fp32T},
1212 SpecificationVersion::V_1_1_DRAFT},
1213 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp32T,
1214 fp32T, fp32T},
1215 SpecificationVersion::V_1_1_DRAFT}},
1216 allOf},
1217 {{Extension::bf16, Extension::mx_common, Extension::mx_int8},
1218 {{{fp16T, bs32_fp8ue8m0_mxint8T, fp16T, fp16T, fp32T, bf16T, fp16T},
1219 SpecificationVersion::V_1_1_DRAFT},
1220 {{bf16T, bs32_fp8ue8m0_mxint8T, fp16T, bf16T, fp32T, bf16T, fp16T},
1221 SpecificationVersion::V_1_1_DRAFT},
1222 {{bf16T, bs32_fp8ue8m0_mxint8T, fp16T, bf16T, fp32T, fp32T, fp16T},
1223 SpecificationVersion::V_1_1_DRAFT},
1224 {{bs32_fp8ue8m0_mxint8T, fp16T, fp16T, fp32T, fp16T, bf16T, fp16T},
1225 SpecificationVersion::V_1_1_DRAFT},
1226 {{bs32_fp8ue8m0_mxint8T, bf16T, fp16T, fp32T, bf16T, bf16T, fp16T},
1227 SpecificationVersion::V_1_1_DRAFT},
1228 {{bs32_fp8ue8m0_mxint8T, bf16T, fp16T, fp32T, bf16T, fp32T, fp16T},
1229 SpecificationVersion::V_1_1_DRAFT},
1230 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_mxint8T, fp16T, fp32T, fp32T,
1231 bf16T, fp16T},
1232 SpecificationVersion::V_1_1_DRAFT},
1233 {{bf16T, bs32_fp8ue8m0_mxint8T, fp32T, bf16T, fp32T, fp32T, fp32T},
1234 SpecificationVersion::V_1_1_DRAFT},
1235 {{bs32_fp8ue8m0_mxint8T, bf16T, fp32T, fp32T, bf16T, fp32T, fp32T},
1236 SpecificationVersion::V_1_1_DRAFT}},
1237 allOf},
1238 {{Extension::mx_common, Extension::mx_int8},
1239 {{{fp16T, bs32_fp8ue8m0_mxint8T, fp16T, fp16T, fp32T, fp32T, fp16T},
1240 SpecificationVersion::V_1_1_DRAFT},
1241 {{bs32_fp8ue8m0_mxint8T, fp16T, fp16T, fp32T, fp16T, fp32T, fp16T},
1242 SpecificationVersion::V_1_1_DRAFT},
1243 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_mxint8T, fp16T, fp32T, fp32T,
1244 fp32T, fp16T},
1245 SpecificationVersion::V_1_1_DRAFT},
1246 {{fp16T, bs32_fp8ue8m0_mxint8T, fp32T, fp16T, fp32T, fp32T, fp32T},
1247 SpecificationVersion::V_1_1_DRAFT},
1248 {{fp32T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp32T, fp32T, fp32T},
1249 SpecificationVersion::V_1_1_DRAFT},
1250 {{bs32_fp8ue8m0_mxint8T, fp16T, fp32T, fp32T, fp16T, fp32T, fp32T},
1251 SpecificationVersion::V_1_1_DRAFT},
1252 {{bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp32T, fp32T, fp32T, fp32T},
1253 SpecificationVersion::V_1_1_DRAFT},
1254 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp32T,
1255 fp32T, fp32T},
1256 SpecificationVersion::V_1_1_DRAFT}},
1257 allOf},
1258 {{Extension::bf16, Extension::mx_common, Extension::mx_fp8e4m3,
1259 Extension::mx_fp8e5m2},
1260 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp32T, fp32T,
1261 bf16T, fp16T},
1262 SpecificationVersion::V_1_1_DRAFT},
1263 {{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp32T, fp32T,
1264 bf16T, fp16T},
1265 SpecificationVersion::V_1_1_DRAFT}},
1266 allOf},
1267 {{Extension::mx_common, Extension::mx_fp8e4m3, Extension::mx_fp8e5m2},
1268 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp32T, fp32T,
1269 fp32T, fp16T},
1270 SpecificationVersion::V_1_1_DRAFT},
1271 {{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp32T, fp32T,
1272 fp32T, fp16T},
1273 SpecificationVersion::V_1_1_DRAFT},
1274 {{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp32T,
1275 fp32T, fp32T},
1276 SpecificationVersion::V_1_1_DRAFT},
1277 {{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp32T,
1278 fp32T, fp32T},
1279 SpecificationVersion::V_1_1_DRAFT}},
1280 allOf},
1281 {{Extension::bf16, Extension::mx_common, Extension::mx_fp6e3m2,
1282 Extension::mx_fp8e4m3},
1283 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp32T, fp32T,
1284 bf16T, fp16T},
1285 SpecificationVersion::V_1_1_DRAFT},
1286 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp32T, fp32T,
1287 bf16T, fp16T},
1288 SpecificationVersion::V_1_1_DRAFT}},
1289 allOf},
1290 {{Extension::mx_common, Extension::mx_fp6e3m2, Extension::mx_fp8e4m3},
1291 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp32T, fp32T,
1292 fp32T, fp16T},
1293 SpecificationVersion::V_1_1_DRAFT},
1294 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp32T, fp32T,
1295 fp32T, fp16T},
1296 SpecificationVersion::V_1_1_DRAFT},
1297 {{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp32T,
1298 fp32T, fp32T},
1299 SpecificationVersion::V_1_1_DRAFT},
1300 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp32T,
1301 fp32T, fp32T},
1302 SpecificationVersion::V_1_1_DRAFT}},
1303 allOf},
1304 {{Extension::bf16, Extension::mx_common, Extension::mx_fp6e2m3,
1305 Extension::mx_fp8e4m3},
1306 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp32T, fp32T,
1307 bf16T, fp16T},
1308 SpecificationVersion::V_1_1_DRAFT},
1309 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp32T, fp32T,
1310 bf16T, fp16T},
1311 SpecificationVersion::V_1_1_DRAFT}},
1312 allOf},
1313 {{Extension::mx_common, Extension::mx_fp6e2m3, Extension::mx_fp8e4m3},
1314 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp32T, fp32T,
1315 fp32T, fp16T},
1316 SpecificationVersion::V_1_1_DRAFT},
1317 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp32T, fp32T,
1318 fp32T, fp16T},
1319 SpecificationVersion::V_1_1_DRAFT},
1320 {{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp32T,
1321 fp32T, fp32T},
1322 SpecificationVersion::V_1_1_DRAFT},
1323 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp32T,
1324 fp32T, fp32T},
1325 SpecificationVersion::V_1_1_DRAFT}},
1326 allOf},
1327 {{Extension::bf16, Extension::mx_common, Extension::mx_fp4e2m1,
1328 Extension::mx_fp8e4m3},
1329 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp32T, fp32T,
1330 bf16T, fp16T},
1331 SpecificationVersion::V_1_1_DRAFT},
1332 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp32T, fp32T,
1333 bf16T, fp16T},
1334 SpecificationVersion::V_1_1_DRAFT}},
1335 allOf},
1336 {{Extension::mx_common, Extension::mx_fp4e2m1, Extension::mx_fp8e4m3},
1337 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp32T, fp32T,
1338 fp32T, fp16T},
1339 SpecificationVersion::V_1_1_DRAFT},
1340 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp32T, fp32T,
1341 fp32T, fp16T},
1342 SpecificationVersion::V_1_1_DRAFT},
1343 {{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp32T,
1344 fp32T, fp32T},
1345 SpecificationVersion::V_1_1_DRAFT},
1346 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp32T,
1347 fp32T, fp32T},
1348 SpecificationVersion::V_1_1_DRAFT}},
1349 allOf},
1350 {{Extension::bf16, Extension::mx_common, Extension::mx_fp8e4m3,
1351 Extension::mx_int8},
1352 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_mxint8T, fp16T, fp32T, fp32T,
1353 bf16T, fp16T},
1354 SpecificationVersion::V_1_1_DRAFT},
1355 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp32T, fp32T,
1356 bf16T, fp16T},
1357 SpecificationVersion::V_1_1_DRAFT}},
1358 allOf},
1359 {{Extension::mx_common, Extension::mx_fp8e4m3, Extension::mx_int8},
1360 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_mxint8T, fp16T, fp32T, fp32T,
1361 fp32T, fp16T},
1362 SpecificationVersion::V_1_1_DRAFT},
1363 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp32T, fp32T,
1364 fp32T, fp16T},
1365 SpecificationVersion::V_1_1_DRAFT},
1366 {{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp32T,
1367 fp32T, fp32T},
1368 SpecificationVersion::V_1_1_DRAFT},
1369 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp32T,
1370 fp32T, fp32T},
1371 SpecificationVersion::V_1_1_DRAFT}},
1372 allOf},
1373 {{Extension::bf16, Extension::mx_common, Extension::mx_fp6e3m2,
1374 Extension::mx_fp8e5m2},
1375 {{{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp32T, fp32T,
1376 bf16T, fp16T},
1377 SpecificationVersion::V_1_1_DRAFT},
1378 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp32T, fp32T,
1379 bf16T, fp16T},
1380 SpecificationVersion::V_1_1_DRAFT}},
1381 allOf},
1382 {{Extension::mx_common, Extension::mx_fp6e3m2, Extension::mx_fp8e5m2},
1383 {{{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp32T, fp32T,
1384 fp32T, fp16T},
1385 SpecificationVersion::V_1_1_DRAFT},
1386 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp32T, fp32T,
1387 fp32T, fp16T},
1388 SpecificationVersion::V_1_1_DRAFT},
1389 {{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp32T,
1390 fp32T, fp32T},
1391 SpecificationVersion::V_1_1_DRAFT},
1392 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp32T,
1393 fp32T, fp32T},
1394 SpecificationVersion::V_1_1_DRAFT}},
1395 allOf},
1396 {{Extension::bf16, Extension::mx_common, Extension::mx_fp6e2m3,
1397 Extension::mx_fp8e5m2},
1398 {{{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp32T, fp32T,
1399 bf16T, fp16T},
1400 SpecificationVersion::V_1_1_DRAFT},
1401 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp32T, fp32T,
1402 bf16T, fp16T},
1403 SpecificationVersion::V_1_1_DRAFT}},
1404 allOf},
1405 {{Extension::mx_common, Extension::mx_fp6e2m3, Extension::mx_fp8e5m2},
1406 {{{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp32T, fp32T,
1407 fp32T, fp16T},
1408 SpecificationVersion::V_1_1_DRAFT},
1409 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp32T, fp32T,
1410 fp32T, fp16T},
1411 SpecificationVersion::V_1_1_DRAFT},
1412 {{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp32T,
1413 fp32T, fp32T},
1414 SpecificationVersion::V_1_1_DRAFT},
1415 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp32T,
1416 fp32T, fp32T},
1417 SpecificationVersion::V_1_1_DRAFT}},
1418 allOf},
1419 {{Extension::bf16, Extension::mx_common, Extension::mx_fp4e2m1,
1420 Extension::mx_fp8e5m2},
1421 {{{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp32T, fp32T,
1422 bf16T, fp16T},
1423 SpecificationVersion::V_1_1_DRAFT},
1424 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp32T, fp32T,
1425 bf16T, fp16T},
1426 SpecificationVersion::V_1_1_DRAFT}},
1427 allOf},
1428 {{Extension::mx_common, Extension::mx_fp4e2m1, Extension::mx_fp8e5m2},
1429 {{{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp32T, fp32T,
1430 fp32T, fp16T},
1431 SpecificationVersion::V_1_1_DRAFT},
1432 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp32T, fp32T,
1433 fp32T, fp16T},
1434 SpecificationVersion::V_1_1_DRAFT},
1435 {{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp32T,
1436 fp32T, fp32T},
1437 SpecificationVersion::V_1_1_DRAFT},
1438 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp32T,
1439 fp32T, fp32T},
1440 SpecificationVersion::V_1_1_DRAFT}},
1441 allOf},
1442 {{Extension::bf16, Extension::mx_common, Extension::mx_fp8e5m2,
1443 Extension::mx_int8},
1444 {{{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_mxint8T, fp16T, fp32T, fp32T,
1445 bf16T, fp16T},
1446 SpecificationVersion::V_1_1_DRAFT},
1447 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp32T, fp32T,
1448 bf16T, fp16T},
1449 SpecificationVersion::V_1_1_DRAFT}},
1450 allOf},
1451 {{Extension::mx_common, Extension::mx_fp8e5m2, Extension::mx_int8},
1452 {{{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_mxint8T, fp16T, fp32T, fp32T,
1453 fp32T, fp16T},
1454 SpecificationVersion::V_1_1_DRAFT},
1455 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp32T, fp32T,
1456 fp32T, fp16T},
1457 SpecificationVersion::V_1_1_DRAFT},
1458 {{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp32T,
1459 fp32T, fp32T},
1460 SpecificationVersion::V_1_1_DRAFT},
1461 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp32T,
1462 fp32T, fp32T},
1463 SpecificationVersion::V_1_1_DRAFT}},
1464 allOf},
1465 {{Extension::bf16, Extension::mx_common, Extension::mx_fp6e2m3,
1466 Extension::mx_fp6e3m2},
1467 {{{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp32T, fp32T,
1468 bf16T, fp16T},
1469 SpecificationVersion::V_1_1_DRAFT},
1470 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp32T, fp32T,
1471 bf16T, fp16T},
1472 SpecificationVersion::V_1_1_DRAFT}},
1473 allOf},
1474 {{Extension::mx_common, Extension::mx_fp6e2m3, Extension::mx_fp6e3m2},
1475 {{{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp32T, fp32T,
1476 fp32T, fp16T},
1477 SpecificationVersion::V_1_1_DRAFT},
1478 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp32T, fp32T,
1479 fp32T, fp16T},
1480 SpecificationVersion::V_1_1_DRAFT},
1481 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp32T,
1482 fp32T, fp32T},
1483 SpecificationVersion::V_1_1_DRAFT},
1484 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp32T,
1485 fp32T, fp32T},
1486 SpecificationVersion::V_1_1_DRAFT}},
1487 allOf},
1488 {{Extension::bf16, Extension::mx_common, Extension::mx_fp4e2m1,
1489 Extension::mx_fp6e3m2},
1490 {{{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp32T, fp32T,
1491 bf16T, fp16T},
1492 SpecificationVersion::V_1_1_DRAFT},
1493 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp32T, fp32T,
1494 bf16T, fp16T},
1495 SpecificationVersion::V_1_1_DRAFT}},
1496 allOf},
1497 {{Extension::mx_common, Extension::mx_fp4e2m1, Extension::mx_fp6e3m2},
1498 {{{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp32T, fp32T,
1499 fp32T, fp16T},
1500 SpecificationVersion::V_1_1_DRAFT},
1501 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp32T, fp32T,
1502 fp32T, fp16T},
1503 SpecificationVersion::V_1_1_DRAFT},
1504 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp32T,
1505 fp32T, fp32T},
1506 SpecificationVersion::V_1_1_DRAFT},
1507 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp32T,
1508 fp32T, fp32T},
1509 SpecificationVersion::V_1_1_DRAFT}},
1510 allOf},
1511 {{Extension::bf16, Extension::mx_common, Extension::mx_fp6e3m2,
1512 Extension::mx_int8},
1513 {{{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_mxint8T, fp16T, fp32T, fp32T,
1514 bf16T, fp16T},
1515 SpecificationVersion::V_1_1_DRAFT},
1516 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp32T, fp32T,
1517 bf16T, fp16T},
1518 SpecificationVersion::V_1_1_DRAFT}},
1519 allOf},
1520 {{Extension::mx_common, Extension::mx_fp6e3m2, Extension::mx_int8},
1521 {{{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_mxint8T, fp16T, fp32T, fp32T,
1522 fp32T, fp16T},
1523 SpecificationVersion::V_1_1_DRAFT},
1524 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp32T, fp32T,
1525 fp32T, fp16T},
1526 SpecificationVersion::V_1_1_DRAFT},
1527 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp32T,
1528 fp32T, fp32T},
1529 SpecificationVersion::V_1_1_DRAFT},
1530 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp32T,
1531 fp32T, fp32T},
1532 SpecificationVersion::V_1_1_DRAFT}},
1533 allOf},
1534 {{Extension::bf16, Extension::mx_common, Extension::mx_fp4e2m1,
1535 Extension::mx_fp6e2m3},
1536 {{{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp32T, fp32T,
1537 bf16T, fp16T},
1538 SpecificationVersion::V_1_1_DRAFT},
1539 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp32T, fp32T,
1540 bf16T, fp16T},
1541 SpecificationVersion::V_1_1_DRAFT}},
1542 allOf},
1543 {{Extension::mx_common, Extension::mx_fp4e2m1, Extension::mx_fp6e2m3},
1544 {{{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp32T, fp32T,
1545 fp32T, fp16T},
1546 SpecificationVersion::V_1_1_DRAFT},
1547 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp32T, fp32T,
1548 fp32T, fp16T},
1549 SpecificationVersion::V_1_1_DRAFT},
1550 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp32T,
1551 fp32T, fp32T},
1552 SpecificationVersion::V_1_1_DRAFT},
1553 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp32T,
1554 fp32T, fp32T},
1555 SpecificationVersion::V_1_1_DRAFT}},
1556 allOf},
1557 {{Extension::bf16, Extension::mx_common, Extension::mx_fp6e2m3,
1558 Extension::mx_int8},
1559 {{{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_mxint8T, fp16T, fp32T, fp32T,
1560 bf16T, fp16T},
1561 SpecificationVersion::V_1_1_DRAFT},
1562 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp32T, fp32T,
1563 bf16T, fp16T},
1564 SpecificationVersion::V_1_1_DRAFT}},
1565 allOf},
1566 {{Extension::mx_common, Extension::mx_fp6e2m3, Extension::mx_int8},
1567 {{{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_mxint8T, fp16T, fp32T, fp32T,
1568 fp32T, fp16T},
1569 SpecificationVersion::V_1_1_DRAFT},
1570 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp32T, fp32T,
1571 fp32T, fp16T},
1572 SpecificationVersion::V_1_1_DRAFT},
1573 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp32T,
1574 fp32T, fp32T},
1575 SpecificationVersion::V_1_1_DRAFT},
1576 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp32T,
1577 fp32T, fp32T},
1578 SpecificationVersion::V_1_1_DRAFT}},
1579 allOf},
1580 {{Extension::bf16, Extension::mx_common, Extension::mx_fp4e2m1,
1581 Extension::mx_int8},
1582 {{{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_mxint8T, fp16T, fp32T, fp32T,
1583 bf16T, fp16T},
1584 SpecificationVersion::V_1_1_DRAFT},
1585 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp32T, fp32T,
1586 bf16T, fp16T},
1587 SpecificationVersion::V_1_1_DRAFT}},
1588 allOf},
1589 {{Extension::mx_common, Extension::mx_fp4e2m1, Extension::mx_int8},
1590 {{{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_mxint8T, fp16T, fp32T, fp32T,
1591 fp32T, fp16T},
1592 SpecificationVersion::V_1_1_DRAFT},
1593 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp32T, fp32T,
1594 fp32T, fp16T},
1595 SpecificationVersion::V_1_1_DRAFT},
1596 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp32T,
1597 fp32T, fp32T},
1598 SpecificationVersion::V_1_1_DRAFT},
1599 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp32T,
1600 fp32T, fp32T},
1601 SpecificationVersion::V_1_1_DRAFT}},
1602 allOf}}},
1603 {"tosa.conv2d_block_scaled",
1604 {{{Extension::mxfp_conv},
1605 {{{fp4e2m1T, fp8ue8m0T, fp4e2m1T, fp8ue8m0T, fp32T, fp32T},
1606 SpecificationVersion::V_1_1_DRAFT},
1607 {{fp6e2m3T, fp8ue8m0T, fp6e2m3T, fp8ue8m0T, fp32T, fp32T},
1608 SpecificationVersion::V_1_1_DRAFT},
1609 {{fp6e3m2T, fp8ue8m0T, fp6e3m2T, fp8ue8m0T, fp32T, fp32T},
1610 SpecificationVersion::V_1_1_DRAFT},
1611 {{fp8e4m3T, fp8ue8m0T, fp8e4m3T, fp8ue8m0T, fp32T, fp32T},
1612 SpecificationVersion::V_1_1_DRAFT},
1613 {{fp8e5m2T, fp8ue8m0T, fp8e5m2T, fp8ue8m0T, fp32T, fp32T},
1614 SpecificationVersion::V_1_1_DRAFT}}}}},
1615 {"tosa.conv3d",
1616 {{{Extension::int4},
1617 {{{i8T, i4T, i32T, i8T, i4T, i32T, i32T}, SpecificationVersion::V_1_0}}},
1618 {{Extension::int16},
1619 {{{i16T, i8T, i48T, i16T, i8T, i48T, i48T},
1620 SpecificationVersion::V_1_0}}},
1621 {{Extension::fp8e4m3},
1622 {{{fp8e4m3T, fp8e4m3T, fp16T, fp8e4m3T, fp8e4m3T, fp16T, fp16T},
1623 SpecificationVersion::V_1_0},
1624 {{fp8e4m3T, fp8e4m3T, fp32T, fp8e4m3T, fp8e4m3T, fp32T, fp32T},
1625 SpecificationVersion::V_1_1_DRAFT}}},
1626 {{Extension::fp8e5m2},
1627 {{{fp8e5m2T, fp8e5m2T, fp16T, fp8e5m2T, fp8e5m2T, fp16T, fp16T},
1628 SpecificationVersion::V_1_0},
1629 {{fp8e5m2T, fp8e5m2T, fp32T, fp8e5m2T, fp8e5m2T, fp32T, fp32T},
1630 SpecificationVersion::V_1_1_DRAFT}}},
1631 {{Extension::bf16},
1632 {{{bf16T, bf16T, bf16T, bf16T, bf16T, fp32T, bf16T},
1633 SpecificationVersion::V_1_0}}}}},
1634 {"tosa.depthwise_conv2d",
1635 {{{Extension::int4},
1636 {{{i8T, i4T, i32T, i8T, i4T, i32T, i32T}, SpecificationVersion::V_1_0}}},
1637 {{Extension::int16},
1638 {{{i16T, i8T, i48T, i16T, i8T, i48T, i48T},
1639 SpecificationVersion::V_1_0}}},
1640 {{Extension::fp8e4m3},
1641 {{{fp8e4m3T, fp8e4m3T, fp16T, fp8e4m3T, fp8e4m3T, fp16T, fp16T},
1642 SpecificationVersion::V_1_0},
1643 {{fp8e4m3T, fp8e4m3T, fp32T, fp8e4m3T, fp8e4m3T, fp32T, fp32T},
1644 SpecificationVersion::V_1_1_DRAFT}}},
1645 {{Extension::fp8e5m2},
1646 {{{fp8e5m2T, fp8e5m2T, fp16T, fp8e5m2T, fp8e5m2T, fp16T, fp16T},
1647 SpecificationVersion::V_1_0},
1648 {{fp8e5m2T, fp8e5m2T, fp32T, fp8e5m2T, fp8e5m2T, fp32T, fp32T},
1649 SpecificationVersion::V_1_1_DRAFT}}},
1650 {{Extension::bf16},
1651 {{{bf16T, bf16T, bf16T, bf16T, bf16T, fp32T, bf16T},
1652 SpecificationVersion::V_1_0}}}}},
1653 {"tosa.fft2d",
1654 {{{Extension::fft},
1655 {{{fp32T, fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
1656 {"tosa.matmul",
1657 {{{Extension::int16},
1658 {{{i16T, i16T, i16T, i16T, i48T}, SpecificationVersion::V_1_0}}},
1659 {{Extension::fp8e4m3},
1660 {{{fp8e4m3T, fp8e4m3T, fp8e4m3T, fp8e4m3T, fp16T},
1661 SpecificationVersion::V_1_0},
1662 {{fp8e4m3T, fp8e4m3T, fp8e4m3T, fp8e4m3T, fp32T},
1663 SpecificationVersion::V_1_1_DRAFT}}},
1664 {{Extension::fp8e5m2},
1665 {{{fp8e5m2T, fp8e5m2T, fp8e5m2T, fp8e5m2T, fp16T},
1666 SpecificationVersion::V_1_0},
1667 {{fp8e5m2T, fp8e5m2T, fp8e5m2T, fp8e5m2T, fp32T},
1668 SpecificationVersion::V_1_1_DRAFT}}},
1669 {{Extension::fp8e4m3, Extension::fp8e5m2},
1670 {{{fp8e4m3T, fp8e5m2T, fp8e4m3T, fp8e5m2T, fp16T},
1671 SpecificationVersion::V_1_1_DRAFT},
1672 {{fp8e4m3T, fp8e5m2T, fp8e4m3T, fp8e5m2T, fp32T},
1673 SpecificationVersion::V_1_1_DRAFT},
1674 {{fp8e5m2T, fp8e4m3T, fp8e5m2T, fp8e4m3T, fp16T},
1675 SpecificationVersion::V_1_1_DRAFT},
1676 {{fp8e5m2T, fp8e4m3T, fp8e5m2T, fp8e4m3T, fp32T},
1677 SpecificationVersion::V_1_1_DRAFT}},
1678 allOf},
1679 {{Extension::bf16},
1680 {{{bf16T, bf16T, bf16T, bf16T, fp32T}, SpecificationVersion::V_1_0}}}}},
1681 {"tosa.matmul_t",
1682 {{{Extension::int16},
1683 {{{i16T, i16T, i16T, i16T, i48T}, SpecificationVersion::V_1_1_DRAFT}}},
1684 {{Extension::fp8e4m3},
1685 {{{fp8e4m3T, fp8e4m3T, fp8e4m3T, fp8e4m3T, fp32T},
1686 SpecificationVersion::V_1_1_DRAFT},
1687 {{fp8e4m3T, fp16T, fp8e4m3T, fp16T, fp32T},
1688 SpecificationVersion::V_1_1_DRAFT},
1689 {{fp8e4m3T, fp32T, fp8e4m3T, fp32T, fp32T},
1690 SpecificationVersion::V_1_1_DRAFT},
1691 {{fp16T, fp8e4m3T, fp16T, fp8e4m3T, fp32T},
1692 SpecificationVersion::V_1_1_DRAFT},
1693 {{fp32T, fp8e4m3T, fp32T, fp8e4m3T, fp32T},
1694 SpecificationVersion::V_1_1_DRAFT}}},
1695 {{Extension::fp8e4m3, Extension::fp8e5m2, Extension::bf16},
1696 {{{fp8e4m3T, fp8e5m2T, fp8e4m3T, fp8e5m2T, bf16T},
1697 SpecificationVersion::V_1_1_DRAFT},
1698 {{fp8e5m2T, fp8e4m3T, fp8e5m2T, fp8e4m3T, bf16T},
1699 SpecificationVersion::V_1_1_DRAFT}},
1700 allOf},
1701 {{Extension::fp8e4m3, Extension::fp8e5m2},
1702 {{{fp8e4m3T, fp8e5m2T, fp8e4m3T, fp8e5m2T, fp32T},
1703 SpecificationVersion::V_1_1_DRAFT},
1704 {{fp8e5m2T, fp8e4m3T, fp8e5m2T, fp8e4m3T, fp32T},
1705 SpecificationVersion::V_1_1_DRAFT}},
1706 allOf},
1707 {{Extension::bf16, Extension::fp8e4m3},
1708 {{{fp8e4m3T, bf16T, fp8e4m3T, bf16T, bf16T},
1709 SpecificationVersion::V_1_1_DRAFT},
1710 {{bf16T, fp8e4m3T, bf16T, fp8e4m3T, bf16T},
1711 SpecificationVersion::V_1_1_DRAFT},
1712 {{fp8e4m3T, bf16T, fp8e4m3T, bf16T, fp32T},
1713 SpecificationVersion::V_1_1_DRAFT},
1714 {{bf16T, fp8e4m3T, bf16T, fp8e4m3T, fp32T},
1715 SpecificationVersion::V_1_1_DRAFT},
1716 {{fp8e4m3T, fp8e4m3T, fp8e4m3T, fp8e4m3T, bf16T},
1717 SpecificationVersion::V_1_1_DRAFT},
1718 {{fp8e4m3T, fp16T, fp8e4m3T, fp16T, bf16T},
1719 SpecificationVersion::V_1_1_DRAFT},
1720 {{fp16T, fp8e4m3T, fp16T, fp8e4m3T, bf16T},
1721 SpecificationVersion::V_1_1_DRAFT}},
1722 allOf},
1723 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp8e4m3,
1724 Extension::bf16},
1725 {{{fp8e4m3T, bs32_fp8ue8m0_fp8e4m3T, fp8e4m3T, fp32T, bf16T},
1726 SpecificationVersion::V_1_1_DRAFT},
1727 {{bs32_fp8ue8m0_fp8e4m3T, fp8e4m3T, fp32T, fp8e4m3T, bf16T},
1728 SpecificationVersion::V_1_1_DRAFT}},
1729 allOf},
1730 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp8e4m3},
1731 {{{fp8e4m3T, bs32_fp8ue8m0_fp8e4m3T, fp8e4m3T, fp32T, fp32T},
1732 SpecificationVersion::V_1_1_DRAFT},
1733 {{bs32_fp8ue8m0_fp8e4m3T, fp8e4m3T, fp32T, fp8e4m3T, fp32T},
1734 SpecificationVersion::V_1_1_DRAFT}},
1735 allOf},
1736 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp8e5m2,
1737 Extension::bf16},
1738 {{{fp8e4m3T, bs32_fp8ue8m0_fp8e5m2T, fp8e4m3T, fp32T, bf16T},
1739 SpecificationVersion::V_1_1_DRAFT},
1740 {{bs32_fp8ue8m0_fp8e5m2T, fp8e4m3T, fp32T, fp8e4m3T, bf16T},
1741 SpecificationVersion::V_1_1_DRAFT}},
1742 allOf},
1743 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp8e5m2},
1744 {{{fp8e4m3T, bs32_fp8ue8m0_fp8e5m2T, fp8e4m3T, fp32T, fp32T},
1745 SpecificationVersion::V_1_1_DRAFT},
1746 {{bs32_fp8ue8m0_fp8e5m2T, fp8e4m3T, fp32T, fp8e4m3T, fp32T},
1747 SpecificationVersion::V_1_1_DRAFT}},
1748 allOf},
1749 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp6e3m2,
1750 Extension::bf16},
1751 {{{fp8e4m3T, bs32_fp8ue8m0_fp6e3m2T, fp8e4m3T, fp32T, bf16T},
1752 SpecificationVersion::V_1_1_DRAFT},
1753 {{bs32_fp8ue8m0_fp6e3m2T, fp8e4m3T, fp32T, fp8e4m3T, bf16T},
1754 SpecificationVersion::V_1_1_DRAFT}},
1755 allOf},
1756 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp6e3m2},
1757 {{{fp8e4m3T, bs32_fp8ue8m0_fp6e3m2T, fp8e4m3T, fp32T, fp32T},
1758 SpecificationVersion::V_1_1_DRAFT},
1759 {{bs32_fp8ue8m0_fp6e3m2T, fp8e4m3T, fp32T, fp8e4m3T, fp32T},
1760 SpecificationVersion::V_1_1_DRAFT}},
1761 allOf},
1762 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp6e2m3,
1763 Extension::bf16},
1764 {{{fp8e4m3T, bs32_fp8ue8m0_fp6e2m3T, fp8e4m3T, fp32T, bf16T},
1765 SpecificationVersion::V_1_1_DRAFT},
1766 {{bs32_fp8ue8m0_fp6e2m3T, fp8e4m3T, fp32T, fp8e4m3T, bf16T},
1767 SpecificationVersion::V_1_1_DRAFT}},
1768 allOf},
1769 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp6e2m3},
1770 {{{fp8e4m3T, bs32_fp8ue8m0_fp6e2m3T, fp8e4m3T, fp32T, fp32T},
1771 SpecificationVersion::V_1_1_DRAFT},
1772 {{bs32_fp8ue8m0_fp6e2m3T, fp8e4m3T, fp32T, fp8e4m3T, fp32T},
1773 SpecificationVersion::V_1_1_DRAFT}},
1774 allOf},
1775 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp4e2m1,
1776 Extension::bf16},
1777 {{{fp8e4m3T, bs32_fp8ue8m0_fp4e2m1T, fp8e4m3T, fp32T, bf16T},
1778 SpecificationVersion::V_1_1_DRAFT},
1779 {{bs32_fp8ue8m0_fp4e2m1T, fp8e4m3T, fp32T, fp8e4m3T, bf16T},
1780 SpecificationVersion::V_1_1_DRAFT}},
1781 allOf},
1782 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp4e2m1},
1783 {{{fp8e4m3T, bs32_fp8ue8m0_fp4e2m1T, fp8e4m3T, fp32T, fp32T},
1784 SpecificationVersion::V_1_1_DRAFT},
1785 {{bs32_fp8ue8m0_fp4e2m1T, fp8e4m3T, fp32T, fp8e4m3T, fp32T},
1786 SpecificationVersion::V_1_1_DRAFT}},
1787 allOf},
1788 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_int8,
1789 Extension::bf16},
1790 {{{fp8e4m3T, bs32_fp8ue8m0_mxint8T, fp8e4m3T, fp32T, bf16T},
1791 SpecificationVersion::V_1_1_DRAFT},
1792 {{bs32_fp8ue8m0_mxint8T, fp8e4m3T, fp32T, fp8e4m3T, bf16T},
1793 SpecificationVersion::V_1_1_DRAFT}},
1794 allOf},
1795 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_int8},
1796 {{{fp8e4m3T, bs32_fp8ue8m0_mxint8T, fp8e4m3T, fp32T, fp32T},
1797 SpecificationVersion::V_1_1_DRAFT},
1798 {{bs32_fp8ue8m0_mxint8T, fp8e4m3T, fp32T, fp8e4m3T, fp32T},
1799 SpecificationVersion::V_1_1_DRAFT}},
1800 allOf},
1801 {{Extension::fp8e5m2},
1802 {{{fp8e5m2T, fp8e5m2T, fp8e5m2T, fp8e5m2T, fp32T},
1803 SpecificationVersion::V_1_1_DRAFT},
1804 {{fp8e5m2T, fp16T, fp8e5m2T, fp16T, fp32T},
1805 SpecificationVersion::V_1_1_DRAFT},
1806 {{fp8e5m2T, fp32T, fp8e5m2T, fp32T, fp32T},
1807 SpecificationVersion::V_1_1_DRAFT},
1808 {{fp16T, fp8e5m2T, fp16T, fp8e5m2T, fp32T},
1809 SpecificationVersion::V_1_1_DRAFT},
1810 {{fp32T, fp8e5m2T, fp32T, fp8e5m2T, fp32T},
1811 SpecificationVersion::V_1_1_DRAFT}}},
1812 {{Extension::bf16, Extension::fp8e5m2},
1813 {{{fp8e5m2T, bf16T, fp8e5m2T, bf16T, bf16T},
1814 SpecificationVersion::V_1_1_DRAFT},
1815 {{bf16T, fp8e5m2T, bf16T, fp8e5m2T, bf16T},
1816 SpecificationVersion::V_1_1_DRAFT},
1817 {{fp8e5m2T, bf16T, fp8e5m2T, bf16T, fp32T},
1818 SpecificationVersion::V_1_1_DRAFT},
1819 {{bf16T, fp8e5m2T, bf16T, fp8e5m2T, fp32T},
1820 SpecificationVersion::V_1_1_DRAFT},
1821 {{fp8e5m2T, fp8e5m2T, fp8e5m2T, fp8e5m2T, bf16T},
1822 SpecificationVersion::V_1_1_DRAFT},
1823 {{fp8e5m2T, fp16T, fp8e5m2T, fp16T, bf16T},
1824 SpecificationVersion::V_1_1_DRAFT},
1825 {{fp16T, fp8e5m2T, fp16T, fp8e5m2T, bf16T},
1826 SpecificationVersion::V_1_1_DRAFT}},
1827 allOf},
1828 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp8e4m3,
1829 Extension::bf16},
1830 {{{fp8e5m2T, bs32_fp8ue8m0_fp8e4m3T, fp8e5m2T, fp32T, bf16T},
1831 SpecificationVersion::V_1_1_DRAFT},
1832 {{bs32_fp8ue8m0_fp8e4m3T, fp8e5m2T, fp32T, fp8e5m2T, bf16T},
1833 SpecificationVersion::V_1_1_DRAFT}},
1834 allOf},
1835 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp8e4m3},
1836 {{{fp8e5m2T, bs32_fp8ue8m0_fp8e4m3T, fp8e5m2T, fp32T, fp32T},
1837 SpecificationVersion::V_1_1_DRAFT},
1838 {{bs32_fp8ue8m0_fp8e4m3T, fp8e5m2T, fp32T, fp8e5m2T, fp32T},
1839 SpecificationVersion::V_1_1_DRAFT}},
1840 allOf},
1841 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp8e5m2,
1842 Extension::bf16},
1843 {{{fp8e5m2T, bs32_fp8ue8m0_fp8e5m2T, fp8e5m2T, fp32T, bf16T},
1844 SpecificationVersion::V_1_1_DRAFT},
1845 {{bs32_fp8ue8m0_fp8e5m2T, fp8e5m2T, fp32T, fp8e5m2T, bf16T},
1846 SpecificationVersion::V_1_1_DRAFT}},
1847 allOf},
1848 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp8e5m2},
1849 {{{fp8e5m2T, bs32_fp8ue8m0_fp8e5m2T, fp8e5m2T, fp32T, fp32T},
1850 SpecificationVersion::V_1_1_DRAFT},
1851 {{bs32_fp8ue8m0_fp8e5m2T, fp8e5m2T, fp32T, fp8e5m2T, fp32T},
1852 SpecificationVersion::V_1_1_DRAFT}},
1853 allOf},
1854 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp6e3m2,
1855 Extension::bf16},
1856 {{{fp8e5m2T, bs32_fp8ue8m0_fp6e3m2T, fp8e5m2T, fp32T, bf16T},
1857 SpecificationVersion::V_1_1_DRAFT},
1858 {{bs32_fp8ue8m0_fp6e3m2T, fp8e5m2T, fp32T, fp8e5m2T, bf16T},
1859 SpecificationVersion::V_1_1_DRAFT}},
1860 allOf},
1861 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp6e3m2},
1862 {{{fp8e5m2T, bs32_fp8ue8m0_fp6e3m2T, fp8e5m2T, fp32T, fp32T},
1863 SpecificationVersion::V_1_1_DRAFT},
1864 {{bs32_fp8ue8m0_fp6e3m2T, fp8e5m2T, fp32T, fp8e5m2T, fp32T},
1865 SpecificationVersion::V_1_1_DRAFT}},
1866 allOf},
1867 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp6e2m3,
1868 Extension::bf16},
1869 {{{fp8e5m2T, bs32_fp8ue8m0_fp6e2m3T, fp8e5m2T, fp32T, bf16T},
1870 SpecificationVersion::V_1_1_DRAFT},
1871 {{bs32_fp8ue8m0_fp6e2m3T, fp8e5m2T, fp32T, fp8e5m2T, bf16T},
1872 SpecificationVersion::V_1_1_DRAFT}},
1873 allOf},
1874 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp6e2m3},
1875 {{{fp8e5m2T, bs32_fp8ue8m0_fp6e2m3T, fp8e5m2T, fp32T, fp32T},
1876 SpecificationVersion::V_1_1_DRAFT},
1877 {{bs32_fp8ue8m0_fp6e2m3T, fp8e5m2T, fp32T, fp8e5m2T, fp32T},
1878 SpecificationVersion::V_1_1_DRAFT}},
1879 allOf},
1880 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp4e2m1,
1881 Extension::bf16},
1882 {{{fp8e5m2T, bs32_fp8ue8m0_fp4e2m1T, fp8e5m2T, fp32T, bf16T},
1883 SpecificationVersion::V_1_1_DRAFT},
1884 {{bs32_fp8ue8m0_fp4e2m1T, fp8e5m2T, fp32T, fp8e5m2T, bf16T},
1885 SpecificationVersion::V_1_1_DRAFT}},
1886 allOf},
1887 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp4e2m1},
1888 {{{fp8e5m2T, bs32_fp8ue8m0_fp4e2m1T, fp8e5m2T, fp32T, fp32T},
1889 SpecificationVersion::V_1_1_DRAFT},
1890 {{bs32_fp8ue8m0_fp4e2m1T, fp8e5m2T, fp32T, fp8e5m2T, fp32T},
1891 SpecificationVersion::V_1_1_DRAFT}},
1892 allOf},
1893 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_int8,
1894 Extension::bf16},
1895 {{{fp8e5m2T, bs32_fp8ue8m0_mxint8T, fp8e5m2T, fp32T, bf16T},
1896 SpecificationVersion::V_1_1_DRAFT},
1897 {{bs32_fp8ue8m0_mxint8T, fp8e5m2T, fp32T, fp8e5m2T, bf16T},
1898 SpecificationVersion::V_1_1_DRAFT}},
1899 allOf},
1900 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_int8},
1901 {{{fp8e5m2T, bs32_fp8ue8m0_mxint8T, fp8e5m2T, fp32T, fp32T},
1902 SpecificationVersion::V_1_1_DRAFT},
1903 {{bs32_fp8ue8m0_mxint8T, fp8e5m2T, fp32T, fp8e5m2T, fp32T},
1904 SpecificationVersion::V_1_1_DRAFT}},
1905 allOf},
1906 {{Extension::bf16},
1907 {{{fp16T, bf16T, fp16T, bf16T, bf16T},
1908 SpecificationVersion::V_1_1_DRAFT},
1909 {{bf16T, fp16T, bf16T, fp16T, bf16T},
1910 SpecificationVersion::V_1_1_DRAFT},
1911 {{bf16T, bf16T, bf16T, bf16T, bf16T},
1912 SpecificationVersion::V_1_1_DRAFT},
1913 {{fp16T, bf16T, fp16T, bf16T, fp32T},
1914 SpecificationVersion::V_1_1_DRAFT},
1915 {{bf16T, fp16T, bf16T, fp16T, fp32T},
1916 SpecificationVersion::V_1_1_DRAFT},
1917 {{bf16T, fp32T, bf16T, fp32T, fp32T},
1918 SpecificationVersion::V_1_1_DRAFT},
1919 {{fp32T, bf16T, fp32T, bf16T, fp32T},
1920 SpecificationVersion::V_1_1_DRAFT},
1921 {{bf16T, bf16T, bf16T, bf16T, fp32T},
1922 SpecificationVersion::V_1_1_DRAFT}}},
1923 {{Extension::mx_common, Extension::mx_fp8e4m3},
1924 {{{fp16T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp32T, fp32T},
1925 SpecificationVersion::V_1_1_DRAFT},
1926 {{fp32T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp32T},
1927 SpecificationVersion::V_1_1_DRAFT},
1928 {{bs32_fp8ue8m0_fp8e4m3T, fp16T, fp32T, fp16T, fp32T},
1929 SpecificationVersion::V_1_1_DRAFT},
1930 {{bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp32T, fp32T},
1931 SpecificationVersion::V_1_1_DRAFT},
1932 {{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp32T},
1933 SpecificationVersion::V_1_1_DRAFT}},
1934 allOf},
1935 {{Extension::mx_common, Extension::mx_fp8e5m2},
1936 {{{fp16T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp32T, fp32T},
1937 SpecificationVersion::V_1_1_DRAFT},
1938 {{fp32T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp32T},
1939 SpecificationVersion::V_1_1_DRAFT},
1940 {{bs32_fp8ue8m0_fp8e5m2T, fp16T, fp32T, fp16T, fp32T},
1941 SpecificationVersion::V_1_1_DRAFT},
1942 {{bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp32T, fp32T},
1943 SpecificationVersion::V_1_1_DRAFT},
1944 {{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp32T},
1945 SpecificationVersion::V_1_1_DRAFT}},
1946 allOf},
1947 {{Extension::mx_common, Extension::mx_fp6e3m2},
1948 {{{fp16T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp32T, fp32T},
1949 SpecificationVersion::V_1_1_DRAFT},
1950 {{fp32T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp32T},
1951 SpecificationVersion::V_1_1_DRAFT},
1952 {{bs32_fp8ue8m0_fp6e3m2T, fp16T, fp32T, fp16T, fp32T},
1953 SpecificationVersion::V_1_1_DRAFT},
1954 {{bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp32T, fp32T},
1955 SpecificationVersion::V_1_1_DRAFT},
1956 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp32T},
1957 SpecificationVersion::V_1_1_DRAFT}},
1958 allOf},
1959 {{Extension::mx_common, Extension::mx_fp6e2m3},
1960 {{{fp16T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp32T, fp32T},
1961 SpecificationVersion::V_1_1_DRAFT},
1962 {{fp32T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp32T},
1963 SpecificationVersion::V_1_1_DRAFT},
1964 {{bs32_fp8ue8m0_fp6e2m3T, fp16T, fp32T, fp16T, fp32T},
1965 SpecificationVersion::V_1_1_DRAFT},
1966 {{bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp32T, fp32T},
1967 SpecificationVersion::V_1_1_DRAFT},
1968 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp32T},
1969 SpecificationVersion::V_1_1_DRAFT}},
1970 allOf},
1971 {{Extension::mx_common, Extension::mx_fp4e2m1},
1972 {{{fp16T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp32T, fp32T},
1973 SpecificationVersion::V_1_1_DRAFT},
1974 {{fp32T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp32T},
1975 SpecificationVersion::V_1_1_DRAFT},
1976 {{bs32_fp8ue8m0_fp4e2m1T, fp16T, fp32T, fp16T, fp32T},
1977 SpecificationVersion::V_1_1_DRAFT},
1978 {{bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp32T, fp32T},
1979 SpecificationVersion::V_1_1_DRAFT},
1980 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp32T},
1981 SpecificationVersion::V_1_1_DRAFT}},
1982 allOf},
1983 {{Extension::mx_common, Extension::mx_int8},
1984 {{{fp16T, bs32_fp8ue8m0_mxint8T, fp16T, fp32T, fp32T},
1985 SpecificationVersion::V_1_1_DRAFT},
1986 {{fp32T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp32T},
1987 SpecificationVersion::V_1_1_DRAFT},
1988 {{bs32_fp8ue8m0_mxint8T, fp16T, fp32T, fp16T, fp32T},
1989 SpecificationVersion::V_1_1_DRAFT},
1990 {{bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp32T, fp32T},
1991 SpecificationVersion::V_1_1_DRAFT},
1992 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp32T},
1993 SpecificationVersion::V_1_1_DRAFT}},
1994 allOf},
1995 {{Extension::bf16, Extension::mx_common, Extension::mx_fp8e4m3},
1996 {{{bf16T, bs32_fp8ue8m0_fp8e4m3T, bf16T, fp32T, bf16T},
1997 SpecificationVersion::V_1_1_DRAFT},
1998 {{bs32_fp8ue8m0_fp8e4m3T, bf16T, fp32T, bf16T, bf16T},
1999 SpecificationVersion::V_1_1_DRAFT},
2000 {{bf16T, bs32_fp8ue8m0_fp8e4m3T, bf16T, fp32T, fp32T},
2001 SpecificationVersion::V_1_1_DRAFT},
2002 {{bs32_fp8ue8m0_fp8e4m3T, bf16T, fp32T, bf16T, fp32T},
2003 SpecificationVersion::V_1_1_DRAFT},
2004 {{fp16T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp32T, bf16T},
2005 SpecificationVersion::V_1_1_DRAFT},
2006 {{bs32_fp8ue8m0_fp8e4m3T, fp16T, fp32T, fp16T, bf16T},
2007 SpecificationVersion::V_1_1_DRAFT},
2008 {{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, bf16T},
2009 SpecificationVersion::V_1_1_DRAFT}},
2010 allOf},
2011 {{Extension::bf16, Extension::mx_common, Extension::mx_fp8e5m2},
2012 {{{bf16T, bs32_fp8ue8m0_fp8e5m2T, bf16T, fp32T, bf16T},
2013 SpecificationVersion::V_1_1_DRAFT},
2014 {{bs32_fp8ue8m0_fp8e5m2T, bf16T, fp32T, bf16T, bf16T},
2015 SpecificationVersion::V_1_1_DRAFT},
2016 {{bf16T, bs32_fp8ue8m0_fp8e5m2T, bf16T, fp32T, fp32T},
2017 SpecificationVersion::V_1_1_DRAFT},
2018 {{bs32_fp8ue8m0_fp8e5m2T, bf16T, fp32T, bf16T, fp32T},
2019 SpecificationVersion::V_1_1_DRAFT},
2020 {{fp16T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp32T, bf16T},
2021 SpecificationVersion::V_1_1_DRAFT},
2022 {{bs32_fp8ue8m0_fp8e5m2T, fp16T, fp32T, fp16T, bf16T},
2023 SpecificationVersion::V_1_1_DRAFT},
2024 {{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, bf16T},
2025 SpecificationVersion::V_1_1_DRAFT}},
2026 allOf},
2027 {{Extension::bf16, Extension::mx_common, Extension::mx_fp6e3m2},
2028 {{{bf16T, bs32_fp8ue8m0_fp6e3m2T, bf16T, fp32T, bf16T},
2029 SpecificationVersion::V_1_1_DRAFT},
2030 {{bs32_fp8ue8m0_fp6e3m2T, bf16T, fp32T, bf16T, bf16T},
2031 SpecificationVersion::V_1_1_DRAFT},
2032 {{bf16T, bs32_fp8ue8m0_fp6e3m2T, bf16T, fp32T, fp32T},
2033 SpecificationVersion::V_1_1_DRAFT},
2034 {{bs32_fp8ue8m0_fp6e3m2T, bf16T, fp32T, bf16T, fp32T},
2035 SpecificationVersion::V_1_1_DRAFT},
2036 {{fp16T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp32T, bf16T},
2037 SpecificationVersion::V_1_1_DRAFT},
2038 {{bs32_fp8ue8m0_fp6e3m2T, fp16T, fp32T, fp16T, bf16T},
2039 SpecificationVersion::V_1_1_DRAFT},
2040 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, bf16T},
2041 SpecificationVersion::V_1_1_DRAFT}},
2042 allOf},
2043 {{Extension::bf16, Extension::mx_common, Extension::mx_fp6e2m3},
2044 {{{bf16T, bs32_fp8ue8m0_fp6e2m3T, bf16T, fp32T, bf16T},
2045 SpecificationVersion::V_1_1_DRAFT},
2046 {{bs32_fp8ue8m0_fp6e2m3T, bf16T, fp32T, bf16T, bf16T},
2047 SpecificationVersion::V_1_1_DRAFT},
2048 {{bf16T, bs32_fp8ue8m0_fp6e2m3T, bf16T, fp32T, fp32T},
2049 SpecificationVersion::V_1_1_DRAFT},
2050 {{bs32_fp8ue8m0_fp6e2m3T, bf16T, fp32T, bf16T, fp32T},
2051 SpecificationVersion::V_1_1_DRAFT},
2052 {{fp16T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp32T, bf16T},
2053 SpecificationVersion::V_1_1_DRAFT},
2054 {{bs32_fp8ue8m0_fp6e2m3T, fp16T, fp32T, fp16T, bf16T},
2055 SpecificationVersion::V_1_1_DRAFT},
2056 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, bf16T},
2057 SpecificationVersion::V_1_1_DRAFT}},
2058 allOf},
2059 {{Extension::bf16, Extension::mx_common, Extension::mx_fp4e2m1},
2060 {{{bf16T, bs32_fp8ue8m0_fp4e2m1T, bf16T, fp32T, bf16T},
2061 SpecificationVersion::V_1_1_DRAFT},
2062 {{bs32_fp8ue8m0_fp4e2m1T, bf16T, fp32T, bf16T, bf16T},
2063 SpecificationVersion::V_1_1_DRAFT},
2064 {{bf16T, bs32_fp8ue8m0_fp4e2m1T, bf16T, fp32T, fp32T},
2065 SpecificationVersion::V_1_1_DRAFT},
2066 {{bs32_fp8ue8m0_fp4e2m1T, bf16T, fp32T, bf16T, fp32T},
2067 SpecificationVersion::V_1_1_DRAFT},
2068 {{fp16T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp32T, bf16T},
2069 SpecificationVersion::V_1_1_DRAFT},
2070 {{bs32_fp8ue8m0_fp4e2m1T, fp16T, fp32T, fp16T, bf16T},
2071 SpecificationVersion::V_1_1_DRAFT},
2072 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, bf16T},
2073 SpecificationVersion::V_1_1_DRAFT}},
2074 allOf},
2075 {{Extension::bf16, Extension::mx_common, Extension::mx_int8},
2076 {{{bf16T, bs32_fp8ue8m0_mxint8T, bf16T, fp32T, bf16T},
2077 SpecificationVersion::V_1_1_DRAFT},
2078 {{bs32_fp8ue8m0_mxint8T, bf16T, fp32T, bf16T, bf16T},
2079 SpecificationVersion::V_1_1_DRAFT},
2080 {{bf16T, bs32_fp8ue8m0_mxint8T, bf16T, fp32T, fp32T},
2081 SpecificationVersion::V_1_1_DRAFT},
2082 {{bs32_fp8ue8m0_mxint8T, bf16T, fp32T, bf16T, fp32T},
2083 SpecificationVersion::V_1_1_DRAFT},
2084 {{fp16T, bs32_fp8ue8m0_mxint8T, fp16T, fp32T, bf16T},
2085 SpecificationVersion::V_1_1_DRAFT},
2086 {{bs32_fp8ue8m0_mxint8T, fp16T, fp32T, fp16T, bf16T},
2087 SpecificationVersion::V_1_1_DRAFT},
2088 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, bf16T},
2089 SpecificationVersion::V_1_1_DRAFT}},
2090 allOf},
2091 {{Extension::mx_common, Extension::mx_fp8e4m3, Extension::mx_fp8e5m2,
2092 Extension::bf16},
2093 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, bf16T},
2094 SpecificationVersion::V_1_1_DRAFT},
2095 {{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, bf16T},
2096 SpecificationVersion::V_1_1_DRAFT}},
2097 allOf},
2098 {{Extension::mx_common, Extension::mx_fp8e4m3, Extension::mx_fp8e5m2},
2099 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp32T},
2100 SpecificationVersion::V_1_1_DRAFT},
2101 {{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp32T},
2102 SpecificationVersion::V_1_1_DRAFT}},
2103 allOf},
2104 {{Extension::mx_common, Extension::mx_fp6e3m2, Extension::mx_fp8e4m3,
2105 Extension::bf16},
2106 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, bf16T},
2107 SpecificationVersion::V_1_1_DRAFT},
2108 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, bf16T},
2109 SpecificationVersion::V_1_1_DRAFT}},
2110 allOf},
2111 {{Extension::mx_common, Extension::mx_fp6e3m2, Extension::mx_fp8e4m3},
2112 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp32T},
2113 SpecificationVersion::V_1_1_DRAFT},
2114 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp32T},
2115 SpecificationVersion::V_1_1_DRAFT}},
2116 allOf},
2117 {{Extension::mx_common, Extension::mx_fp6e2m3, Extension::mx_fp8e4m3,
2118 Extension::bf16},
2119 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, bf16T},
2120 SpecificationVersion::V_1_1_DRAFT},
2121 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, bf16T},
2122 SpecificationVersion::V_1_1_DRAFT}},
2123 allOf},
2124 {{Extension::mx_common, Extension::mx_fp6e2m3, Extension::mx_fp8e4m3},
2125 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp32T},
2126 SpecificationVersion::V_1_1_DRAFT},
2127 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp32T},
2128 SpecificationVersion::V_1_1_DRAFT}},
2129 allOf},
2130 {{Extension::mx_common, Extension::mx_fp4e2m1, Extension::mx_fp8e4m3,
2131 Extension::bf16},
2132 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, bf16T},
2133 SpecificationVersion::V_1_1_DRAFT},
2134 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, bf16T},
2135 SpecificationVersion::V_1_1_DRAFT}},
2136 allOf},
2137 {{Extension::mx_common, Extension::mx_fp4e2m1, Extension::mx_fp8e4m3},
2138 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp32T},
2139 SpecificationVersion::V_1_1_DRAFT},
2140 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp32T},
2141 SpecificationVersion::V_1_1_DRAFT}},
2142 allOf},
2143 {{Extension::mx_common, Extension::mx_fp8e4m3, Extension::mx_int8,
2144 Extension::bf16},
2145 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, bf16T},
2146 SpecificationVersion::V_1_1_DRAFT},
2147 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, bf16T},
2148 SpecificationVersion::V_1_1_DRAFT}},
2149 allOf},
2150 {{Extension::mx_common, Extension::mx_fp8e4m3, Extension::mx_int8},
2151 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp32T},
2152 SpecificationVersion::V_1_1_DRAFT},
2153 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp8e4m3T, fp32T, fp32T, fp32T},
2154 SpecificationVersion::V_1_1_DRAFT}},
2155 allOf},
2156 {{Extension::mx_common, Extension::mx_fp6e3m2, Extension::mx_fp8e5m2,
2157 Extension::bf16},
2158 {{{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, bf16T},
2159 SpecificationVersion::V_1_1_DRAFT},
2160 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, bf16T},
2161 SpecificationVersion::V_1_1_DRAFT}},
2162 allOf},
2163 {{Extension::mx_common, Extension::mx_fp6e3m2, Extension::mx_fp8e5m2},
2164 {{{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp32T},
2165 SpecificationVersion::V_1_1_DRAFT},
2166 {{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp32T},
2167 SpecificationVersion::V_1_1_DRAFT}},
2168 allOf},
2169 {{Extension::mx_common, Extension::mx_fp6e2m3, Extension::mx_fp8e5m2,
2170 Extension::bf16},
2171 {{{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, bf16T},
2172 SpecificationVersion::V_1_1_DRAFT},
2173 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, bf16T},
2174 SpecificationVersion::V_1_1_DRAFT}},
2175 allOf},
2176 {{Extension::mx_common, Extension::mx_fp6e2m3, Extension::mx_fp8e5m2},
2177 {{{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp32T},
2178 SpecificationVersion::V_1_1_DRAFT},
2179 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp32T},
2180 SpecificationVersion::V_1_1_DRAFT}},
2181 allOf},
2182 {{Extension::mx_common, Extension::mx_fp4e2m1, Extension::mx_fp8e5m2,
2183 Extension::bf16},
2184 {{{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, bf16T},
2185 SpecificationVersion::V_1_1_DRAFT},
2186 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, bf16T},
2187 SpecificationVersion::V_1_1_DRAFT}},
2188 allOf},
2189 {{Extension::mx_common, Extension::mx_fp4e2m1, Extension::mx_fp8e5m2},
2190 {{{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp32T},
2191 SpecificationVersion::V_1_1_DRAFT},
2192 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp32T},
2193 SpecificationVersion::V_1_1_DRAFT}},
2194 allOf},
2195 {{Extension::mx_common, Extension::mx_fp8e5m2, Extension::mx_int8,
2196 Extension::bf16},
2197 {{{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, bf16T},
2198 SpecificationVersion::V_1_1_DRAFT},
2199 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, bf16T},
2200 SpecificationVersion::V_1_1_DRAFT}},
2201 allOf},
2202 {{Extension::mx_common, Extension::mx_fp8e5m2, Extension::mx_int8},
2203 {{{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp32T},
2204 SpecificationVersion::V_1_1_DRAFT},
2205 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp8e5m2T, fp32T, fp32T, fp32T},
2206 SpecificationVersion::V_1_1_DRAFT}},
2207 allOf},
2208 {{Extension::mx_common, Extension::mx_fp6e2m3, Extension::mx_fp6e3m2,
2209 Extension::bf16},
2210 {{{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, bf16T},
2211 SpecificationVersion::V_1_1_DRAFT},
2212 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, bf16T},
2213 SpecificationVersion::V_1_1_DRAFT}},
2214 allOf},
2215 {{Extension::mx_common, Extension::mx_fp6e2m3, Extension::mx_fp6e3m2},
2216 {{{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp32T},
2217 SpecificationVersion::V_1_1_DRAFT},
2218 {{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp32T},
2219 SpecificationVersion::V_1_1_DRAFT}},
2220 allOf},
2221 {{Extension::mx_common, Extension::mx_fp4e2m1, Extension::mx_fp6e3m2,
2222 Extension::bf16},
2223 {{{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, bf16T},
2224 SpecificationVersion::V_1_1_DRAFT},
2225 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, bf16T},
2226 SpecificationVersion::V_1_1_DRAFT}},
2227 allOf},
2228 {{Extension::mx_common, Extension::mx_fp4e2m1, Extension::mx_fp6e3m2},
2229 {{{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp32T},
2230 SpecificationVersion::V_1_1_DRAFT},
2231 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp32T},
2232 SpecificationVersion::V_1_1_DRAFT}},
2233 allOf},
2234 {{Extension::mx_common, Extension::mx_fp6e3m2, Extension::mx_int8,
2235 Extension::bf16},
2236 {{{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, bf16T},
2237 SpecificationVersion::V_1_1_DRAFT},
2238 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, bf16T},
2239 SpecificationVersion::V_1_1_DRAFT}},
2240 allOf},
2241 {{Extension::mx_common, Extension::mx_fp6e3m2, Extension::mx_int8},
2242 {{{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp32T},
2243 SpecificationVersion::V_1_1_DRAFT},
2244 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp6e3m2T, fp32T, fp32T, fp32T},
2245 SpecificationVersion::V_1_1_DRAFT}},
2246 allOf},
2247 {{Extension::mx_common, Extension::mx_fp4e2m1, Extension::mx_fp6e2m3,
2248 Extension::bf16},
2249 {{{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, bf16T},
2250 SpecificationVersion::V_1_1_DRAFT},
2251 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, bf16T},
2252 SpecificationVersion::V_1_1_DRAFT}},
2253 allOf},
2254 {{Extension::mx_common, Extension::mx_fp4e2m1, Extension::mx_fp6e2m3},
2255 {{{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp32T},
2256 SpecificationVersion::V_1_1_DRAFT},
2257 {{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp32T},
2258 SpecificationVersion::V_1_1_DRAFT}},
2259 allOf},
2260 {{Extension::mx_common, Extension::mx_fp6e2m3, Extension::mx_int8,
2261 Extension::bf16},
2262 {{{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, bf16T},
2263 SpecificationVersion::V_1_1_DRAFT},
2264 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, bf16T},
2265 SpecificationVersion::V_1_1_DRAFT}},
2266 allOf},
2267 {{Extension::mx_common, Extension::mx_fp6e2m3, Extension::mx_int8},
2268 {{{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp32T},
2269 SpecificationVersion::V_1_1_DRAFT},
2270 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp6e2m3T, fp32T, fp32T, fp32T},
2271 SpecificationVersion::V_1_1_DRAFT}},
2272 allOf},
2273 {{Extension::mx_common, Extension::mx_fp4e2m1, Extension::mx_int8,
2274 Extension::bf16},
2275 {{{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, bf16T},
2276 SpecificationVersion::V_1_1_DRAFT},
2277 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, bf16T},
2278 SpecificationVersion::V_1_1_DRAFT}},
2279 allOf},
2280 {{Extension::mx_common, Extension::mx_fp4e2m1, Extension::mx_int8},
2281 {{{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_mxint8T, fp32T, fp32T, fp32T},
2282 SpecificationVersion::V_1_1_DRAFT},
2283 {{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_fp4e2m1T, fp32T, fp32T, fp32T},
2284 SpecificationVersion::V_1_1_DRAFT}},
2285 allOf}}},
2286 {"tosa.matmul_t_block_scaled",
2287 {{{Extension::mxfp},
2288 {{{fp4e2m1T, fp8ue8m0T, fp4e2m1T, fp8ue8m0T, fp32T},
2289 SpecificationVersion::V_1_1_DRAFT},
2290 {{fp6e2m3T, fp8ue8m0T, fp6e2m3T, fp8ue8m0T, fp32T},
2291 SpecificationVersion::V_1_1_DRAFT},
2292 {{fp6e3m2T, fp8ue8m0T, fp6e3m2T, fp8ue8m0T, fp32T},
2293 SpecificationVersion::V_1_1_DRAFT},
2294 {{fp8e4m3T, fp8ue8m0T, fp8e4m3T, fp8ue8m0T, fp32T},
2295 SpecificationVersion::V_1_1_DRAFT},
2296 {{fp8e5m2T, fp8ue8m0T, fp8e5m2T, fp8ue8m0T, fp32T},
2297 SpecificationVersion::V_1_1_DRAFT},
2298 {{mxint8T, fp8ue8m0T, mxint8T, fp8ue8m0T, fp32T},
2299 SpecificationVersion::V_1_1_DRAFT}}}}},
2300 {"tosa.max_pool2d",
2301 {{{Extension::int16}, {{{i16T, i16T}, SpecificationVersion::V_1_0}}},
2302 {{Extension::fp8e4m3},
2303 {{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
2304 {{Extension::fp8e5m2},
2305 {{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
2306 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2307 {"tosa.max_pool2d_adaptive",
2308 {{{Extension::int16}, {{{i16T, i16T}, SpecificationVersion::V_1_1_DRAFT}}},
2309 {{Extension::fp8e4m3},
2310 {{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT}}},
2311 {{Extension::fp8e5m2},
2312 {{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT}}},
2313 {{Extension::bf16},
2314 {{{bf16T, bf16T}, SpecificationVersion::V_1_1_DRAFT}}}}},
2315 {"tosa.rfft2d",
2316 {{{Extension::fft},
2317 {{{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
2318 {"tosa.transpose_conv2d",
2319 {{{Extension::int4},
2320 {{{i8T, i4T, i32T, i8T, i4T, i32T, i32T}, SpecificationVersion::V_1_0}}},
2321 {{Extension::int16},
2322 {{{i16T, i8T, i48T, i16T, i8T, i48T, i48T},
2323 SpecificationVersion::V_1_0}}},
2324 {{Extension::fp8e4m3},
2325 {{{fp8e4m3T, fp8e4m3T, fp16T, fp8e4m3T, fp8e4m3T, fp16T, fp16T},
2326 SpecificationVersion::V_1_0},
2327 {{fp8e4m3T, fp8e4m3T, fp32T, fp8e4m3T, fp8e4m3T, fp32T, fp32T},
2328 SpecificationVersion::V_1_1_DRAFT}}},
2329 {{Extension::fp8e5m2},
2330 {{{fp8e5m2T, fp8e5m2T, fp16T, fp8e5m2T, fp8e5m2T, fp16T, fp16T},
2331 SpecificationVersion::V_1_0},
2332 {{fp8e5m2T, fp8e5m2T, fp32T, fp8e5m2T, fp8e5m2T, fp32T, fp32T},
2333 SpecificationVersion::V_1_1_DRAFT}}},
2334 {{Extension::bf16},
2335 {{{bf16T, bf16T, bf16T, bf16T, bf16T, fp32T, bf16T},
2336 SpecificationVersion::V_1_0}}}}},
2337 {"tosa.clamp",
2338 {{{Extension::int16}, {{{i16T, i16T}, SpecificationVersion::V_1_0}}},
2339 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2340 {"tosa.erf",
2341 {{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2342 {"tosa.sigmoid",
2343 {{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2344 {"tosa.tanh",
2345 {{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2346 {"tosa.add",
2347 {{{Extension::int64},
2348 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
2349 {{Extension::bf16},
2350 {{{bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2351 {"tosa.arithmetic_right_shift",
2352 {{{Extension::int64},
2353 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
2354 {"tosa.bitwise_and",
2355 {{{Extension::int64},
2356 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
2357 {"tosa.bitwise_or",
2358 {{{Extension::int64},
2359 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
2360 {"tosa.bitwise_xor",
2361 {{{Extension::int64},
2362 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
2363 {"tosa.intdiv",
2364 {{{Extension::int64},
2365 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
2366 {"tosa.logical_left_shift",
2367 {{{Extension::int64},
2368 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
2369 {"tosa.logical_right_shift",
2370 {{{Extension::int64},
2371 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
2372 {"tosa.maximum",
2373 {{{Extension::int64},
2374 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
2375 {{Extension::bf16},
2376 {{{bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2377 {"tosa.minimum",
2378 {{{Extension::int64},
2379 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
2380 {{Extension::bf16},
2381 {{{bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2382 {"tosa.mul",
2383 {{{Extension::int64},
2384 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
2385 {{Extension::bf16},
2386 {{{bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2387 {"tosa.pow",
2388 {{{Extension::bf16},
2389 {{{bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2390 {"tosa.sub",
2391 {{{Extension::int64},
2392 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
2393 {{Extension::bf16},
2394 {{{bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2395 {"tosa.table",
2396 {{{Extension::int16},
2397 {{{i16T, i16T, i32T}, SpecificationVersion::V_1_0}}}}},
2398 {"tosa.abs",
2399 {{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
2400 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2401 {"tosa.bitwise_not",
2402 {{{Extension::int64},
2403 {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
2404 {"tosa.ceil",
2405 {{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2406 {"tosa.clz",
2407 {{{Extension::int64},
2408 {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
2409 {"tosa.cos",
2410 {{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2411 {"tosa.exp",
2412 {{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2413 {"tosa.floor",
2414 {{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2415 {"tosa.log",
2416 {{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2417 {"tosa.negate",
2418 {{{Extension::int64},
2419 {{{i64T, i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
2420 {{Extension::bf16},
2421 {{{bf16T, bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2422 {"tosa.reciprocal",
2423 {{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2424 {"tosa.rsqrt",
2425 {{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2426 {"tosa.sin",
2427 {{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2428 {"tosa.select",
2429 {{{Extension::int64},
2430 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
2431 {{Extension::bf16},
2432 {{{bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2433 {"tosa.equal",
2434 {{{Extension::int64},
2435 {{{i64T, i64T, boolT}, SpecificationVersion::V_1_1_DRAFT}}},
2436 {{Extension::bf16},
2437 {{{bf16T, bf16T, boolT}, SpecificationVersion::V_1_0}}}}},
2438 {"tosa.greater",
2439 {{{Extension::int64},
2440 {{{i64T, i64T, boolT}, SpecificationVersion::V_1_1_DRAFT}}},
2441 {{Extension::bf16},
2442 {{{bf16T, bf16T, boolT}, SpecificationVersion::V_1_0}}}}},
2443 {"tosa.greater_equal",
2444 {{{Extension::int64},
2445 {{{i64T, i64T, boolT}, SpecificationVersion::V_1_1_DRAFT}}},
2446 {{Extension::bf16},
2447 {{{bf16T, bf16T, boolT}, SpecificationVersion::V_1_0}}}}},
2448 {"tosa.reduce_max",
2449 {{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
2450 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2451 {"tosa.reduce_min",
2452 {{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
2453 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2454 {"tosa.reduce_product",
2455 {{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2456 {"tosa.reduce_sum",
2457 {{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
2458 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2459 {"tosa.concat",
2460 {{{Extension::int16}, {{{i16T, i16T}, SpecificationVersion::V_1_0}}},
2461 {{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
2462 {{Extension::fp8e4m3},
2463 {{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
2464 {{Extension::fp8e5m2},
2465 {{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
2466 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}},
2467 {{Extension::mx_common, Extension::mx_fp8e4m3},
2468 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp8e4m3T},
2469 SpecificationVersion::V_1_1_DRAFT}},
2470 allOf},
2471 {{Extension::mx_common, Extension::mx_fp8e5m2},
2472 {{{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp8e5m2T},
2473 SpecificationVersion::V_1_1_DRAFT}},
2474 allOf},
2475 {{Extension::mx_common, Extension::mx_fp6e3m2},
2476 {{{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp6e3m2T},
2477 SpecificationVersion::V_1_1_DRAFT}},
2478 allOf},
2479 {{Extension::mx_common, Extension::mx_fp6e2m3},
2480 {{{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp6e2m3T},
2481 SpecificationVersion::V_1_1_DRAFT}},
2482 allOf},
2483 {{Extension::mx_common, Extension::mx_fp4e2m1},
2484 {{{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp4e2m1T},
2485 SpecificationVersion::V_1_1_DRAFT}},
2486 allOf},
2487 {{Extension::mx_common, Extension::mx_int8},
2488 {{{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_mxint8T},
2489 SpecificationVersion::V_1_1_DRAFT}},
2490 allOf}}},
2491 {"tosa.pad",
2492 {{{Extension::int64},
2493 {{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
2494 {{Extension::fp8e4m3},
2495 {{{fp8e4m3T, fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
2496 {{Extension::fp8e5m2},
2497 {{{fp8e5m2T, fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
2498 {{Extension::bf16},
2499 {{{bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2500 {"tosa.reshape",
2501 {{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
2502 {{Extension::fp8e4m3},
2503 {{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
2504 {{Extension::fp8e5m2},
2505 {{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
2506 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}},
2507 {{Extension::mx_common, Extension::mx_fp8e4m3},
2508 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp8e4m3T},
2509 SpecificationVersion::V_1_1_DRAFT}},
2510 allOf},
2511 {{Extension::mx_common, Extension::mx_fp8e5m2},
2512 {{{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp8e5m2T},
2513 SpecificationVersion::V_1_1_DRAFT}},
2514 allOf},
2515 {{Extension::mx_common, Extension::mx_fp6e3m2},
2516 {{{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp6e3m2T},
2517 SpecificationVersion::V_1_1_DRAFT}},
2518 allOf},
2519 {{Extension::mx_common, Extension::mx_fp6e2m3},
2520 {{{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp6e2m3T},
2521 SpecificationVersion::V_1_1_DRAFT}},
2522 allOf},
2523 {{Extension::mx_common, Extension::mx_fp4e2m1},
2524 {{{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp4e2m1T},
2525 SpecificationVersion::V_1_1_DRAFT}},
2526 allOf},
2527 {{Extension::mx_common, Extension::mx_int8},
2528 {{{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_mxint8T},
2529 SpecificationVersion::V_1_1_DRAFT}},
2530 allOf}}},
2531 {"tosa.reverse",
2532 {{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
2533 {{Extension::fp8e4m3},
2534 {{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
2535 {{Extension::fp8e5m2},
2536 {{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
2537 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2538 {"tosa.slice",
2539 {{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
2540 {{Extension::fp8e4m3},
2541 {{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
2542 {{Extension::fp8e5m2},
2543 {{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
2544 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2545 {"tosa.tile",
2546 {{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
2547 {{Extension::fp8e4m3},
2548 {{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
2549 {{Extension::fp8e5m2},
2550 {{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
2551 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2552 {"tosa.transpose",
2553 {{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
2554 {{Extension::fp8e4m3},
2555 {{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
2556 {{Extension::fp8e5m2},
2557 {{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
2558 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2559 {"tosa.gather",
2560 {{{Extension::fp8e4m3},
2561 {{{fp8e4m3T, i32T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
2562 {{Extension::fp8e5m2},
2563 {{{fp8e5m2T, i32T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
2564 {{Extension::bf16},
2565 {{{bf16T, i32T, bf16T}, SpecificationVersion::V_1_0}}},
2566 {{Extension::int64},
2567 {{{i8T, i64T, i8T}, SpecificationVersion::V_1_1_DRAFT},
2568 {{i16T, i64T, i16T}, SpecificationVersion::V_1_1_DRAFT},
2569 {{i32T, i64T, i32T}, SpecificationVersion::V_1_1_DRAFT},
2570 {{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT},
2571 {{fp16T, i64T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
2572 {{fp32T, i64T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
2573 {{boolT, i64T, boolT}, SpecificationVersion::V_1_1_DRAFT}}},
2574 {{Extension::fp8e4m3, Extension::int64},
2575 {{{fp8e4m3T, i64T, fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT}},
2576 allOf},
2577 {{Extension::fp8e5m2, Extension::int64},
2578 {{{fp8e5m2T, i64T, fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT}},
2579 allOf},
2580 {{Extension::bf16, Extension::int64},
2581 {{{bf16T, i64T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
2582 allOf}}},
2583 {"tosa.row_gather",
2584 {{{Extension::int16},
2585 {{{i16T, i32T, i16T}, SpecificationVersion::V_1_1_DRAFT}}},
2586 {{Extension::int64},
2587 {{{i64T, i32T, i64T}, SpecificationVersion::V_1_1_DRAFT},
2588 {{boolT, i64T, boolT}, SpecificationVersion::V_1_1_DRAFT},
2589 {{i8T, i64T, i8T}, SpecificationVersion::V_1_1_DRAFT},
2590 {{i32T, i64T, i32T}, SpecificationVersion::V_1_1_DRAFT},
2591 {{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT},
2592 {{fp16T, i64T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
2593 {{fp32T, i64T, fp32T}, SpecificationVersion::V_1_1_DRAFT}}},
2594 {{Extension::int16, Extension::int64},
2595 {{{i16T, i64T, i16T}, SpecificationVersion::V_1_1_DRAFT}},
2596 allOf},
2597 {{Extension::fp8e4m3},
2598 {{{fp8e4m3T, i32T, fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT}}},
2599 {{Extension::fp8e5m2},
2600 {{{fp8e5m2T, i32T, fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT}}},
2601 {{Extension::bf16},
2602 {{{bf16T, i32T, bf16T}, SpecificationVersion::V_1_1_DRAFT}}},
2603 {{Extension::mx_common, Extension::mx_fp8e4m3},
2604 {{{bs32_fp8ue8m0_fp8e4m3T, i32T, bs32_fp8ue8m0_fp8e4m3T},
2605 SpecificationVersion::V_1_1_DRAFT}},
2606 allOf},
2607 {{Extension::mx_common, Extension::mx_fp8e5m2},
2608 {{{bs32_fp8ue8m0_fp8e5m2T, i32T, bs32_fp8ue8m0_fp8e5m2T},
2609 SpecificationVersion::V_1_1_DRAFT}},
2610 allOf},
2611 {{Extension::mx_common, Extension::mx_fp6e3m2},
2612 {{{bs32_fp8ue8m0_fp6e3m2T, i32T, bs32_fp8ue8m0_fp6e3m2T},
2613 SpecificationVersion::V_1_1_DRAFT}},
2614 allOf},
2615 {{Extension::mx_common, Extension::mx_fp6e2m3},
2616 {{{bs32_fp8ue8m0_fp6e2m3T, i32T, bs32_fp8ue8m0_fp6e2m3T},
2617 SpecificationVersion::V_1_1_DRAFT}},
2618 allOf},
2619 {{Extension::mx_common, Extension::mx_fp4e2m1},
2620 {{{bs32_fp8ue8m0_fp4e2m1T, i32T, bs32_fp8ue8m0_fp4e2m1T},
2621 SpecificationVersion::V_1_1_DRAFT}},
2622 allOf},
2623 {{Extension::mx_common, Extension::mx_int8},
2624 {{{bs32_fp8ue8m0_mxint8T, i32T, bs32_fp8ue8m0_mxint8T},
2625 SpecificationVersion::V_1_1_DRAFT}},
2626 allOf},
2627 {{Extension::fp8e4m3, Extension::int64},
2628 {{{fp8e4m3T, i64T, fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT}},
2629 allOf},
2630 {{Extension::fp8e5m2, Extension::int64},
2631 {{{fp8e5m2T, i64T, fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT}},
2632 allOf},
2633 {{Extension::bf16, Extension::int64},
2634 {{{bf16T, i64T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
2635 allOf},
2636 {{Extension::int64, Extension::mx_common, Extension::mx_fp8e4m3},
2637 {{{bs32_fp8ue8m0_fp8e4m3T, i64T, bs32_fp8ue8m0_fp8e4m3T},
2638 SpecificationVersion::V_1_1_DRAFT}},
2639 allOf},
2640 {{Extension::int64, Extension::mx_common, Extension::mx_fp8e5m2},
2641 {{{bs32_fp8ue8m0_fp8e5m2T, i64T, bs32_fp8ue8m0_fp8e5m2T},
2642 SpecificationVersion::V_1_1_DRAFT}},
2643 allOf},
2644 {{Extension::int64, Extension::mx_common, Extension::mx_fp6e3m2},
2645 {{{bs32_fp8ue8m0_fp6e3m2T, i64T, bs32_fp8ue8m0_fp6e3m2T},
2646 SpecificationVersion::V_1_1_DRAFT}},
2647 allOf},
2648 {{Extension::int64, Extension::mx_common, Extension::mx_fp6e2m3},
2649 {{{bs32_fp8ue8m0_fp6e2m3T, i64T, bs32_fp8ue8m0_fp6e2m3T},
2650 SpecificationVersion::V_1_1_DRAFT}},
2651 allOf},
2652 {{Extension::int64, Extension::mx_common, Extension::mx_fp4e2m1},
2653 {{{bs32_fp8ue8m0_fp4e2m1T, i64T, bs32_fp8ue8m0_fp4e2m1T},
2654 SpecificationVersion::V_1_1_DRAFT}},
2655 allOf},
2656 {{Extension::int64, Extension::mx_common, Extension::mx_int8},
2657 {{{bs32_fp8ue8m0_mxint8T, i64T, bs32_fp8ue8m0_mxint8T},
2658 SpecificationVersion::V_1_1_DRAFT}},
2659 allOf}}},
2660 {"tosa.row_gather_block_scaled",
2661 {{{Extension::fp8e4m3},
2662 {{{fp8e4m3T, i32T, i32T, fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT}}},
2663 {{Extension::fp8e5m2},
2664 {{{fp8e5m2T, i32T, i32T, fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT}}},
2665 {{Extension::bf16},
2666 {{{bf16T, i32T, i32T, bf16T}, SpecificationVersion::V_1_1_DRAFT}}},
2667 {{Extension::int64},
2668 {{{i8T, i64T, i32T, i8T}, SpecificationVersion::V_1_1_DRAFT},
2669 {{i16T, i64T, i32T, i16T}, SpecificationVersion::V_1_1_DRAFT},
2670 {{i32T, i64T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT},
2671 {{i64T, i64T, i32T, i64T}, SpecificationVersion::V_1_1_DRAFT},
2672 {{fp16T, i64T, i32T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
2673 {{fp32T, i64T, i32T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
2674 {{boolT, i64T, i32T, boolT}, SpecificationVersion::V_1_1_DRAFT}}},
2675 {{Extension::fp8e4m3, Extension::int64},
2676 {{{fp8e4m3T, i64T, i32T, fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT}},
2677 allOf},
2678 {{Extension::fp8e5m2, Extension::int64},
2679 {{{fp8e5m2T, i64T, i32T, fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT}},
2680 allOf},
2681 {{Extension::bf16, Extension::int64},
2682 {{{bf16T, i64T, i32T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
2683 allOf},
2684 {{Extension::mxfp},
2685 {{{fp4e2m1T, fp8ue8m0T, i32T, i32T, fp4e2m1T, fp8ue8m0T},
2686 SpecificationVersion::V_1_1_DRAFT},
2687 {{fp6e2m3T, fp8ue8m0T, i32T, i32T, fp6e2m3T, fp8ue8m0T},
2688 SpecificationVersion::V_1_1_DRAFT},
2689 {{fp6e3m2T, fp8ue8m0T, i32T, i32T, fp6e3m2T, fp8ue8m0T},
2690 SpecificationVersion::V_1_1_DRAFT},
2691 {{fp8e4m3T, fp8ue8m0T, i32T, i32T, fp8e4m3T, fp8ue8m0T},
2692 SpecificationVersion::V_1_1_DRAFT},
2693 {{fp8e5m2T, fp8ue8m0T, i32T, i32T, fp8e5m2T, fp8ue8m0T},
2694 SpecificationVersion::V_1_1_DRAFT},
2695 {{mxint8T, fp8ue8m0T, i32T, i32T, mxint8T, fp8ue8m0T},
2696 SpecificationVersion::V_1_1_DRAFT}}},
2697 {{Extension::mxfp, Extension::int64},
2698 {{{fp4e2m1T, fp8ue8m0T, i64T, i32T, fp4e2m1T, fp8ue8m0T},
2699 SpecificationVersion::V_1_1_DRAFT},
2700 {{fp6e2m3T, fp8ue8m0T, i64T, i32T, fp6e2m3T, fp8ue8m0T},
2701 SpecificationVersion::V_1_1_DRAFT},
2702 {{fp6e3m2T, fp8ue8m0T, i64T, i32T, fp6e3m2T, fp8ue8m0T},
2703 SpecificationVersion::V_1_1_DRAFT},
2704 {{fp8e4m3T, fp8ue8m0T, i64T, i32T, fp8e4m3T, fp8ue8m0T},
2705 SpecificationVersion::V_1_1_DRAFT},
2706 {{fp8e5m2T, fp8ue8m0T, i64T, i32T, fp8e5m2T, fp8ue8m0T},
2707 SpecificationVersion::V_1_1_DRAFT},
2708 {{mxint8T, fp8ue8m0T, i64T, i32T, mxint8T, fp8ue8m0T},
2709 SpecificationVersion::V_1_1_DRAFT}},
2710 allOf}}},
2711 {"tosa.scatter",
2712 {{{Extension::fp8e4m3},
2713 {{{fp8e4m3T, i32T, fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
2714 {{Extension::fp8e5m2},
2715 {{{fp8e5m2T, i32T, fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
2716 {{Extension::bf16},
2717 {{{bf16T, i32T, bf16T, bf16T}, SpecificationVersion::V_1_0}}},
2718 {{Extension::int64},
2719 {{{i8T, i64T, i8T, i8T}, SpecificationVersion::V_1_1_DRAFT},
2720 {{i16T, i64T, i16T, i16T}, SpecificationVersion::V_1_1_DRAFT},
2721 {{i32T, i64T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT},
2722 {{i64T, i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT},
2723 {{fp16T, i64T, fp16T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
2724 {{fp32T, i64T, fp32T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
2725 {{boolT, i64T, boolT, boolT}, SpecificationVersion::V_1_1_DRAFT}}},
2726 {{Extension::fp8e4m3, Extension::int64},
2727 {{{fp8e4m3T, i64T, fp8e4m3T, fp8e4m3T},
2728 SpecificationVersion::V_1_1_DRAFT}},
2729 allOf},
2730 {{Extension::fp8e5m2, Extension::int64},
2731 {{{fp8e5m2T, i64T, fp8e5m2T, fp8e5m2T},
2732 SpecificationVersion::V_1_1_DRAFT}},
2733 allOf},
2734 {{Extension::bf16, Extension::int64},
2735 {{{bf16T, i64T, bf16T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
2736 allOf}}},
2737 {"tosa.resize",
2738 {{{Extension::int16},
2739 {{{i16T, i48T}, SpecificationVersion::V_1_0},
2740 {{i16T, i16T}, SpecificationVersion::V_1_0}}},
2741 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}},
2742 {{Extension::fp8e4m3},
2743 {{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT}}},
2744 {{Extension::fp8e5m2},
2745 {{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT}}},
2746 {{Extension::mx_common, Extension::mx_fp8e4m3},
2747 {{{bs32_fp8ue8m0_fp8e4m3T, bs32_fp8ue8m0_fp8e4m3T},
2748 SpecificationVersion::V_1_1_DRAFT}},
2749 allOf},
2750 {{Extension::mx_common, Extension::mx_fp8e5m2},
2751 {{{bs32_fp8ue8m0_fp8e5m2T, bs32_fp8ue8m0_fp8e5m2T},
2752 SpecificationVersion::V_1_1_DRAFT}},
2753 allOf},
2754 {{Extension::mx_common, Extension::mx_fp6e3m2},
2755 {{{bs32_fp8ue8m0_fp6e3m2T, bs32_fp8ue8m0_fp6e3m2T},
2756 SpecificationVersion::V_1_1_DRAFT}},
2757 allOf},
2758 {{Extension::mx_common, Extension::mx_fp6e2m3},
2759 {{{bs32_fp8ue8m0_fp6e2m3T, bs32_fp8ue8m0_fp6e2m3T},
2760 SpecificationVersion::V_1_1_DRAFT}},
2761 allOf},
2762 {{Extension::mx_common, Extension::mx_fp4e2m1},
2763 {{{bs32_fp8ue8m0_fp4e2m1T, bs32_fp8ue8m0_fp4e2m1T},
2764 SpecificationVersion::V_1_1_DRAFT}},
2765 allOf},
2766 {{Extension::mx_common, Extension::mx_int8},
2767 {{{bs32_fp8ue8m0_mxint8T, bs32_fp8ue8m0_mxint8T},
2768 SpecificationVersion::V_1_1_DRAFT}},
2769 allOf}}},
2770 {"tosa.cast",
2771 {{{Extension::bf16},
2772 {{{i8T, bf16T}, SpecificationVersion::V_1_0},
2773 {{i16T, bf16T}, SpecificationVersion::V_1_0},
2774 {{i32T, bf16T}, SpecificationVersion::V_1_0},
2775 {{bf16T, i8T}, SpecificationVersion::V_1_0},
2776 {{bf16T, i16T}, SpecificationVersion::V_1_0},
2777 {{bf16T, i32T}, SpecificationVersion::V_1_0},
2778 {{bf16T, fp32T}, SpecificationVersion::V_1_0},
2779 {{fp32T, bf16T}, SpecificationVersion::V_1_0}}},
2780 {{Extension::int64},
2781 {{{i32T, i64T}, SpecificationVersion::V_1_1_DRAFT},
2782 {{i64T, i32T}, SpecificationVersion::V_1_1_DRAFT},
2783 {{boolT, i64T}, SpecificationVersion::V_1_1_DRAFT},
2784 {{i64T, boolT}, SpecificationVersion::V_1_1_DRAFT}}},
2785 {{Extension::bf16, Extension::fp8e4m3},
2786 {{{bf16T, fp8e4m3T}, SpecificationVersion::V_1_0},
2787 {{fp8e4m3T, bf16T}, SpecificationVersion::V_1_0}},
2788 allOf},
2789 {{Extension::bf16, Extension::fp8e5m2},
2790 {{{bf16T, fp8e5m2T}, SpecificationVersion::V_1_0},
2791 {{fp8e5m2T, bf16T}, SpecificationVersion::V_1_0}},
2792 allOf},
2793 {{Extension::fp8e4m3},
2794 {{{fp8e4m3T, fp16T}, SpecificationVersion::V_1_0},
2795 {{fp8e4m3T, fp32T}, SpecificationVersion::V_1_0},
2796 {{fp16T, fp8e4m3T}, SpecificationVersion::V_1_0},
2797 {{fp32T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
2798 {{Extension::fp8e5m2},
2799 {{{fp8e5m2T, fp16T}, SpecificationVersion::V_1_0},
2800 {{fp8e5m2T, fp32T}, SpecificationVersion::V_1_0},
2801 {{fp16T, fp8e5m2T}, SpecificationVersion::V_1_0},
2802 {{fp32T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
2803 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp8e4m3},
2804 {{{fp8e4m3T, bs32_fp8ue8m0_fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT},
2805 {{bs32_fp8ue8m0_fp8e4m3T, fp8e4m3T},
2806 SpecificationVersion::V_1_1_DRAFT}},
2807 allOf},
2808 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp8e5m2},
2809 {{{fp8e4m3T, bs32_fp8ue8m0_fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT},
2810 {{bs32_fp8ue8m0_fp8e5m2T, fp8e4m3T},
2811 SpecificationVersion::V_1_1_DRAFT}},
2812 allOf},
2813 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp6e3m2},
2814 {{{fp8e4m3T, bs32_fp8ue8m0_fp6e3m2T}, SpecificationVersion::V_1_1_DRAFT},
2815 {{bs32_fp8ue8m0_fp6e3m2T, fp8e4m3T},
2816 SpecificationVersion::V_1_1_DRAFT}},
2817 allOf},
2818 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp6e2m3},
2819 {{{fp8e4m3T, bs32_fp8ue8m0_fp6e2m3T}, SpecificationVersion::V_1_1_DRAFT},
2820 {{bs32_fp8ue8m0_fp6e2m3T, fp8e4m3T},
2821 SpecificationVersion::V_1_1_DRAFT}},
2822 allOf},
2823 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp4e2m1},
2824 {{{fp8e4m3T, bs32_fp8ue8m0_fp4e2m1T}, SpecificationVersion::V_1_1_DRAFT},
2825 {{bs32_fp8ue8m0_fp4e2m1T, fp8e4m3T},
2826 SpecificationVersion::V_1_1_DRAFT}},
2827 allOf},
2828 {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_int8},
2829 {{{fp8e4m3T, bs32_fp8ue8m0_mxint8T}, SpecificationVersion::V_1_1_DRAFT},
2830 {{bs32_fp8ue8m0_mxint8T, fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT}},
2831 allOf},
2832 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp8e4m3},
2833 {{{fp8e5m2T, bs32_fp8ue8m0_fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT},
2834 {{bs32_fp8ue8m0_fp8e4m3T, fp8e5m2T},
2835 SpecificationVersion::V_1_1_DRAFT}},
2836 allOf},
2837 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp8e5m2},
2838 {{{fp8e5m2T, bs32_fp8ue8m0_fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT},
2839 {{bs32_fp8ue8m0_fp8e5m2T, fp8e5m2T},
2840 SpecificationVersion::V_1_1_DRAFT}},
2841 allOf},
2842 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp6e3m2},
2843 {{{fp8e5m2T, bs32_fp8ue8m0_fp6e3m2T}, SpecificationVersion::V_1_1_DRAFT},
2844 {{bs32_fp8ue8m0_fp6e3m2T, fp8e5m2T},
2845 SpecificationVersion::V_1_1_DRAFT}},
2846 allOf},
2847 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp6e2m3},
2848 {{{fp8e5m2T, bs32_fp8ue8m0_fp6e2m3T}, SpecificationVersion::V_1_1_DRAFT},
2849 {{bs32_fp8ue8m0_fp6e2m3T, fp8e5m2T},
2850 SpecificationVersion::V_1_1_DRAFT}},
2851 allOf},
2852 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp4e2m1},
2853 {{{fp8e5m2T, bs32_fp8ue8m0_fp4e2m1T}, SpecificationVersion::V_1_1_DRAFT},
2854 {{bs32_fp8ue8m0_fp4e2m1T, fp8e5m2T},
2855 SpecificationVersion::V_1_1_DRAFT}},
2856 allOf},
2857 {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_int8},
2858 {{{fp8e5m2T, bs32_fp8ue8m0_mxint8T}, SpecificationVersion::V_1_1_DRAFT},
2859 {{bs32_fp8ue8m0_mxint8T, fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT}},
2860 allOf},
2861 {{Extension::mx_common, Extension::mx_fp8e4m3},
2862 {{{fp16T, bs32_fp8ue8m0_fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT},
2863 {{fp32T, bs32_fp8ue8m0_fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT},
2864 {{bs32_fp8ue8m0_fp8e4m3T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
2865 {{bs32_fp8ue8m0_fp8e4m3T, fp32T}, SpecificationVersion::V_1_1_DRAFT}},
2866 allOf},
2867 {{Extension::mx_common, Extension::mx_fp8e5m2},
2868 {{{fp16T, bs32_fp8ue8m0_fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT},
2869 {{fp32T, bs32_fp8ue8m0_fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT},
2870 {{bs32_fp8ue8m0_fp8e5m2T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
2871 {{bs32_fp8ue8m0_fp8e5m2T, fp32T}, SpecificationVersion::V_1_1_DRAFT}},
2872 allOf},
2873 {{Extension::mx_common, Extension::mx_fp6e3m2},
2874 {{{fp16T, bs32_fp8ue8m0_fp6e3m2T}, SpecificationVersion::V_1_1_DRAFT},
2875 {{fp32T, bs32_fp8ue8m0_fp6e3m2T}, SpecificationVersion::V_1_1_DRAFT},
2876 {{bs32_fp8ue8m0_fp6e3m2T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
2877 {{bs32_fp8ue8m0_fp6e3m2T, fp32T}, SpecificationVersion::V_1_1_DRAFT}},
2878 allOf},
2879 {{Extension::mx_common, Extension::mx_fp6e2m3},
2880 {{{fp16T, bs32_fp8ue8m0_fp6e2m3T}, SpecificationVersion::V_1_1_DRAFT},
2881 {{fp32T, bs32_fp8ue8m0_fp6e2m3T}, SpecificationVersion::V_1_1_DRAFT},
2882 {{bs32_fp8ue8m0_fp6e2m3T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
2883 {{bs32_fp8ue8m0_fp6e2m3T, fp32T}, SpecificationVersion::V_1_1_DRAFT}},
2884 allOf},
2885 {{Extension::mx_common, Extension::mx_fp4e2m1},
2886 {{{fp16T, bs32_fp8ue8m0_fp4e2m1T}, SpecificationVersion::V_1_1_DRAFT},
2887 {{fp32T, bs32_fp8ue8m0_fp4e2m1T}, SpecificationVersion::V_1_1_DRAFT},
2888 {{bs32_fp8ue8m0_fp4e2m1T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
2889 {{bs32_fp8ue8m0_fp4e2m1T, fp32T}, SpecificationVersion::V_1_1_DRAFT}},
2890 allOf},
2891 {{Extension::mx_common, Extension::mx_int8},
2892 {{{fp16T, bs32_fp8ue8m0_mxint8T}, SpecificationVersion::V_1_1_DRAFT},
2893 {{fp32T, bs32_fp8ue8m0_mxint8T}, SpecificationVersion::V_1_1_DRAFT},
2894 {{bs32_fp8ue8m0_mxint8T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
2895 {{bs32_fp8ue8m0_mxint8T, fp32T}, SpecificationVersion::V_1_1_DRAFT}},
2896 allOf},
2897 {{Extension::bf16, Extension::mx_common, Extension::mx_fp8e4m3},
2898 {{{bf16T, bs32_fp8ue8m0_fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT},
2899 {{bs32_fp8ue8m0_fp8e4m3T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
2900 allOf},
2901 {{Extension::bf16, Extension::mx_common, Extension::mx_fp8e5m2},
2902 {{{bf16T, bs32_fp8ue8m0_fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT},
2903 {{bs32_fp8ue8m0_fp8e5m2T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
2904 allOf},
2905 {{Extension::bf16, Extension::mx_common, Extension::mx_fp6e3m2},
2906 {{{bf16T, bs32_fp8ue8m0_fp6e3m2T}, SpecificationVersion::V_1_1_DRAFT},
2907 {{bs32_fp8ue8m0_fp6e3m2T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
2908 allOf},
2909 {{Extension::bf16, Extension::mx_common, Extension::mx_fp6e2m3},
2910 {{{bf16T, bs32_fp8ue8m0_fp6e2m3T}, SpecificationVersion::V_1_1_DRAFT},
2911 {{bs32_fp8ue8m0_fp6e2m3T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
2912 allOf},
2913 {{Extension::bf16, Extension::mx_common, Extension::mx_fp4e2m1},
2914 {{{bf16T, bs32_fp8ue8m0_fp4e2m1T}, SpecificationVersion::V_1_1_DRAFT},
2915 {{bs32_fp8ue8m0_fp4e2m1T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
2916 allOf},
2917 {{Extension::bf16, Extension::mx_common, Extension::mx_int8},
2918 {{{bf16T, bs32_fp8ue8m0_mxint8T}, SpecificationVersion::V_1_1_DRAFT},
2919 {{bs32_fp8ue8m0_mxint8T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
2920 allOf}}},
2921 {"tosa.cast_from_block_scaled",
2922 {{{Extension::bf16, Extension::mxfp},
2923 {{{fp4e2m1T, fp8ue8m0T, bf16T}, SpecificationVersion::V_1_1_DRAFT},
2924 {{fp6e2m3T, fp8ue8m0T, bf16T}, SpecificationVersion::V_1_1_DRAFT},
2925 {{fp6e3m2T, fp8ue8m0T, bf16T}, SpecificationVersion::V_1_1_DRAFT},
2926 {{fp8e4m3T, fp8ue8m0T, bf16T}, SpecificationVersion::V_1_1_DRAFT},
2927 {{fp8e5m2T, fp8ue8m0T, bf16T}, SpecificationVersion::V_1_1_DRAFT},
2928 {{mxint8T, fp8ue8m0T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
2929 allOf},
2930 {{Extension::mxfp},
2931 {{{fp4e2m1T, fp8ue8m0T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
2932 {{fp6e2m3T, fp8ue8m0T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
2933 {{fp6e3m2T, fp8ue8m0T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
2934 {{fp8e4m3T, fp8ue8m0T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
2935 {{fp8e5m2T, fp8ue8m0T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
2936 {{mxint8T, fp8ue8m0T, fp32T}, SpecificationVersion::V_1_1_DRAFT}}}}},
2937 {"tosa.cast_to_block_scaled",
2938 {{{Extension::mxfp},
2939 {{{bf16T, fp4e2m1T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
2940 {{fp32T, fp4e2m1T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
2941 {{fp32T, fp6e2m3T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
2942 {{fp32T, fp6e3m2T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
2943 {{fp32T, fp8e4m3T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
2944 {{fp32T, fp8e5m2T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
2945 {{fp32T, mxint8T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT}}},
2946 {{Extension::bf16, Extension::mxfp},
2947 {{{bf16T, fp6e2m3T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
2948 {{bf16T, fp6e3m2T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
2949 {{bf16T, fp8e4m3T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
2950 {{bf16T, fp8e5m2T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
2951 {{bf16T, mxint8T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT}},
2952 allOf}}},
2953 {"tosa.rescale",
2954 {{{Extension::int16},
2955 {{{i48T, i48T, i8T, i8T}, SpecificationVersion::V_1_0},
2956 {{i48T, i48T, i16T, i16T}, SpecificationVersion::V_1_0},
2957 {{i48T, i48T, i32T, i32T}, SpecificationVersion::V_1_0}}}}},
2958 {"tosa.const",
2959 {{{Extension::int4}, {{{i4T}, SpecificationVersion::V_1_0}}},
2960 {{Extension::int16}, {{{i48T}, SpecificationVersion::V_1_0}}},
2961 {{Extension::int64}, {{{i64T}, SpecificationVersion::V_1_1_DRAFT}}},
2962 {{Extension::fp8e4m3}, {{{fp8e4m3T}, SpecificationVersion::V_1_0}}},
2963 {{Extension::fp8e5m2}, {{{fp8e5m2T}, SpecificationVersion::V_1_0}}},
2964 {{Extension::bf16}, {{{bf16T}, SpecificationVersion::V_1_0}}},
2965 {{Extension::mxfp},
2966 {{{fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
2967 {{fp6e3m2T}, SpecificationVersion::V_1_1_DRAFT},
2968 {{fp6e2m3T}, SpecificationVersion::V_1_1_DRAFT},
2969 {{fp4e2m1T}, SpecificationVersion::V_1_1_DRAFT},
2970 {{mxint8T}, SpecificationVersion::V_1_1_DRAFT}}},
2971 {{Extension::mx_common, Extension::mx_fp8e4m3},
2972 {{{bs32_fp8ue8m0_fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT}},
2973 allOf},
2974 {{Extension::mx_common, Extension::mx_fp8e5m2},
2975 {{{bs32_fp8ue8m0_fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT}},
2976 allOf},
2977 {{Extension::mx_common, Extension::mx_fp6e3m2},
2978 {{{bs32_fp8ue8m0_fp6e3m2T}, SpecificationVersion::V_1_1_DRAFT}},
2979 allOf},
2980 {{Extension::mx_common, Extension::mx_fp6e2m3},
2981 {{{bs32_fp8ue8m0_fp6e2m3T}, SpecificationVersion::V_1_1_DRAFT}},
2982 allOf},
2983 {{Extension::mx_common, Extension::mx_fp4e2m1},
2984 {{{bs32_fp8ue8m0_fp4e2m1T}, SpecificationVersion::V_1_1_DRAFT}},
2985 allOf},
2986 {{Extension::mx_common, Extension::mx_int8},
2987 {{{bs32_fp8ue8m0_mxint8T}, SpecificationVersion::V_1_1_DRAFT}},
2988 allOf}}},
2989 {"tosa.identity",
2990 {{{Extension::int4}, {{{i4T, i4T}, SpecificationVersion::V_1_0}}},
2991 {{Extension::int16}, {{{i48T, i48T}, SpecificationVersion::V_1_0}}},
2992 {{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
2993 {{Extension::fp8e4m3},
2994 {{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
2995 {{Extension::fp8e5m2},
2996 {{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
2997 {{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
2998 {"tosa.variable",
2999 {{{Extension::variable},
3000 {{{i8T}, SpecificationVersion::V_1_0},
3001 {{fp16T}, SpecificationVersion::V_1_0},
3002 {{fp32T}, SpecificationVersion::V_1_0}}}}},
3003 {"tosa.variable_write",
3004 {{{Extension::variable},
3005 {{{i8T}, SpecificationVersion::V_1_0},
3006 {{fp16T}, SpecificationVersion::V_1_0},
3007 {{fp32T}, SpecificationVersion::V_1_0}}}}},
3008 {"tosa.variable_read",
3009 {{{Extension::variable},
3010 {{{i8T}, SpecificationVersion::V_1_0},
3011 {{fp16T}, SpecificationVersion::V_1_0},
3012 {{fp32T}, SpecificationVersion::V_1_0}}}}},
3013 {"tosa.add_shape", {{{Extension::shape}, {}}}},
3014 {"tosa.assert_equal_shape", {{{Extension::shape}, {}}}},
3015 {"tosa.concat_shape", {{{Extension::shape}, {}}}},
3016 {"tosa.div_ceil_shape", {{{Extension::shape}, {}}}},
3017 {"tosa.div_floor_shape", {{{Extension::shape}, {}}}},
3018 {"tosa.exp2_shape", {{{Extension::shape}, {}}}},
3019 {"tosa.log2_ceil_shape", {{{Extension::shape}, {}}}},
3020 {"tosa.log2_floor_shape", {{{Extension::shape}, {}}}},
3021 {"tosa.max_shape", {{{Extension::shape}, {}}}},
3022 {"tosa.min_shape", {{{Extension::shape}, {}}}},
3023 {"tosa.mod_shape", {{{Extension::shape}, {}}}},
3024 {"tosa.mul_shape", {{{Extension::shape}, {}}}},
3025 {"tosa.slice_shape", {{{Extension::shape}, {}}}},
3026 {"tosa.sub_shape", {{{Extension::shape}, {}}}},
3027 {"tosa.while_loop", {{{Extension::controlflow}, {}}}},
3028 {"tosa.cond_if", {{{Extension::controlflow}, {}}}},
3029 {"tosa.dim",
3030 {{{Extension::shape},
3031 {{{boolT}, SpecificationVersion::V_1_1_DRAFT},
3032 {{i8T}, SpecificationVersion::V_1_1_DRAFT},
3033 {{i16T}, SpecificationVersion::V_1_1_DRAFT},
3034 {{i32T}, SpecificationVersion::V_1_1_DRAFT},
3035 {{fp16T}, SpecificationVersion::V_1_1_DRAFT},
3036 {{fp32T}, SpecificationVersion::V_1_1_DRAFT}}},
3037 {{Extension::fp8e4m3, Extension::shape},
3038 {{{fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT}},
3039 allOf},
3040 {{Extension::fp8e5m2, Extension::shape},
3041 {{{fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT}},
3042 allOf},
3043 {{Extension::bf16, Extension::shape},
3044 {{{bf16T}, SpecificationVersion::V_1_1_DRAFT}},
3045 allOf},
3046 {{Extension::mxfp, Extension::shape},
3047 {{{fp6e3m2T}, SpecificationVersion::V_1_1_DRAFT},
3048 {{fp6e2m3T}, SpecificationVersion::V_1_1_DRAFT},
3049 {{fp4e2m1T}, SpecificationVersion::V_1_1_DRAFT},
3050 {{mxint8T}, SpecificationVersion::V_1_1_DRAFT},
3051 {{fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT}},
3052 allOf},
3053 {{Extension::int64, Extension::shape},
3054 {{{i64T}, SpecificationVersion::V_1_1_DRAFT}},
3055 allOf}}}};
3056
3057// End of auto-generated metadata
Generated on for MLIR by doxygen 1.14.0