Lines Matching defs:tileSizes

52 // checking whether the corresponding `tileSizes[pos]` is non-zero.
59   TileCheck(ArrayRef<OpFoldResult> tileSizes) : tileSizes(tileSizes) {}
62     isTiled |= !isZeroIndex(tileSizes[expr.getPosition()]);
72   ArrayRef<OpFoldResult> tileSizes;
77 static bool isTiled(AffineExpr expr, ArrayRef<OpFoldResult> tileSizes) {
80   TileCheck t(tileSizes);
86 static bool isTiled(AffineMap map, ArrayRef<OpFoldResult> tileSizes) {
90     if (isTiled(map.getResult(r), tileSizes))
555                           ArrayRef<OpFoldResult> tileSizes, AffineMap map,
561       computeSliceParameters(builder, loc, valueToTile, tileSizes, map, lbs,
568                        ArrayRef<OpFoldResult> tileSizes, AffineMap map,
584     if (!isTiled(map.getSubMap({r}), tileSizes)) {
686                                              ArrayRef<OpFoldResult> tileSizes) {
688   for (unsigned idx = 0, idxIvs = 0, e = tileSizes.size(); idx < e; ++idx) {
690     bool isTiled = !isZeroIndex(tileSizes[idx]);
699                                            ArrayRef<OpFoldResult> tileSizes,
702   for (unsigned idx = 0, e = tileSizes.size(); idx < e; ++idx) {
703     bool isTiled = !isZeroIndex(tileSizes[idx]);
705     OpFoldResult size = isTiled ? tileSizes[idx] : sizeBounds[idx];
754                           ArrayRef<OpFoldResult> tileSizes,
758                            llvm::make_range(tileSizes.begin(), tileSizes.end()),
765       computeTileOffsets(builder, loc, ivs, tileSizes);
767       computeTileSizes(builder, loc, tileSizes, sizeBounds);
786     if (!isTiled(map, tileSizes) && !(isa<RankedTensorType>(operandType) &&
796         builder, loc, shapedOp, tileSizes, map, lbs, sizeBounds, subShapeSizes,
806                                    ArrayRef<OpFoldResult> tileSizes,
811                                 tileSizes, sizeBounds, omitPartialTileCheck);