Lines Matching defs:tileSizes

56   SmallVector<OpFoldResult> tileSizes(allTileSizes);
60   for (int idx = 0, e = tileSizes.size(), zerosCount = 0; idx < e; ++idx) {
61     if (getConstantIntValue(tileSizes[idx - zerosCount]) ==
64       tileSizes.erase(tileSizes.begin() + idx - zerosCount);
73   for (unsigned idx = 0, e = tileSizes.size(); idx < e; ++idx)
74     res.push_back(Range{b.getIndexAttr(0), shapeSizes[idx], tileSizes[idx]});
126   spec.tileSizes.push_back(tileSize);
139       spec.tileSizes.push_back(tileSize);
146   auto tripCountCheck = [&](SmallVector<int64_t> tileSizes,
150     for (auto [tileSize, tripCount] : llvm::zip(tileSizes, tripCounts))
155   if (!tripCountCheck(spec.tileSizes, spec.tripCounts, loopRange))
208   spec.tileSizes.push_back(targetSizeValue);
227         spec.tileSizes.push_back(constStepOp);
231       spec.tileSizes.push_back(constStepOp);
438 tileLinalgOpImpl(RewriterBase &b, LinalgOp op, ArrayRef<OpFoldResult> tileSizes,
444   tileSizes = tileSizes.take_front(nLoops);
446   if (llvm::all_of(tileSizes, [](OpFoldResult ofr) {
464       b, op.getLoc(), shapeSizesToLoopsMap, allShapeSizes, tileSizes);
474       AffineMap::getMultiDimIdentityMap(tileSizes.size(), b.getContext());
552         b, loc, op, valuesToTile, getAsOpFoldResult(interchangedIvs), tileSizes,
596     ArrayRef<OpFoldResult> numThreads, ArrayRef<OpFoldResult> tileSizes,
632   if (!tileSizes.empty() && tileSizes.size() != numThreads.size())
714     if (tileSizes.empty()) {
729           b, cast<LinalgOp>(clonedOp), tileSizes, options);