Searched defs:reduceOp (Results 1 – 8 of 8) sorted by relevance
339 gpu::AllReduceOp reduceOp; member359 auto callback = [&](gpu::AllReduceOp reduceOp) -> WalkResult { in matchAndRewrite()371 for (gpu::AllReduceOp reduceOp : reduceOps) in matchAndRewrite() local
163 Operation *reduceOp = thenBlock.getTerminator(); in tileParallelLoop() local
248 LogicalResult ReductionTreePass::reduceOp(ModuleOp module, Region ®ion) { in reduceOp() function in ReductionTreePass
250 auto reduceOp = rewriter.replaceOpWithNewOp<scf::ReduceOp>( in matchAndRewrite() local
571 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...]
357 extractAllElements(vector::ReductionOp reduceOp,vector::ReductionOp::Adaptor adaptor,VectorType srcVectorType,ConversionPatternRewriter & rewriter) extractAllElements() argument [all...]
2895 auto reduceOp = verifyAndGetTerminator<scf::ReduceOp>( verify() local 3079 auto reduceOp = cast<ReduceOp>(op.getBody()->getTerminator()); matchAndRewrite() local
1807 matchAndRewrite(SrcOp reduceOp,PatternRewriter & rewriter) const matchAndRewrite() argument