Lines Matching defs:OpFoldResult
59 TileCheck(ArrayRef<OpFoldResult> tileSizes) : tileSizes(tileSizes) {}
72 ArrayRef<OpFoldResult> tileSizes;
77 static bool isTiled(AffineExpr expr, ArrayRef<OpFoldResult> tileSizes) {
86 static bool isTiled(AffineMap map, ArrayRef<OpFoldResult> tileSizes) {
220 if (llvm::any_of(padOp.getMixedLowPad(), [](OpFoldResult ofr) {
236 [](std::tuple<OpFoldResult, OpFoldResult> it) {
555 ArrayRef<OpFoldResult> tileSizes, AffineMap map,
556 ArrayRef<OpFoldResult> lbs,
557 ArrayRef<OpFoldResult> ubs,
558 ArrayRef<OpFoldResult> subShapeSizes,
568 ArrayRef<OpFoldResult> tileSizes, AffineMap map,
569 ArrayRef<OpFoldResult> lbs, ArrayRef<OpFoldResult> ubs,
570 ArrayRef<OpFoldResult> subShapeSizes,
586 OpFoldResult dim = createFoldedDimOp(builder, loc, valueToTile, r);
606 OpFoldResult offset = makeComposedFoldedAffineApply(
610 OpFoldResult closedIntSize =
614 OpFoldResult size =
661 SmallVector<OpFoldResult> maxIndices =
662 llvm::to_vector(llvm::map_range(ubs, [&](OpFoldResult ub) {
666 OpFoldResult maxIndex =
668 OpFoldResult d =
684 SmallVector<OpFoldResult> computeTileOffsets(OpBuilder &b, Location loc,
685 ArrayRef<OpFoldResult> ivs,
686 ArrayRef<OpFoldResult> tileSizes) {
687 SmallVector<OpFoldResult> offsets;
698 SmallVector<OpFoldResult> computeTileSizes(OpBuilder &b, Location loc,
699 ArrayRef<OpFoldResult> tileSizes,
700 ArrayRef<OpFoldResult> sizeBounds) {
701 SmallVector<OpFoldResult> sizes;
705 OpFoldResult size = isTiled ? tileSizes[idx] : sizeBounds[idx];
753 ValueRange valuesToTile, ArrayRef<OpFoldResult> ivs,
754 ArrayRef<OpFoldResult> tileSizes,
755 ArrayRef<OpFoldResult> sizeBounds,
759 [](OpFoldResult v) { return !isZeroIndex(v); })) &&
764 SmallVector<OpFoldResult> lbs =
766 SmallVector<OpFoldResult> subShapeSizes =
805 ArrayRef<OpFoldResult> ivs,
806 ArrayRef<OpFoldResult> tileSizes,
807 ArrayRef<OpFoldResult> sizeBounds,
826 ArrayRef<OpFoldResult> offsets) {
832 ArrayRef<OpFoldResult> offsets) {
843 OpFoldResult applied = makeComposedFoldedAffineApply(
861 getReassociationMapForFoldingUnitDims(ArrayRef<OpFoldResult> mixedSizes) {