Searched refs:reduce_or_tuple (Results 1 – 1 of 1) sorted by relevance
2701 HloInstruction* reduce_or_tuple = unnested_hlo->opcode() == HloOpcode::kFusion in EmitPrologueForReduction() local2705 GetOutputInstructions(&reduce_or_tuple); in EmitPrologueForReduction()2723 if (reduce_or_tuple->opcode() == HloOpcode::kTuple) { in EmitPrologueForReduction()2857 HloInstruction* reduce_or_tuple = unnested_hlo->opcode() == HloOpcode::kFusion in EmitTileElementForReduction() local2888 GetOutputInstructions(&reduce_or_tuple); in EmitTileElementForReduction()2898 if (reduce_or_tuple->opcode() == HloOpcode::kTuple) { in EmitTileElementForReduction()3732 HloInstruction* reduce_or_tuple = unnested_hlo->opcode() == HloOpcode::kFusion in EmitReductionToVector() local3736 GetOutputInstructions(&reduce_or_tuple); in EmitReductionToVector()3754 (output_instructions[i] == reduce_or_tuple) in EmitReductionToVector()