Searched refs:output_instructions (Results 1 – 1 of 1) sorted by relevance
2704 absl::Span<HloInstruction* const> output_instructions = in EmitPrologueForReduction() local2712 for (int i = 0, e = output_instructions.size(); i != e; ++i) { in EmitPrologueForReduction()2713 if (output_instructions[i]->opcode() != HloOpcode::kReduce) { in EmitPrologueForReduction()2716 HloInstruction* reduce_inst = output_instructions[i]; in EmitPrologueForReduction()2887 absl::Span<HloInstruction* const> output_instructions = in EmitTileElementForReduction() local2895 for (int i = 0, e = output_instructions.size(); i != e; ++i) { in EmitTileElementForReduction()2896 const HloInstruction* inst = output_instructions[i]; in EmitTileElementForReduction()2918 GetFirstReduceInstruction(output_instructions)->operand(0)->shape()); in EmitTileElementForReduction()3465 absl::Span<HloInstruction* const> output_instructions, in AreFusedReductionOutputsConsistent() argument3467 for (const HloInstruction* inst : output_instructions) { in AreFusedReductionOutputsConsistent()[all …]