|
FailureOr< SCFTilingResult > | mlir::scf::tileUsingSCF (RewriterBase &rewriter, TilingInterface op, const SCFTilingOptions &options) |
| Method to tile an op that implements the TilingInterface using scf.for for iterating over the tiles. More...
|
|
std::optional< SCFFuseProducerOfSliceResult > | mlir::scf::tileAndFuseProducerOfSlice (RewriterBase &rewriter, tensor::ExtractSliceOp candidateSliceOp, MutableArrayRef< LoopLikeOpInterface > loops) |
| Implementation of fusing producer of a single slice by computing the slice of the producer in-place. More...
|
|
FailureOr< SmallVector< Operation * > > | mlir::scf::yieldReplacementForFusedProducer (RewriterBase &rewriter, tensor::ExtractSliceOp sliceOp, scf::SCFFuseProducerOfSliceResult fusedProducerInfo, MutableArrayRef< LoopLikeOpInterface > loops, ArrayRef< unsigned > yieldResultNumber=ArrayRef< unsigned >{}) |
| Reconstruct the fused producer from within the tiled-and-fused code. More...
|
|
FailureOr< SCFTileAndFuseResult > | mlir::scf::tileConsumerAndFuseProducersUsingSCF (RewriterBase &rewriter, TilingInterface consumer, const SCFTileAndFuseOptions &options) |
| Method to tile and fuse a sequence of operations, by tiling the consumer and fusing its producers. More...
|
|
FailureOr< scf::SCFFuseConsumerOfSliceResult > | mlir::scf::tileAndFuseConsumerOfSlice (RewriterBase &rewriter, Operation *candidateSliceOp) |
| Implementation of fusing consumer of a single slice by computing the slice of the consumer in-place for scf loop. More...
|
|
FailureOr< SmallVector< scf::ForOp > > | mlir::scf::lowerToLoopsUsingSCFForOp (RewriterBase &rewriter, TilingInterface op) |
| Method to lower an op that implements the TilingInterface to loops/scalars. More...
|
|
FailureOr< scf::SCFTilingResult > | mlir::scf::tileReductionUsingScf (RewriterBase &b, PartialReductionOpInterface op, ArrayRef< OpFoldResult > tileSize) |
| Method to tile a reduction and generate a parallel op within a serial loop. More...
|
|