Searched refs:allShapeSizes (Results 1 – 5 of 5) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | BubbleUpExtractSlice.cpp | 91 auto allShapeSizes = in matchAndRewrite() local
99 shapeSizesToLoopsMap, allShapeSizes); in matchAndRewrite()
|
| H A D | Tiling.cpp | 47 ValueRange allShapeSizes, in makeTiledLoopRanges() argument
51 auto shapeSizes = applyMapToValues(b, loc, map, allShapeSizes); in makeTiledLoopRanges()
399 auto allShapeSizes = op.createFlatListOfOperandDims(b, op.getLoc()); in tileLinalgOpImpl() local
407 b, op.getLoc(), shapeSizesToLoopsMap, allShapeSizes, tileSizes); in tileLinalgOpImpl()
465 applyMapToValues(b, loc, shapeSizesToLoopsMap, allShapeSizes); in tileLinalgOpImpl()
|
| H A D | Transforms.cpp | 145 auto allShapeSizes = linalgOp.createFlatListOfOperandDims(b, loc); in scalarizeDynamicDims() local
149 auto shapeSizes = applyMapToValues(b, loc, map, allShapeSizes); in scalarizeDynamicDims()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgInterfaces.cpp | 529 SmallVector<int64_t, 4> allShapeSizes = createFlatListOfOperandStaticDims(); in computeStaticLoopSizes() local
534 res[d.getPosition()] = allShapeSizes[idx]; in computeStaticLoopSizes()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Transforms/ |
| H A D | Transforms.h | 411 ValueRange allShapeSizes, ValueRange allTileSizes);
|