| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorInferTypeOpInterfaceImpl.cpp | 55 return applyMapToValues(builder, loc, in getCollapsedOutputDimFromInputShape()
103 return applyMapToValues( in getExpandedOutputDimFromInputShape()
196 shapes.push_back(applyMapToValues( in reifyResultShapes()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Split.cpp | 58 offsets = applyMapToValues(b, loc, indexing, offsets); in getOperandSlices()
109 applyMapToValues(rewriter, op.getLoc(), shapesToLoops, allShapes); in splitOp()
|
| H A D | BubbleUpExtractSlice.cpp | 98 auto sizeBounds = applyMapToValues(rewriter, linalgLoc, in matchAndRewrite()
|
| H A D | Tiling.cpp | 51 auto shapeSizes = applyMapToValues(b, loc, map, allShapeSizes); in makeTiledLoopRanges()
131 applyMapToValues(b, op.getLoc(), shapesToLoops, allShapes); in computeMultiTileSizes()
454 interchangedIvs = applyMapToValues(b, loc, invPermutationMap, ivs); in tileLinalgOpImpl()
465 applyMapToValues(b, loc, shapeSizesToLoopsMap, allShapeSizes); in tileLinalgOpImpl()
|
| H A D | TilingInterfaceImpl.cpp | 58 applyMapToValues(b, loc, map, allShapesSizes), [&](Value v) { in getIterationDomain()
|
| H A D | DecomposeLinalgOps.cpp | 112 return getAsOpFoldResult(applyMapToValues(b, loc, map, allShapesSizes)); in getGenericOpLoopRange()
|
| H A D | Transforms.cpp | 149 auto shapeSizes = applyMapToValues(b, loc, map, allShapeSizes); in scalarizeDynamicDims()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/IR/ |
| H A D | AffineOps.h | 409 SmallVector<Value, 4> applyMapToValues(OpBuilder &b, Location loc,
414 SmallVector<OpFoldResult> applyMapToValues(RewriterBase &b, Location loc,
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 829 auto offset = applyMapToValues(builder, loc, m, lbs).front(); in makeTiledShape()
832 applyMapToValues(builder, loc, m, subShapeSizes).front(); in makeTiledShape()
884 Value maxIndex = applyMapToValues(builder, loc, m, maxIndices).front(); in makeTiledShape()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgInterfaces.cpp | 595 applyMapToValues(b, loc, resultShapesFromInputShapesMap, in reifyResultShapes()
|
| H A D | LinalgOps.cpp | 638 applyMapToValues(rewriter, loc, addMap, {offsetValue, padValue})[0]); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/ |
| H A D | AffineOps.cpp | 884 SmallVector<Value, 4> mlir::applyMapToValues(OpBuilder &b, Location loc, in applyMapToValues() function in mlir
900 mlir::applyMapToValues(RewriterBase &b, Location loc, AffineMap map, in applyMapToValues() function in mlir
|