Searched defs:fusion_instruction (Results 1 – 10 of 10) sorted by relevance
45 Status Defuse(HloInstruction* fusion_instruction) { in Defuse()102 HloInstruction* fusion_instruction = in Run() local
542 HloInstruction* fusion_instruction = nullptr; in Run() local622 HloInstruction* fusion_instruction; in AddFusionInstruction() local638 HloInstruction* fusion_instruction, HloInstruction* producer) { in FuseInstruction()646 HloInstruction* fusion_instruction = AddFusionInstruction(producer, consumer); in Fuse() local658 HloInstruction* fusion_instruction = AddFusionInstruction(producer, consumer); in FuseIntoMultiOutput() local
42 HloInstruction* FuseInstruction(HloInstruction* fusion_instruction, in FuseInstruction()
465 void SetFusionInstruction(HloInstruction* fusion_instruction) { in SetFusionInstruction()
70 HloInstruction* root_instruction, HloInstruction* fusion_instruction) in HloComputation()725 HloInstruction* fusion_instruction) { in FuseInstructionsInto()746 HloInstruction* fusion_instruction = AddInstruction( in CreateFusionInstruction() local
4770 HloInstruction* fusion_instruction = in TEST_P() local
134 HloInstruction* fusion_instruction, HloInstruction* producer) { in FuseInstruction()
206 HloInstruction* fusion_instruction, HloInstruction* producer) { in FuseInstruction()
300 HloInstruction* fusion_instruction = in RunDotOutputFusion() local
76 auto fusion_instruction = computation->root_instruction(); in TEST_F() local123 auto fusion_instruction = computation->root_instruction(); in TEST_F() local