Lines Matching full:context
22 MLIRContext context; in TEST() local
25 bindDims(&context, m, n, k); in TEST()
26 auto mapA = AffineMapAttr::get(AffineMap::get(3, 0, {m, k}, &context)); in TEST()
27 auto mapB = AffineMapAttr::get(AffineMap::get(3, 0, {k, n}, &context)); in TEST()
28 auto mapC = AffineMapAttr::get(AffineMap::get(3, 0, {m, n}, &context)); in TEST()
29 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
35 MLIRContext context; in TEST() local
38 bindDims(&context, k, m, n); // bind in different order in TEST()
39 auto mapA = AffineMapAttr::get(AffineMap::get(3, 0, {m, k}, &context)); in TEST()
40 auto mapB = AffineMapAttr::get(AffineMap::get(3, 0, {k, n}, &context)); in TEST()
41 auto mapC = AffineMapAttr::get(AffineMap::get(3, 0, {m, n}, &context)); in TEST()
42 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
48 MLIRContext context; in TEST() local
51 bindDims(&context, k, n, m); // bind in different order in TEST()
52 auto mapA = AffineMapAttr::get(AffineMap::get(3, 0, {m, k}, &context)); in TEST()
53 auto mapB = AffineMapAttr::get(AffineMap::get(3, 0, {k, n}, &context)); in TEST()
54 auto mapC = AffineMapAttr::get(AffineMap::get(3, 0, {m, n}, &context)); in TEST()
55 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
61 MLIRContext context; in TEST() local
64 bindDims(&context, m, n, k); in TEST()
65 auto mapA = AffineMapAttr::get(AffineMap::get(3, 0, {k, n}, &context)); in TEST()
66 auto mapB = AffineMapAttr::get(AffineMap::get(3, 0, {m, k}, &context)); in TEST()
67 auto mapC = AffineMapAttr::get(AffineMap::get(3, 0, {m, n}, &context)); in TEST()
68 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
74 MLIRContext context; in TEST() local
77 bindDims(&context, m, n, k); in TEST()
78 auto mapA = AffineMapAttr::get(AffineMap::get(3, 0, {k, m}, &context)); in TEST()
79 auto mapB = AffineMapAttr::get(AffineMap::get(3, 0, {k, n}, &context)); in TEST()
80 auto mapC = AffineMapAttr::get(AffineMap::get(3, 0, {m, n}, &context)); in TEST()
81 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
87 MLIRContext context; in TEST() local
90 bindDims(&context, m, n, k); in TEST()
91 auto mapA = AffineMapAttr::get(AffineMap::get(3, 0, {m, k}, &context)); in TEST()
92 auto mapB = AffineMapAttr::get(AffineMap::get(3, 0, {k, n}, &context)); in TEST()
93 auto maps = ArrayAttr::get(&context, {mapA, mapB}); in TEST()
99 MLIRContext context; in TEST() local
102 bindDims(&context, m, n, k); in TEST()
103 auto mapA = AffineMapAttr::get(AffineMap::get(3, 0, {m, k}, &context)); in TEST()
104 auto mapB = AffineMapAttr::get(AffineMap::get(3, 0, {k, n}, &context)); in TEST()
105 auto mapC = AffineMapAttr::get(AffineMap::get(3, 0, {m, n}, &context)); in TEST()
106 auto mapD = AffineMapAttr::get(AffineMap::get(3, 0, {k, m}, &context)); in TEST()
108 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC, mapD}); in TEST()
114 MLIRContext context; in TEST() local
117 bindDims(&context, m, n, k); in TEST()
118 auto mapA = AffineMapAttr::get(AffineMap::get(3, 0, {m}, &context)); in TEST()
119 auto mapB = AffineMapAttr::get(AffineMap::get(3, 0, {k, n}, &context)); in TEST()
120 auto mapC = AffineMapAttr::get(AffineMap::get(3, 0, {m, n}, &context)); in TEST()
121 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
127 MLIRContext context; in TEST() local
130 bindDims(&context, m, n, k); in TEST()
131 auto mapA = AffineMapAttr::get(AffineMap::get(3, 0, {k, n}, &context)); in TEST()
132 auto mapB = AffineMapAttr::get(AffineMap::get(3, 0, {m, k}, &context)); in TEST()
133 auto mapC = AffineMapAttr::get(AffineMap::get(3, 0, {m, n}, &context)); in TEST()
134 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
140 MLIRContext context; in TEST() local
143 bindDims(&context, k, m, n); // bind in different order in TEST()
144 auto mapA = AffineMapAttr::get(AffineMap::get(3, 0, {k, n}, &context)); in TEST()
145 auto mapB = AffineMapAttr::get(AffineMap::get(3, 0, {m, k}, &context)); in TEST()
146 auto mapC = AffineMapAttr::get(AffineMap::get(3, 0, {m, n}, &context)); in TEST()
147 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
153 MLIRContext context; in TEST() local
156 bindDims(&context, k, n, m); // bind in different order in TEST()
157 auto mapA = AffineMapAttr::get(AffineMap::get(3, 0, {k, n}, &context)); in TEST()
158 auto mapB = AffineMapAttr::get(AffineMap::get(3, 0, {m, k}, &context)); in TEST()
159 auto mapC = AffineMapAttr::get(AffineMap::get(3, 0, {m, n}, &context)); in TEST()
160 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
166 MLIRContext context; in TEST() local
169 bindDims(&context, m, n, k); in TEST()
170 auto mapA = AffineMapAttr::get(AffineMap::get(3, 0, {m, k}, &context)); in TEST()
171 auto mapB = AffineMapAttr::get(AffineMap::get(3, 0, {k, n}, &context)); in TEST()
172 auto mapC = AffineMapAttr::get(AffineMap::get(3, 0, {m, n}, &context)); in TEST()
173 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
179 MLIRContext context; in TEST() local
182 bindDims(&context, m, n, k); in TEST()
183 auto mapA = AffineMapAttr::get(AffineMap::get(3, 0, {n, k}, &context)); in TEST()
184 auto mapB = AffineMapAttr::get(AffineMap::get(3, 0, {m, k}, &context)); in TEST()
185 auto mapC = AffineMapAttr::get(AffineMap::get(3, 0, {m, n}, &context)); in TEST()
186 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
192 MLIRContext context; in TEST() local
195 bindDims(&context, batch, m, n, k); in TEST()
196 auto mapA = AffineMapAttr::get(AffineMap::get(4, 0, {batch, m, k}, &context)); in TEST()
197 auto mapB = AffineMapAttr::get(AffineMap::get(4, 0, {batch, k, n}, &context)); in TEST()
198 auto mapC = AffineMapAttr::get(AffineMap::get(4, 0, {batch, m, n}, &context)); in TEST()
199 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
205 MLIRContext context; in TEST() local
208 bindDims(&context, k, batch, m, n); // bind in different order in TEST()
209 auto mapA = AffineMapAttr::get(AffineMap::get(4, 0, {batch, m, k}, &context)); in TEST()
210 auto mapB = AffineMapAttr::get(AffineMap::get(4, 0, {batch, k, n}, &context)); in TEST()
211 auto mapC = AffineMapAttr::get(AffineMap::get(4, 0, {batch, m, n}, &context)); in TEST()
212 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
218 MLIRContext context; in TEST() local
221 bindDims(&context, batch, k, n, m); // bind in different order in TEST()
222 auto mapA = AffineMapAttr::get(AffineMap::get(4, 0, {batch, m, k}, &context)); in TEST()
223 auto mapB = AffineMapAttr::get(AffineMap::get(4, 0, {batch, k, n}, &context)); in TEST()
224 auto mapC = AffineMapAttr::get(AffineMap::get(4, 0, {batch, m, n}, &context)); in TEST()
225 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
231 MLIRContext context; in TEST() local
234 bindDims(&context, batch, m, n, k); in TEST()
235 auto mapA = AffineMapAttr::get(AffineMap::get(4, 0, {batch, k, m}, &context)); in TEST()
236 auto mapB = AffineMapAttr::get(AffineMap::get(4, 0, {batch, k, n}, &context)); in TEST()
237 auto mapC = AffineMapAttr::get(AffineMap::get(4, 0, {batch, m, n}, &context)); in TEST()
238 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
244 MLIRContext context; in TEST() local
247 bindDims(&context, k, n); in TEST()
248 auto mapA = AffineMapAttr::get(AffineMap::get(2, 0, {k}, &context)); in TEST()
249 auto mapB = AffineMapAttr::get(AffineMap::get(2, 0, {k, n}, &context)); in TEST()
250 auto mapC = AffineMapAttr::get(AffineMap::get(2, 0, {n}, &context)); in TEST()
251 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
257 MLIRContext context; in TEST() local
260 bindDims(&context, n, k); // bind in different order in TEST()
261 auto mapA = AffineMapAttr::get(AffineMap::get(2, 0, {k}, &context)); in TEST()
262 auto mapB = AffineMapAttr::get(AffineMap::get(2, 0, {k, n}, &context)); in TEST()
263 auto mapC = AffineMapAttr::get(AffineMap::get(2, 0, {n}, &context)); in TEST()
264 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
270 MLIRContext context; in TEST() local
273 bindDims(&context, k, n); in TEST()
274 auto mapA = AffineMapAttr::get(AffineMap::get(2, 0, {k}, &context)); in TEST()
275 auto mapB = AffineMapAttr::get(AffineMap::get(2, 0, {n, k}, &context)); in TEST()
276 auto mapC = AffineMapAttr::get(AffineMap::get(2, 0, {n}, &context)); in TEST()
277 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
283 MLIRContext context; in TEST() local
286 bindDims(&context, k, n); in TEST()
287 auto mapA = AffineMapAttr::get(AffineMap::get(2, 0, {n, k}, &context)); in TEST()
288 auto mapB = AffineMapAttr::get(AffineMap::get(2, 0, {k}, &context)); in TEST()
289 auto mapC = AffineMapAttr::get(AffineMap::get(2, 0, {n}, &context)); in TEST()
290 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
296 MLIRContext context; in TEST() local
299 bindDims(&context, n, k); // bind in different order in TEST()
300 auto mapA = AffineMapAttr::get(AffineMap::get(2, 0, {n, k}, &context)); in TEST()
301 auto mapB = AffineMapAttr::get(AffineMap::get(2, 0, {k}, &context)); in TEST()
302 auto mapC = AffineMapAttr::get(AffineMap::get(2, 0, {n}, &context)); in TEST()
303 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
309 MLIRContext context; in TEST() local
312 bindDims(&context, k, n); in TEST()
313 auto mapA = AffineMapAttr::get(AffineMap::get(2, 0, {k, n}, &context)); in TEST()
314 auto mapB = AffineMapAttr::get(AffineMap::get(2, 0, {k}, &context)); in TEST()
315 auto mapC = AffineMapAttr::get(AffineMap::get(2, 0, {n}, &context)); in TEST()
316 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
322 MLIRContext context; in TEST() local
325 bindDims(&context, batch, k, n); in TEST()
326 auto mapA = AffineMapAttr::get(AffineMap::get(3, 0, {batch, n, k}, &context)); in TEST()
327 auto mapB = AffineMapAttr::get(AffineMap::get(3, 0, {batch, k}, &context)); in TEST()
328 auto mapC = AffineMapAttr::get(AffineMap::get(3, 0, {batch, n}, &context)); in TEST()
329 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
335 MLIRContext context; in TEST() local
338 bindDims(&context, batch, n, k); // bind in different order in TEST()
339 auto mapA = AffineMapAttr::get(AffineMap::get(3, 0, {batch, n, k}, &context)); in TEST()
340 auto mapB = AffineMapAttr::get(AffineMap::get(3, 0, {batch, k}, &context)); in TEST()
341 auto mapC = AffineMapAttr::get(AffineMap::get(3, 0, {batch, n}, &context)); in TEST()
342 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
348 MLIRContext context; in TEST() local
351 bindDims(&context, m, n, k); in TEST()
352 auto mapA = AffineMapAttr::get(AffineMap::get(3, 0, {m, k}, &context)); in TEST()
353 auto mapB = AffineMapAttr::get(AffineMap::get(3, 0, {k, n}, &context)); in TEST()
354 auto mapC = AffineMapAttr::get(AffineMap::get(3, 0, {m, n}, &context)); in TEST()
355 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
361 MLIRContext context; in TEST() local
364 bindDims(&context, batch, k, n); in TEST()
365 auto mapA = AffineMapAttr::get(AffineMap::get(3, 0, {batch, k, n}, &context)); in TEST()
366 auto mapB = AffineMapAttr::get(AffineMap::get(3, 0, {batch, k}, &context)); in TEST()
367 auto mapC = AffineMapAttr::get(AffineMap::get(3, 0, {batch, n}, &context)); in TEST()
368 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
374 MLIRContext context; in TEST() local
377 bindDims(&context, batch, k, n); in TEST()
378 auto mapA = AffineMapAttr::get(AffineMap::get(3, 0, {batch, k}, &context)); in TEST()
379 auto mapB = AffineMapAttr::get(AffineMap::get(3, 0, {batch, k, n}, &context)); in TEST()
380 auto mapC = AffineMapAttr::get(AffineMap::get(3, 0, {batch, n}, &context)); in TEST()
381 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
387 MLIRContext context; in TEST() local
390 bindDims(&context, batch, n, k); // bind in different order in TEST()
391 auto mapA = AffineMapAttr::get(AffineMap::get(3, 0, {batch, k}, &context)); in TEST()
392 auto mapB = AffineMapAttr::get(AffineMap::get(3, 0, {batch, k, n}, &context)); in TEST()
393 auto mapC = AffineMapAttr::get(AffineMap::get(3, 0, {batch, n}, &context)); in TEST()
394 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
400 MLIRContext context; in TEST() local
403 bindDims(&context, m, n, k); in TEST()
404 auto mapA = AffineMapAttr::get(AffineMap::get(3, 0, {m, k}, &context)); in TEST()
405 auto mapB = AffineMapAttr::get(AffineMap::get(3, 0, {k, n}, &context)); in TEST()
406 auto mapC = AffineMapAttr::get(AffineMap::get(3, 0, {m, n}, &context)); in TEST()
407 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
413 MLIRContext context; in TEST() local
416 bindDims(&context, batch, k, n); in TEST()
417 auto mapA = AffineMapAttr::get(AffineMap::get(3, 0, {batch, k}, &context)); in TEST()
418 auto mapB = AffineMapAttr::get(AffineMap::get(3, 0, {batch, n, k}, &context)); in TEST()
419 auto mapC = AffineMapAttr::get(AffineMap::get(3, 0, {batch, n}, &context)); in TEST()
420 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()