Searched refs:reduce_hlo (Results 1 – 1 of 1) sorted by relevance
4266 const HloInstruction* reduce_hlo = fused_computation->root_instruction(); in EmitPrologueForReduction() local4267 if (reduce_hlo->opcode() == HloOpcode::kTuple) { in EmitPrologueForReduction()4268 reduce_hlo = reduce_hlo->operand(index); in EmitPrologueForReduction()4270 const HloInstruction* init_value = reduce_hlo->operand(1); in EmitPrologueForReduction()4419 mlir::Operation* reduce_hlo = in EmitEpilogueForReduction() local4421 if (!IsReductionFromOrToContiguousDimensions(reduce_hlo)) { in EmitEpilogueForReduction()4425 auto operand_shape = TypeToShape(reduce_hlo->getOperand(0).getType()); in EmitEpilogueForReduction()4429 reduce_hlo->getAttrOfType<mlir::DenseIntElementsAttr>( in EmitEpilogueForReduction()