| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | HoistPadding.cpp | 542 RewriterBase &rewriter, IRMapping &bvm, tensor::PadOp opToHoist, in buildPackingLoopNestImpl() argument
678 buildPackingLoopNestImpl(RewriterBase & rewriter,IRMapping & bvm,tensor::PadOp opToHoist,ArrayRef<int64_t> transposeVector,const HoistPaddingAnalysis & analysis) buildPackingLoopNestImpl() argument
732 IRMapping bvm; buildPackingLoopNest() local
876 replaceByPackingResult(RewriterBase & rewriter,const IRMapping & bvm,tensor::PadOp opToHoist,RankedTensorType transposedTensorType,const HoistPaddingAnalysis & analysis,const PackingResult & packingResult) replaceByPackingResult() argument
954 IRMapping bvm; hoistPaddingOnTensors() local
[all...] |
| H A D | Vectorization.cpp | 665 vectorizeLinalgYield(RewriterBase & rewriter,Operation * op,const IRMapping & bvm,VectorizationState & state,LinalgOp linalgOp,SmallVectorImpl<Value> & newResults) vectorizeLinalgYield() argument
768 calculateGatherOffset(RewriterBase & rewriter,VectorizationState & state,tensor::ExtractOp extractOp,const IRMapping & bvm) calculateGatherOffset() argument
1010 vectorizeTensorExtract(RewriterBase & rewriter,VectorizationState & state,Operation * op,LinalgOp linalgOp,const IRMapping & bvm) vectorizeTensorExtract() argument
1135 reduceIfNeeded(OpBuilder & b,LinalgOp linalgOp,Operation * op,Value reduceValue,Value initialValue,const IRMapping & bvm) reduceIfNeeded() argument
1170 vectorizeOneOp(RewriterBase & rewriter,VectorizationState & state,LinalgOp linalgOp,Operation * op,const IRMapping & bvm,ArrayRef<CustomVectorizationHook> customVectorizationHooks) vectorizeOneOp() argument
1291 IRMapping bvm; vectorizeAsLinalgGeneric() local
1368 __anon729377361502(Operation *op, const IRMapping &bvm) vectorizeAsLinalgGeneric() argument
1375 __anon729377361602(Operation *op, const IRMapping &bvm) vectorizeAsLinalgGeneric() argument
1382 __anon729377361702(Operation *op, const IRMapping &bvm) vectorizeAsLinalgGeneric() argument
[all...] |
| H A D | Transforms.cpp | 938 padOp.getRegion().cloneInto(&generateOp.getRegion(), bvm); in createFillOrGenerateOp() local
|
| /llvm-project/mlir/lib/Dialect/Utils/ |
| H A D | StructuredOpsUtils.cpp | 199 IRMapping bvm; in clone() local
|
| /llvm-project/mlir/lib/Dialect/Shape/Transforms/ |
| H A D | OutlineShapeComputation.cpp | 75 IRMapping bvm; in createFuncFromCluster() local
|
| /llvm-project/mlir/lib/Dialect/SCF/Utils/ |
| H A D | Utils.cpp | 194 IRMapping bvm; outlineSingleBlockRegion() local
210 IRMapping bvm; outlineSingleBlockRegion() local
|
| /llvm-project/mlir/lib/Dialect/GPU/TransformOps/ |
| H A D | GPUTransformOps.cpp | 485 IRMapping bvm; in rewriteOneForallCommonImpl() local
|
| /llvm-project/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorTilingInterfaceImpl.cpp | 748 IRMapping bvm; bubbleUpPadSlice() local
|
| /llvm-project/mlir/lib/Dialect/Linalg/TransformOps/ |
| H A D | LinalgTransformOps.cpp | 838 IRMapping bvm; tileAndFuseFirstExtractUseThroughContainingOpBlockArgument() local
|
| /llvm-project/mlir/lib/Dialect/Vector/IR/ |
| H A D | VectorOps.cpp | 1181 IRMapping bvm; matchAndRewrite() local
|