Home
last modified time | relevance | path

Searched defs:reduceOp (Results 1 – 8 of 8) sorted by relevance

/llvm-project/mlir/lib/Dialect/GPU/Transforms/
H A DAllReduceLowering.cpp339 gpu::AllReduceOp reduceOp; member
359 auto callback = [&](gpu::AllReduceOp reduceOp) -> WalkResult { in matchAndRewrite()
371 for (gpu::AllReduceOp reduceOp : reduceOps) in matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/SCF/Transforms/
H A DParallelLoopTiling.cpp163 Operation *reduceOp = thenBlock.getTerminator(); in tileParallelLoop() local
/llvm-project/mlir/lib/Reducer/
H A DReductionTreePass.cpp248 LogicalResult ReductionTreePass::reduceOp(ModuleOp module, Region &region) { in reduceOp() function in ReductionTreePass
/llvm-project/mlir/lib/Conversion/AffineToStandard/
H A DAffineToStandard.cpp250 auto reduceOp = rewriter.replaceOpWithNewOp<scf::ReduceOp>( in matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DVectorization.cpp571 buildMultiDimReduce(OpBuilder & b,Operation * reduceOp,Value valueToReduce,Value acc,ArrayRef<bool> dimsToMask) buildMultiDimReduce() argument
1210 Operation *reduceOp = vectorizeOneOp() local
1733 Operation *reduceOp = matchLinalgReduction(&opOperand); reductionPreconditions() local
2848 Operation *reduceOp = matchLinalgReduction(linalgOp.getDpsInitOperand(0)); Conv1DGenerator() local
[all...]
/llvm-project/mlir/lib/Conversion/VectorToSPIRV/
H A DVectorToSPIRV.cpp357 extractAllElements(vector::ReductionOp reduceOp,vector::ReductionOp::Adaptor adaptor,VectorType srcVectorType,ConversionPatternRewriter & rewriter) extractAllElements() argument
[all...]
/llvm-project/mlir/lib/Dialect/SCF/IR/
H A DSCF.cpp2895 auto reduceOp = verifyAndGetTerminator<scf::ReduceOp>( verify() local
3079 auto reduceOp = cast<ReduceOp>(op.getBody()->getTerminator()); matchAndRewrite() local
/llvm-project/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalg.cpp1807 matchAndRewrite(SrcOp reduceOp,PatternRewriter & rewriter) const matchAndRewrite() argument