Home
last modified time | relevance | path

Searched defs:tileSizes (Results 1 – 16 of 16) sorted by relevance

/llvm-project/mlir/lib/Dialect/Affine/Transforms/
H A DLoopTiling.cpp79 SmallVectorImpl<unsigned> *tileSizes) { in adjustToDivisorsOfTripCounts() argument
166 getTileSizes(ArrayRef<AffineForOp> band,SmallVectorImpl<unsigned> * tileSizes) getTileSizes() argument
252 SmallVector<unsigned, 6> tileSizes; runOnOperation() local
[all...]
/llvm-project/mlir/test/lib/Interfaces/TilingInterface/
H A DTestTilingInterfaceTransformOps.cpp60 ArrayRef<OpFoldResult> tileSizes, in applyTileAndFuseToAll() argument
147 SmallVector<int64_t> tileSizes = apply() local
224 applyTileToAll(RewriterBase & rewriter,Operation * transformOp,Range && payloadOps,ArrayRef<OpFoldResult> tileSizes,ArrayRef<int64_t> interchange,std::optional<ArrayAttr> mapping,TransformResults & transformResults) applyTileToAll() argument
271 SmallVector<int64_t> tileSizes = apply() local
349 SmallVector<int64_t> tileSizes = apply() local
[all...]
/llvm-project/mlir/lib/Dialect/SCF/Transforms/
H A DParallelLoopTiling.cpp58 mlir::scf::tileParallelLoop(ParallelOp op, ArrayRef<int64_t> tileSizes, in tileParallelLoop()
222 mlir::createParallelLoopTilingPass(ArrayRef<int64_t> tileSizes, in createParallelLoopTilingPass()
H A DTileUsingInterface.cpp38 auto tileSizes = llvm::to_vector(ts); setTileSizes() local
148 generateLoopNestUsingForOp(RewriterBase & rewriter,Location loc,ArrayRef<Range> loopRanges,ArrayRef<OpFoldResult> tileSizes,ValueRange destinationTensors,YieldTiledValuesFn yieldTiledValuesFn,SmallVector<LoopLikeOpInterface> & loops) generateLoopNestUsingForOp() argument
227 generateLoopNestUsingForallOp(RewriterBase & rewriter,Location loc,ArrayRef<Range> loopRanges,ArrayRef<OpFoldResult> tileSizes,ArrayRef<Attribute> mappingVector,ValueRange destinationTensors,YieldTiledValuesFn tiledBodyFn,SmallVector<LoopLikeOpInterface> & loops) generateLoopNestUsingForallOp() argument
292 generateLoopNest(RewriterBase & rewriter,Location loc,const scf::SCFTilingOptions & options,ArrayRef<Range> loopRanges,ArrayRef<OpFoldResult> tileSizes,ValueRange destinationTensors,YieldTiledValuesFn tiledBodyFn,SmallVector<LoopLikeOpInterface> & loops) generateLoopNest() argument
550 SmallVector<OpFoldResult> tileSizes = tileUsingSCF() local
689 tileReductionUsingScf(RewriterBase & b,PartialReductionOpInterface op,ArrayRef<OpFoldResult> tileSizes) tileReductionUsingScf() argument
[all...]
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DBubbleUpExtractSlice.cpp108 SmallVector<OpFoldResult> tileSizes = sizeBounds; in matchAndRewrite() local
H A DTiling.cpp56 SmallVector<OpFoldResult> tileSizes(allTileSizes); in makeTiledLoopRanges() local
148 __anon638b672c0102(SmallVector<int64_t> tileSizes, SmallVector<int64_t> tripCounts, int64_t range) computeStaticContinuousTileSizes() argument
599 tileToForallOpUsingTileSizes(RewriterBase & b,TilingInterface op,ArrayRef<OpFoldResult> tileSizes,std::optional<ArrayAttr> mapping) tileToForallOpUsingTileSizes() argument
622 tileLinalgOpImpl(RewriterBase & b,LinalgOp op,ArrayRef<OpFoldResult> tileSizes,const LinalgTilingOptions & options) tileLinalgOpImpl() argument
780 tileReductionUsingForall(RewriterBase & b,PartialReductionOpInterface op,ArrayRef<OpFoldResult> numThreads,ArrayRef<OpFoldResult> tileSizes,std::optional<ArrayAttr> mapping) tileReductionUsingForall() argument
[all...]
H A DFusion.cpp115 SmallVector<OpFoldResult> ivs, tileSizes, sizeBounds; in fuse() local
H A DTransforms.cpp907 tileSizeComputationFunction = [tileSizes](OpBuilder &b, Operation *op) { in setTileSizes() local
1282 SmallVector<OpFoldResult> tileSizes; matchAndRewrite() local
[all...]
H A DDataLayoutPropagation.cpp601 ArrayRef<int64_t> tileSizes) { in isDimsDivisibleByTileSizes() argument
/llvm-project/mlir/lib/Dialect/Linalg/Utils/
H A DUtils.cpp72 ArrayRef<OpFoldResult> tileSizes; member
77 static bool isTiled(AffineExpr expr, ArrayRef<OpFoldResult> tileSizes) { in isTiled() argument
86 static bool isTiled(AffineMap map, ArrayRef<OpFoldResult> tileSizes) { in isTiled() argument
591 makeTiledShape(OpBuilder & builder,Location loc,Value valueToTile,ArrayRef<OpFoldResult> tileSizes,AffineMap map,ArrayRef<OpFoldResult> lbs,ArrayRef<OpFoldResult> ubs,ArrayRef<OpFoldResult> subShapeSizes,bool omitPartialTileCheck) makeTiledShape() argument
603 computeSliceParameters(OpBuilder & builder,Location loc,Value valueToTile,ArrayRef<OpFoldResult> tileSizes,AffineMap map,ArrayRef<OpFoldResult> lbs,ArrayRef<OpFoldResult> ubs,ArrayRef<OpFoldResult> subShapeSizes,bool omitPartialTileCheck) computeSliceParameters() argument
712 computeTileOffsets(OpBuilder & b,Location loc,ArrayRef<OpFoldResult> ivs,ArrayRef<OpFoldResult> tileSizes) computeTileOffsets() argument
725 computeTileSizes(OpBuilder & b,Location loc,ArrayRef<OpFoldResult> tileSizes,ArrayRef<OpFoldResult> sizeBounds) computeTileSizes() argument
780 computeAllSliceParameters(OpBuilder & builder,Location loc,LinalgOp linalgOp,ValueRange valuesToTile,ArrayRef<OpFoldResult> ivs,ArrayRef<OpFoldResult> tileSizes,ArrayRef<OpFoldResult> sizeBounds,bool omitPartialTileCheck) computeAllSliceParameters() argument
832 makeTiledShapes(OpBuilder & builder,Location loc,LinalgOp linalgOp,ValueRange valuesToTile,ArrayRef<OpFoldResult> ivs,ArrayRef<OpFoldResult> tileSizes,ArrayRef<OpFoldResult> sizeBounds,bool omitPartialTileCheck) makeTiledShapes() argument
[all...]
/llvm-project/mlir/lib/Dialect/Tensor/Transforms/
H A DExtractSliceFromReshapeUtils.cpp108 SmallVector<Value> tileSizes; in create() local
/llvm-project/mlir/test/lib/Dialect/Linalg/
H A DTestLinalgTransforms.cpp96 ListOption<int64_t> tileSizes{ member
/llvm-project/mlir/include/mlir/Dialect/Linalg/Transforms/
H A DTransforms.h261 SmallVector<int64_t> tileSizes; member
807 SmallVector<T> tileSizes; global() member
/llvm-project/mlir/lib/Dialect/Affine/Utils/
H A DLoopUtils.cpp380 performPreTilingChecks(MutableArrayRef<AffineForOp> input,ArrayRef<t> tileSizes) performPreTilingChecks() argument
659 constructParametricallyTiledIndexSetHyperRect(MutableArrayRef<AffineForOp> origLoops,MutableArrayRef<AffineForOp> newLoops,ArrayRef<Value> tileSizes) constructParametricallyTiledIndexSetHyperRect() argument
687 constructTiledIndexSetHyperRect(MutableArrayRef<AffineForOp> origLoops,MutableArrayRef<AffineForOp> newLoops,ArrayRef<unsigned> tileSizes) constructTiledIndexSetHyperRect() argument
770 tilePerfectlyNested(MutableArrayRef<AffineForOp> input,ArrayRef<unsigned> tileSizes,SmallVectorImpl<AffineForOp> * tiledNest) tilePerfectlyNested() argument
812 tilePerfectlyNestedParametric(MutableArrayRef<AffineForOp> input,ArrayRef<Value> tileSizes,SmallVectorImpl<AffineForOp> * tiledNest) tilePerfectlyNestedParametric() argument
[all...]
/llvm-project/mlir/lib/Dialect/Linalg/TransformOps/
H A DLinalgTransformOps.cpp547 SmallVector<int64_t> tileSizes = apply() local
2168 SmallVector<OpFoldResult> tileSizes; applyToOne() local
2649 SmallVector<OpFoldResult> tileSizes = applyToOne() local
2900 ArrayRef<int64_t> tileSizes = getStaticSizes(); apply() local
3045 ArrayRef<int64_t> tileSizes = getStaticSizes(); getMixedSizes() local
[all...]
/llvm-project/mlir/lib/Dialect/SCF/Utils/
H A DUtils.cpp1237 SmallVector<Value, 4> tileSizes; extractFixedOuterLoops() local