Searched defs:fusion_instruction (Results 1 – 6 of 6) sorted by relevance
45 Status Defuse(HloInstruction* fusion_instruction) { in Defuse()102 HloInstruction* fusion_instruction = in Run() local
494 HloInstruction* fusion_instruction; in Run() local537 HloInstruction* fusion_instruction; in AddFusionInstruction() local556 HloInstruction* fusion_instruction = AddFusionInstruction(producer, consumer); in Fuse() local565 HloInstruction* fusion_instruction = AddFusionInstruction(producer, consumer); in FuseIntoMultiOutput() local
399 void SetFusionInstruction(HloInstruction* fusion_instruction) { in SetFusionInstruction()
70 HloInstruction* root_instruction, HloInstruction* fusion_instruction) in HloComputation()597 HloInstruction* fusion_instruction) { in FuseInstructionsInto()618 HloInstruction* fusion_instruction = AddInstruction( in CreateFusionInstruction() local
76 auto fusion_instruction = computation->root_instruction(); in TEST_F() local123 auto fusion_instruction = computation->root_instruction(); in TEST_F() local
293 HloInstruction* fusion_instruction = in RunDotOutputFusion() local