Searched defs:first_reduce (Results 1 – 3 of 3) sorted by relevance
131 const HloInstruction* first_reduce) { in AreFusedReductionOutputsConsistent()
459 const HloInstruction* first_reduce) { in IsFusedReductionOutputConsistent()
4570 mlir::lmhlo::FusionOp fusion, mlir::mhlo::ReduceOp first_reduce) { in ComputeReductionCodegenInfo()4921 mlir::mhlo::ReduceOp first_reduce = mlir::cast<mlir::mhlo::ReduceOp>( in EmitUnnestedReduction() local