Lines Matching defs:dims

103 // Checks whether the input Affine `map` contains two consecutive dims that
181   auto dims = *res;
182   if (dims.m.size() != 1 || dims.n.size() != 1 || dims.k.size() != 1)
197   if (llvm::any_of(indexingMaps, [&dims](AffineMap m) {
199                dims.batch.size() + 2 /* any two of {m,n,k} */;
203   auto numOfBatchDims = dims.batch.size();
224       matchOperandMap(indexingMaps[0], numOfBatchDims, dims.m[0], dims.k[0]);
226       matchOperandMap(indexingMaps[1], numOfBatchDims, dims.k[0], dims.n[0]);
228       matchOperandMap(indexingMaps[2], numOfBatchDims, dims.m[0], dims.n[0]);
280     auto dims = *equivalentToBroadcast;
283         dims);