| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | RuntimeOpVerification.cpp | 38 SmallVector<Range> loopRanges = linalgOp.createLoopRanges(builder, loc); in generateRuntimeVerification() local
|
| H A D | Tiling.cpp | 167 SmallVector<Range> loopRanges = op.getIterationDomain(builder); in computeContinuousTileSizes() local
291 SmallVector<OpFoldResult> loopRanges = in computeMultiTileSizes() local
369 ArrayRef<OpFoldResult> numThreads, SmallVector<Range> loopRanges, in calculateTileOffsetsAndSizes() argument
481 SmallVector<Range> loopRanges = op.getIterationDomain(b); tileToForallOpImpl() local
601 SmallVector<Range> loopRanges = op.getIterationDomain(b); tileToForallOpUsingTileSizes() local
[all...] |
| H A D | SplitReduction.cpp | 52 SmallVector<int64_t, 4> loopRanges = op.getStaticLoopRanges(); in splitReduction() local
263 SmallVector<int64_t> loopRanges = op.getStaticLoopRanges(); in splitReductionByScaling() local
|
| H A D | Fusion.cpp | 116 SmallVector<Range> loopRanges; in fuse() local
|
| H A D | Loops.cpp | 223 auto loopRanges = linalgOp.createLoopRanges(rewriter, linalgOp.getLoc()); in linalgOpToLoopsImpl() local
|
| H A D | Transforms.cpp | 862 SmallVector<Range, 4> loopRanges = packMatmulGreedily() local
|
| H A D | ElementwiseOpFusion.cpp | 1660 SmallVector<Range> loopRanges = op.createLoopRanges(rewriter, op.getLoc()); collapseOpIterationDims() local
|
| /llvm-project/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 314 doit(OpBuilder & b,Location loc,ArrayRef<Range> loopRanges,LinalgOp linalgOp,ArrayRef<utils::IteratorType> iteratorTypes,function_ref<scf::ValueVector (OpBuilder &,Location,ValueRange,ValueRange)> bodyBuilderFn,ArrayRef<linalg::ProcInfo> procInfo) doit() argument
357 doit(OpBuilder & b,Location loc,ArrayRef<Range> loopRanges,LinalgOp linalgOp,ArrayRef<utils::IteratorType> iteratorTypes,function_ref<scf::ValueVector (OpBuilder &,Location,ValueRange,ValueRange)> bodyBuilderFn,ArrayRef<linalg::ProcInfo>) doit() argument
524 doit(OpBuilder & b,Location loc,ArrayRef<Range> loopRanges,LinalgOp linalgOp,ArrayRef<utils::IteratorType> iteratorTypes,function_ref<scf::ValueVector (OpBuilder &,Location,ValueRange,ValueRange)> bodyBuilderFn,ArrayRef<linalg::ProcInfo> procInfo) doit() argument
[all...] |
| /llvm-project/mlir/lib/Dialect/SCF/Transforms/ |
| H A D | TileUsingInterface.cpp | 147 generateLoopNestUsingForOp(RewriterBase & rewriter,Location loc,ArrayRef<Range> loopRanges,ArrayRef<OpFoldResult> tileSizes,ValueRange destinationTensors,YieldTiledValuesFn yieldTiledValuesFn,SmallVector<LoopLikeOpInterface> & loops) generateLoopNestUsingForOp() argument
226 generateLoopNestUsingForallOp(RewriterBase & rewriter,Location loc,ArrayRef<Range> loopRanges,ArrayRef<OpFoldResult> tileSizes,ArrayRef<Attribute> mappingVector,ValueRange destinationTensors,YieldTiledValuesFn tiledBodyFn,SmallVector<LoopLikeOpInterface> & loops) generateLoopNestUsingForallOp() argument
291 generateLoopNest(RewriterBase & rewriter,Location loc,const scf::SCFTilingOptions & options,ArrayRef<Range> loopRanges,ArrayRef<OpFoldResult> tileSizes,ValueRange destinationTensors,YieldTiledValuesFn tiledBodyFn,SmallVector<LoopLikeOpInterface> & loops) generateLoopNest() argument
[all...] |
| /llvm-project/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorTilingInterfaceImpl.cpp | 43 SmallVector<Range> loopRanges(reifiedShapes[0].size(), {zero, one, one}); getIterationDomain() local
|