Home
last modified time | relevance | path

Searched defs:controlFn (Results 1 – 11 of 11) sorted by relevance

/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DBlockPackMatmul.cpp233 ControlBlockPackMatmulFn controlFn; member
275 ControlBlockPackMatmulFn controlFn; member
287 ControlBlockPackMatmulFn controlFn = in runOnOperation() local
311 RewritePatternSet &patterns, const ControlBlockPackMatmulFn &controlFn) { in populateBlockPackMatmulPatterns() argument
[all...]
H A DConstantFold.cpp53 FoldConstantBase(MLIRContext *context, const ControlFusionFn &controlFn, in FoldConstantBase()
262 ControlFusionFn controlFn; member in __anond55d2d340111::FoldConstantBase
302 ControlFusionFn controlFn; member
307 RewritePatternSet &patterns, const ControlFusionFn &controlFn) { in populateConstantFoldLinalgOperations()
H A DDataLayoutPropagation.cpp375 const ControlPropagationFn &controlFn) { in bubbleUpPackOpThroughGenericOp() argument
473 ControlPropagationFn controlFn; member
566 ControlPropagationFn controlFn; member in __anon32e33ec10111::BubbleUpPackThroughPadOp
872 ControlPropagationFn controlFn; global() member in __anon32e33ec10111::BubbleUpPackOpThroughReshapeOp
896 pushDownUnPackOpThroughExpandShape(tensor::UnPackOp unPackOp,tensor::ExpandShapeOp expandOp,PatternRewriter & rewriter,ControlPropagationFn controlFn) pushDownUnPackOpThroughExpandShape() argument
996 ControlPropagationFn controlFn; global() member in __anon32e33ec10111::PushDownUnPackOpThroughReshapeOp
1053 pushDownUnPackOpThroughGenericOp(RewriterBase & rewriter,GenericOp genericOp,ControlPropagationFn controlFn) pushDownUnPackOpThroughGenericOp() argument
1150 ControlPropagationFn controlFn; global() member
1214 ControlPropagationFn controlFn; global() member
[all...]
H A DElementwiseOpFusion.cpp455 ControlFusionFn controlFn; global() member in __anona62ab6fa0511::FuseElementwiseOps
/llvm-project/mlir/lib/Dialect/Tensor/Transforms/
H A DRewriteAsConstant.cpp206 ControlFoldFn controlFn; member
212 RewritePatternSet &patterns, const ControlFoldFn &controlFn) { in populateRewriteAsConstantPatterns()
/llvm-project/mlir/lib/Dialect/Vector/Transforms/
H A DVectorInsertExtractStridedSliceRewritePatterns.cpp242 Convert1DExtractStridedSliceIntoExtractInsertChain(MLIRContext * context,std::function<bool (ExtractStridedSliceOp)> controlFn,PatternBenefit benefit) Convert1DExtractStridedSliceIntoExtractInsertChain() argument
277 std::function<bool(ExtractStridedSliceOp)> controlFn; global() member in Convert1DExtractStridedSliceIntoExtractInsertChain
341 populateVectorExtractStridedSliceToExtractInsertChainPatterns(RewritePatternSet & patterns,std::function<bool (ExtractStridedSliceOp)> controlFn,PatternBenefit benefit) populateVectorExtractStridedSliceToExtractInsertChainPatterns() argument
H A DVectorTransforms.cpp955 std::function<bool(BitCastOp)> controlFn; global() member
1931 populateBreakDownVectorBitCastOpPatterns(RewritePatternSet & patterns,std::function<bool (vector::BitCastOp)> controlFn,PatternBenefit benefit) populateBreakDownVectorBitCastOpPatterns() argument
/llvm-project/mlir/test/lib/Dialect/Linalg/
H A DTestLinalgElementwiseFusion.cpp156 auto controlFn = [](OpOperand *operand) { return true; }; in runOnOperation() local
233 linalg::ControlFusionFn controlFn = [](OpOperand *fusedOperand) -> bool { runOnOperation() local
/llvm-project/mlir/test/lib/Interfaces/TilingInterface/
H A DTestTilingInterfaceTransformOps.cpp92 scf::SCFTileAndFuseOptions::ControlFnTy controlFn = in applyTileAndFuseToAll() local
/llvm-project/mlir/test/lib/Dialect/Tensor/
H A DTestTensorTransforms.cpp113 tensor::ControlConstantExtractSliceFusionFn controlFn = applyFoldConstantExtractSlicePatterns() local
/llvm-project/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp2307 ConstantOpExtractSliceFolder(MLIRContext * context,ControlConstantExtractSliceFusionFn controlFn) ConstantOpExtractSliceFolder() argument
2383 ControlConstantExtractSliceFusionFn controlFn; global() member in __anon6f30b6c21411::ConstantOpExtractSliceFolder
2390 populateFoldConstantExtractSlicePatterns(RewritePatternSet & patterns,const ControlConstantExtractSliceFusionFn & controlFn) populateFoldConstantExtractSlicePatterns() argument