Lines Matching refs:tileSizes

52   SmallVector<Value, 4> tileSizes(allTileSizes.begin(), allTileSizes.end());  in makeTiledLoopRanges()  local
56 for (int idx = 0, e = tileSizes.size(), zerosCount = 0; idx < e; ++idx) { in makeTiledLoopRanges()
57 if (isZero(tileSizes[idx - zerosCount])) { in makeTiledLoopRanges()
59 tileSizes.erase(tileSizes.begin() + idx - zerosCount); in makeTiledLoopRanges()
68 for (unsigned idx = 0, e = tileSizes.size(); idx < e; ++idx) in makeTiledLoopRanges()
70 shapeSizes[idx], tileSizes[idx]}); in makeTiledLoopRanges()
348 RewriterBase &b, TilingInterface op, ArrayRef<OpFoldResult> tileSizes, in tileToForeachThreadOpUsingTileSizes() argument
357 for (const auto &it : llvm::zip(tileSizes, loopRanges)) { in tileToForeachThreadOpUsingTileSizes()
365 /*nominalTileSizes=*/tileSizes, in tileToForeachThreadOpUsingTileSizes()
384 tileLinalgOpImpl(RewriterBase &b, LinalgOp op, ValueRange tileSizes, in tileLinalgOpImpl() argument
388 tileSizes = tileSizes.take_front(nLoops); in tileLinalgOpImpl()
390 if (llvm::all_of(tileSizes, isZero)) { in tileLinalgOpImpl()
407 b, op.getLoc(), shapeSizesToLoopsMap, allShapeSizes, tileSizes); in tileLinalgOpImpl()
418 AffineMap::getMultiDimIdentityMap(tileSizes.size(), b.getContext()); in tileLinalgOpImpl()
467 makeTiledShapes(b, loc, op, valuesToTile, interchangedIvs, tileSizes, in tileLinalgOpImpl()
559 SmallVector<Value> tileSizes = in tilePadOp() local
563 tileSizes.append(rank - tileSizes.size(), zero); in tilePadOp()
571 if (!isZero(tileSizes[i])) { in tilePadOp()
574 steps.push_back(tileSizes[i]); in tilePadOp()
586 computeTileOffsets(b, loc, localIvs, tileSizes); in tilePadOp()
587 SmallVector<Value> sizes = computeTileSizes(b, loc, tileSizes, allDims); in tilePadOp()
593 b, loc, newPadOp->getResult(0), tileSizes, map, offsets, allDims, in tilePadOp()
719 LinalgTilingPass(ArrayRef<int64_t> tileSizes, LinalgTilingLoopType loopType) { in LinalgTilingPass()
720 this->tileSizes = tileSizes; in LinalgTilingPass()
734 LinalgTilingOptions().setTileSizes(tileSizes).setLoopType(type); in runOnOperation()
758 mlir::createLinalgTilingPass(ArrayRef<int64_t> tileSizes, in createLinalgTilingPass() argument
760 return std::make_unique<LinalgTilingPass>(tileSizes, loopType); in createLinalgTilingPass()