Home
last modified time | relevance | path

Searched refs:distributedReductionFn (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/include/mlir/Dialect/Vector/Transforms/
H A DVectorDistribution.h80 DistributedReductionFn distributedReductionFn);
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorDistribute.cpp863 DistributedReductionFn distributedReductionFn, in WarpOpReduction()
866 distributedReductionFn(distributedReductionFn) {} in WarpOpReduction()
914 distributedReductionFn(reductionOp.getLoc(), rewriter, perLaneReduction, in matchAndRewrite()
926 DistributedReductionFn distributedReductionFn; member
951 DistributedReductionFn distributedReductionFn) { in populateDistributeReduction() argument
952 patterns.add<WarpOpReduction>(patterns.getContext(), distributedReductionFn); in populateDistributeReduction()