Searched refs:controlFn (Results 1 – 8 of 8) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | ConstantFold.cpp | 52 FoldConstantBase(MLIRContext *context, const ControlFusionFn &controlFn, in FoldConstantBase() argument
54 : OpRewritePattern<GenericOp>(context, benefit), controlFn(controlFn) {} in FoldConstantBase()
128 if (!controlFn(producer, *consumer)) in matchAndRewrite()
262 ControlFusionFn controlFn; member in __anonb7b609f10111::FoldConstantBase
300 ControlFusionFn controlFn; member
305 RewritePatternSet &patterns, const ControlFusionFn &controlFn) { in populateConstantFoldLinalgOperations() argument
307 patterns.insert<FoldConstantTranspose>(context, controlFn); in populateConstantFoldLinalgOperations()
|
| H A D | ElementwiseOpFusion.cpp | 251 const ControlFusionFn &controlFn, in fuseElementwiseOpsImpl() argument
255 !controlFn(producer->getResult(0), *consumerOpOperand)) in fuseElementwiseOpsImpl()
355 GenericOp producer, const ControlFusionFn &controlFn) { in fuseElementwiseOps() argument
359 return fuseElementwiseOpsImpl(producer, consumerOpOperand, controlFn, in fuseElementwiseOps()
370 controlFn(std::move(fun)) {} in FuseElementwiseOps()
381 fuseElementwiseOps(rewriter, opOperand, producer, controlFn); in matchAndRewrite()
391 ControlFusionFn controlFn; member in __anon750e3d170311::FuseElementwiseOps
|
| H A D | Transforms.cpp | 824 if (controlFn) { in matchAndRewrite()
825 if (Optional<bool> control = controlFn(sliceOp)) in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/test/lib/Dialect/Tensor/ |
| H A D | TestTensorTransforms.cpp | 63 tensor::ControlConstantExtractSliceFusionFn controlFn = in applyFoldConstantExtractSlicePatterns() local
73 tensor::populateFoldConstantExtractSlicePatterns(patterns, controlFn); in applyFoldConstantExtractSlicePatterns()
|
| /llvm-project-15.0.7/mlir/test/lib/Dialect/Linalg/ |
| H A D | TestLinalgElementwiseFusion.cpp | 165 linalg::ControlFusionFn controlFn = [](const OpResult &producer, in runOnOperation() local
173 linalg::populateFoldReshapeOpsByCollapsingPatterns(patterns, controlFn); in runOnOperation()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Tensor/IR/ |
| H A D | Tensor.h | 133 const ControlConstantExtractSliceFusionFn &controlFn =
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Transforms/ |
| H A D | Transforms.h | 92 const ControlFusionFn &controlFn);
1328 ControlFn controlFn = nullptr,
1330 : OpRewritePattern(context, benefit), controlFn(std::move(controlFn)) {} in OpRewritePattern()
1336 ControlFn controlFn;
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorOps.cpp | 1263 ControlConstantExtractSliceFusionFn controlFn) in ConstantOpExtractSliceFolder() argument
1265 controlFn(std::move(controlFn)) {} in ConstantOpExtractSliceFolder()
1284 if (!controlFn(op)) in matchAndRewrite()
1339 ControlConstantExtractSliceFusionFn controlFn; member in __anon3fb9f79f0a11::ConstantOpExtractSliceFolder
1346 const ControlConstantExtractSliceFusionFn &controlFn) { in populateFoldConstantExtractSlicePatterns() argument
1347 patterns.add<ConstantOpExtractSliceFolder>(patterns.getContext(), controlFn); in populateFoldConstantExtractSlicePatterns()
|