Searched refs:instruction_to_merge (Results 1 – 4 of 4) sorted by relevance
1232 HloFusionInstruction* instruction_to_merge) { in MergeFusionInstruction() argument1233 CHECK(absl::c_linear_search(operands(), instruction_to_merge)); in MergeFusionInstruction()1235 std::unique_ptr<HloInstruction> cloned = instruction_to_merge->Clone(); in MergeFusionInstruction()1259 instruction_to_merge->ReplaceUseWith(this, unfused_instructions.front())); in MergeFusionInstruction()1270 HloFusionInstruction* instruction_to_merge) { in MergeFusionInstructionIntoMultiOutput() argument1280 instruction_to_merge->fused_instructions_computation(); in MergeFusionInstructionIntoMultiOutput()1286 instruction_to_merge->mutable_operand( in MergeFusionInstructionIntoMultiOutput()1308 TF_CHECK_OK(instruction_to_merge->ReplaceAllUsesWith(unfused_root)); in MergeFusionInstructionIntoMultiOutput()1311 instruction_to_merge->parent()->RemoveInstruction(instruction_to_merge)); in MergeFusionInstructionIntoMultiOutput()
1438 void MergeFusionInstruction(HloInstruction* instruction_to_merge);1442 HloInstruction* instruction_to_merge);
725 void MergeFusionInstruction(HloFusionInstruction* instruction_to_merge);733 HloFusionInstruction* instruction_to_merge);
3294 HloInstruction* instruction_to_merge) { in MergeFusionInstruction() argument3296 Cast<HloFusionInstruction>(instruction_to_merge)); in MergeFusionInstruction()3301 HloInstruction* instruction_to_merge) { in MergeFusionInstructionIntoMultiOutput() argument3304 Cast<HloFusionInstruction>(instruction_to_merge)); in MergeFusionInstructionIntoMultiOutput()