Searched defs:reduceOp (Results 1 – 3 of 3) sorted by relevance
70 void convertToReductionOperator(Location loc, scf::ReduceOp reduceOp, in convertToReductionOperator()199 lmhlo::ReduceOp reduceOp, OpAdaptor /*adaptor*/, in matchAndRewrite()233 lmhlo::ReduceOp reduceOp, ConversionPatternRewriter* rewriter) const { in createReduceOpInNestedParallelLoops()379 scf::ReduceOp reduceOp = createReduceOpInNestedParallelLoops( in matchAndRewrite() local
55 ReduceOp reduceOp, OpAdaptor /*adaptor*/, in matchAndRewrite()
345 auto reduceOp = op.getDefiningOp<mhlo::ReduceOp>(); in backwardReduceShape() local352 auto reduceOp = op.getDefiningOp<mhlo::ReduceOp>(); in forwardReduceShape() local