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