Searched refs:FuseInstruction (Results 1 – 8 of 8) sorted by relevance
557 fusion_instruction->FuseInstruction(producer); in Fuse()
781 fusion->FuseInstruction(map_2_x); in TEST_F()784 fusion->FuseInstruction(map_1_x); in TEST_F()
607 fusion_instruction->FuseInstruction(instruction); in FuseInstructionsInto()
1445 HloInstruction* FuseInstruction(HloInstruction* instruction_to_fuse);
1262 FuseInstruction(instruction); in MergeFusionInstruction()1322 FuseInstruction(instruction); in MergeFusionInstructionIntoMultiOutput()
742 HloInstruction* FuseInstruction(HloInstruction* instruction_to_fuse) { in FuseInstruction() function
3307 HloInstruction* HloInstruction::FuseInstruction( in FuseInstruction() function in xla::HloInstruction3309 return Cast<HloFusionInstruction>(this)->FuseInstruction(instruction_to_fuse); in FuseInstruction()
301 HloInstruction* fused_dot = fusion_instruction->FuseInstruction(dot_result); in RunDotOutputFusion()