Searched refs:instruction_to_merge (Results 1 – 4 of 4) sorted by relevance
1920 HloFusionInstruction* instruction_to_merge) { in MergeFusionInstruction() argument1921 CHECK(absl::c_linear_search(operands(), instruction_to_merge)); in MergeFusionInstruction()1923 std::unique_ptr<HloInstruction> cloned = instruction_to_merge->Clone(); in MergeFusionInstruction()1950 ? instruction_to_merge->mutable_operand( in MergeFusionInstruction()1951 instruction_to_merge->fused_instructions_computation() in MergeFusionInstruction()1958 TF_CHECK_OK(instruction_to_merge->ReplaceUseWith(this, unfused_root)); in MergeFusionInstruction()1985 HloFusionInstruction* instruction_to_merge) { in MergeFusionInstructionIntoMultiOutput() argument1995 instruction_to_merge->fused_instructions_computation(); in MergeFusionInstructionIntoMultiOutput()2001 instruction_to_merge->mutable_operand( in MergeFusionInstructionIntoMultiOutput()2027 ? instruction_to_merge->mutable_operand( in MergeFusionInstructionIntoMultiOutput()[all …]
1934 void MergeFusionInstruction(HloInstruction* instruction_to_merge);1938 HloInstruction* instruction_to_merge);
1236 void MergeFusionInstruction(HloFusionInstruction* instruction_to_merge);1244 HloFusionInstruction* instruction_to_merge);
4561 HloInstruction* instruction_to_merge) { in MergeFusionInstruction() argument4563 Cast<HloFusionInstruction>(instruction_to_merge)); in MergeFusionInstruction()4568 HloInstruction* instruction_to_merge) { in MergeFusionInstructionIntoMultiOutput() argument4571 Cast<HloFusionInstruction>(instruction_to_merge)); in MergeFusionInstructionIntoMultiOutput()