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