Searched defs:first_reduce (Results 1 – 3 of 3) sorted by relevance
269 const HloInstruction* first_reduce) { in AreFusedReductionOutputsConsistent()
680 const HloInstruction* first_reduce) { in IsFusedReductionOutputConsistent()699 mlir::mhlo::ReduceOp inst, mlir::mhlo::ReduceOp first_reduce, in IsFusedReductionOutputConsistent()
3728 mlir::Operation* first_reduce = nullptr; in EmitPrologueForReduction() local4824 mlir::Operation* unnested_hlo, mlir::Operation* first_reduce, in ComputeReductionCodegenInfo()5062 mlir::Operation* first_reduce = nullptr; in EmitUnnestedReduction() local