Searched refs:instruction_to_fuse (Results 1 – 3 of 3) sorted by relevance
1253 HloInstruction* FuseInstruction(HloInstruction* instruction_to_fuse) { in FuseInstruction() argument1254 CHECK(instruction_to_fuse->IsFusible()) << instruction_to_fuse->ToString(); in FuseInstruction()1255 return AppendInstructionIntoCalledComputation(instruction_to_fuse); in FuseInstruction()1264 HloInstruction* instruction_to_fuse) { in FuseInstructionIntoMultiOutput() argument1265 return AppendInstructionIntoCalledComputation(instruction_to_fuse, in FuseInstructionIntoMultiOutput()
1941 HloInstruction* FuseInstruction(HloInstruction* instruction_to_fuse);1945 HloInstruction* instruction_to_fuse);
4575 HloInstruction* instruction_to_fuse) { in FuseInstruction() argument4576 return Cast<HloFusionInstruction>(this)->FuseInstruction(instruction_to_fuse); in FuseInstruction()4580 HloInstruction* instruction_to_fuse) { in FuseInstructionIntoMultiOutput() argument4582 instruction_to_fuse); in FuseInstructionIntoMultiOutput()