Searched refs:controlSplitReductionFn (Results 1 – 2 of 2) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | SplitReduction.cpp | 63 const ControlSplitReductionFn &controlSplitReductionFn, in splitReduction() argument
71 splitReduction(b, op, controlSplitReductionFn, useAlloc); in splitReduction()
83 const ControlSplitReductionFn &controlSplitReductionFn, bool useAlloc) { in splitReduction() argument
87 std::pair<int64_t, unsigned> control = controlSplitReductionFn(op); in splitReduction()
273 const ControlSplitReductionFn &controlSplitReductionFn, bool useAlloc) { in splitReductionByScaling() argument
278 std::pair<int64_t, unsigned> control = controlSplitReductionFn(op); in splitReductionByScaling()
454 ControlSplitReductionFn controlSplitReductionFn, in LinalgSplitReduction()
458 controlSplitReductionFn(std::move(controlSplitReductionFn)), in LinalgSplitReduction()
463 return splitReduction(rewriter, op, controlSplitReductionFn, filter, in matchAndRewrite()
468 ControlSplitReductionFn controlSplitReductionFn; member
[all …]
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Transforms/ |
| H A D | Transforms.h | 1400 const ControlSplitReductionFn &controlSplitReductionFn,
1445 const ControlSplitReductionFn &controlSplitReductionFn,
1459 const ControlSplitReductionFn &controlSplitReductionFn,
1510 const ControlSplitReductionFn &controlSplitReductionFn,
|