Searched defs:loopRanges (Results 1 – 8 of 8) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Tiling.cpp | 130 SmallVector<Value, 4> loopRanges = in computeMultiTileSizes() local
225 SmallVector<Range> loopRanges = op.getIterationDomain(b); in tileToForeachThreadOpImpl() local
350 SmallVector<Range> loopRanges = op.getIterationDomain(b); in tileToForeachThreadOpUsingTileSizes() local
404 SmallVector<Range, 4> loopRanges; in tileLinalgOpImpl() local
|
| H A D | SplitReduction.cpp | 97 SmallVector<int64_t, 4> loopRanges = op.getStaticLoopRanges(); in splitReduction() local
290 SmallVector<int64_t> loopRanges = op.getStaticLoopRanges(); in splitReductionByScaling() local
|
| H A D | Fusion.cpp | 117 SmallVector<Range, 8> loopRanges; in fuse() local
|
| H A D | Loops.cpp | 211 auto loopRanges = linalgOp.createLoopRanges(rewriter, linalgOp.getLoc()); in linalgOpToLoopsImpl() local
|
| H A D | ElementwiseOpFusion.cpp | 1393 SmallVector<Range> loopRanges = in collapseGenericOpIterationDims() local
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorTilingInterfaceImpl.cpp | 55 SmallVector<Range> loopRanges(reifiedShapes[0].size(), {zero, one, one}); in getIterationDomain() local
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/ |
| H A D | TileUsingInterface.cpp | 100 ArrayRef<Range> loopRanges, ArrayRef<Value> tileSizeVals, in generateTileLoopNest()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 501 OpBuilder &b, Location loc, ArrayRef<Range> loopRanges, LinalgOp linalgOp, in doit()
561 OpBuilder &b, Location loc, ArrayRef<Range> loopRanges, LinalgOp linalgOp, in doit()
726 OpBuilder &b, Location loc, ArrayRef<Range> loopRanges, LinalgOp linalgOp, in doit()
|