Lines Matching refs: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()