Searched refs:allShapeSizes (Results 1 – 1 of 1) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Tiling.cpp | 59 ValueRange allShapeSizes, ValueRange allTileSizes) { in makeTiledLoopRanges() argument
62 auto shapeSizes = applyMapToValues(b, loc, map, allShapeSizes); in makeTiledLoopRanges()
224 ValueRange tileSizes, ValueRange allShapeSizes) { in makeTiledShapes() argument
233 auto shapeSizes = applyMapToValues(b, loc, map, allShapeSizes); in makeTiledShapes()
335 auto allShapeSizes = op.createFlatListOfOperandDims(b, op.getLoc()); in tileLinalgOpImpl() local
343 b, op.getLoc(), shapeSizesToLoopsMap, allShapeSizes, tileSizes); in tileLinalgOpImpl()
404 interchangedIvs, tileSizes, allShapeSizes); in tileLinalgOpImpl()
|