Lines Matching refs:controlSplitReductionFn
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
477 const ControlSplitReductionFn &controlSplitReductionFn, in populateSplitReductionPattern() argument
480 controlSplitReductionFn, f, useAlloc); in populateSplitReductionPattern()