Home
last modified time | relevance | path

Searched refs:ControlFusionFn (Results 1 – 4 of 4) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DConstantFold.cpp52 FoldConstantBase(MLIRContext *context, const ControlFusionFn &controlFn, in FoldConstantBase()
262 ControlFusionFn controlFn;
300 ControlFusionFn controlFn;
305 RewritePatternSet &patterns, const ControlFusionFn &controlFn) { in populateConstantFoldLinalgOperations()
H A DElementwiseOpFusion.cpp251 const ControlFusionFn &controlFn, in fuseElementwiseOpsImpl()
367 FuseElementwiseOps(MLIRContext *context, ControlFusionFn fun, in FuseElementwiseOps()
391 ControlFusionFn controlFn;
821 ControlFusionFn foldReshapes, in FoldWithProducerReshapeOpByExpansion()
851 ControlFusionFn controlFoldingReshapes;
860 ControlFusionFn foldReshapes, in FoldReshapeWithGenericOpByExpansion()
884 ControlFusionFn controlFoldingReshapes;
1477 ControlFusionFn controlFoldingReshapes;
1672 const ControlFusionFn &controlFoldingReshapes) { in populateFoldReshapeOpsByExpansionPatterns()
1681 const ControlFusionFn &controlFoldingReshapes) { in populateFoldReshapeOpsByCollapsingPatterns()
[all …]
/llvm-project-15.0.7/mlir/test/lib/Dialect/Linalg/
H A DTestLinalgElementwiseFusion.cpp127 linalg::ControlFusionFn controlReshapeFusionFn = in runOnOperation()
165 linalg::ControlFusionFn controlFn = [](const OpResult &producer, in runOnOperation()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Transforms/
H A DTransforms.h68 using ControlFusionFn = variable
77 const ControlFusionFn &controlElementwiseOpFusion);
83 RewritePatternSet &patterns, const ControlFusionFn &controlFoldingReshapes);
88 RewritePatternSet &patterns, const ControlFusionFn &controlFoldingReshapes);
92 const ControlFusionFn &controlFn);