Searched defs:fusion_instruction (Results 1 – 11 of 11) sorted by relevance
565 HloInstruction* fusion_instruction = nullptr; in Run() local708 HloInstruction* fusion_instruction; in AddFusionInstruction() local727 HloInstruction* fusion_instruction, HloInstruction* producer) { in FuseInstruction()732 HloInstruction* producer, HloInstruction* fusion_instruction) { in UpdateReusedOperandsForFusion()758 HloInstruction* fusion_instruction = in Fuse() local773 HloInstruction* fusion_instruction = in FuseIntoMultiOutput() local
44 Status Defuse(HloInstruction* fusion_instruction) { in Defuse()103 HloInstruction* fusion_instruction = in Run() local
39 HloInstruction* FuseInstruction(HloInstruction* fusion_instruction, in FuseInstruction()
78 HloInstruction* root_instruction, HloInstruction* fusion_instruction) in HloComputation()742 HloInstruction* fusion_instruction = AddInstruction( in CreateFusionInstruction() local
570 void SetFusionInstruction(HloInstruction* fusion_instruction) { in SetFusionInstruction()
689 HloInstruction* fusion_instruction = comp->AddInstruction( in MakeFusionInstruction() local
5699 HloInstruction* fusion_instruction = in TEST_P() local
75 auto fusion_instruction = computation->root_instruction(); in TEST_F() local122 auto fusion_instruction = computation->root_instruction(); in TEST_F() local
128 HloInstruction* fusion_instruction, HloInstruction* producer) { in FuseInstruction()
201 HloInstruction* fusion_instruction, HloInstruction* producer) { in FuseInstruction()
299 HloInstruction* fusion_instruction = in RunDotOutputFusion() local