Searched defs:first_reduce (Results 1 – 3 of 3) sorted by relevance
237 const HloInstruction* first_reduce) { in AreFusedReductionOutputsConsistent()
595 const HloInstruction* first_reduce) { in IsFusedReductionOutputConsistent()614 mlir::mhlo::ReduceOp first_reduce) { in IsFusedReductionOutputConsistent()
4222 mlir::Operation* first_reduce = nullptr; in EmitPrologueForReduction() local5295 mlir::Operation* unnested_hlo, mlir::Operation* first_reduce) { in ComputeReductionCodegenInfo()5569 mlir::Operation* first_reduce = nullptr; in EmitReductionFromOrToContiguousDimensions() local