Lines Matching full:batch
194 AffineExpr batch, m, n, k; 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()
207 AffineExpr batch, m, n, k; 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()
220 AffineExpr batch, m, n, k; 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()
233 AffineExpr batch, m, n, k; 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()
324 AffineExpr batch, k, n; 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()
337 AffineExpr batch, k, n; 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()
363 AffineExpr batch, k, n; 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()
376 AffineExpr batch, k, n; 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()
389 AffineExpr batch, k, n; 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()
415 AffineExpr batch, k, n; 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()