Lines Matching refs:tileSizes

62   TileCheck(ValueRange tileSizes) : tileSizes(tileSizes) {}  in TileCheck()
65 isTiled |= !isZero(tileSizes[expr.getPosition()]); in visitDimExpr()
75 ValueRange tileSizes; member
80 static bool isTiled(AffineExpr expr, ValueRange tileSizes) { in isTiled() argument
83 TileCheck t(tileSizes); in isTiled()
89 static bool isTiled(AffineMap map, ValueRange tileSizes) { in isTiled() argument
93 if (isTiled(map.getResult(r), tileSizes)) in isTiled()
800 ValueRange tileSizes, AffineMap map, ValueRange lbs, in makeTiledShape() argument
815 if (!isTiled(map.getSubMap({r}), tileSizes)) { in makeTiledShape()
932 ValueRange ivs, ValueRange tileSizes) { in computeTileOffsets() argument
934 for (unsigned idx = 0, idxIvs = 0, e = tileSizes.size(); idx < e; ++idx) { in computeTileOffsets()
936 bool isTiled = !isZero(tileSizes[idx]); in computeTileOffsets()
947 ValueRange tileSizes, in computeTileSizes() argument
950 for (unsigned idx = 0, e = tileSizes.size(); idx < e; ++idx) { in computeTileSizes()
951 bool isTiled = !isZero(tileSizes[idx]); in computeTileSizes()
953 Value size = isTiled ? tileSizes[idx] : sizeBounds[idx]; in computeTileSizes()
1007 ValueRange ivs, ValueRange tileSizes, in makeTiledShapes() argument
1011 llvm::make_range(tileSizes.begin(), tileSizes.end()), in makeTiledShapes()
1017 SmallVector<Value> lbs = computeTileOffsets(b, loc, ivs, tileSizes); in makeTiledShapes()
1019 computeTileSizes(b, loc, tileSizes, sizeBounds); in makeTiledShapes()
1035 if (!isTiled(map, tileSizes) && !linalgOp.isOutputTensor(opOperand)) { in makeTiledShapes()
1043 tiledShapes.push_back(makeTiledShape(b, loc, shapedOp, tileSizes, map, lbs, in makeTiledShapes()