Searched defs:fusion_instruction (Results 1 – 11 of 11) sorted by relevance
45 Status Defuse(HloInstruction* fusion_instruction) { in Defuse()102 HloInstruction* fusion_instruction = in Run() local
552 HloInstruction* fusion_instruction = nullptr; in Run() local632 HloInstruction* fusion_instruction; in AddFusionInstruction() local648 HloInstruction* fusion_instruction, HloInstruction* producer) { in FuseInstruction()656 HloInstruction* fusion_instruction = AddFusionInstruction(producer, consumer); in Fuse() local668 HloInstruction* fusion_instruction = AddFusionInstruction(producer, consumer); in FuseIntoMultiOutput() local
42 HloInstruction* FuseInstruction(HloInstruction* fusion_instruction, in FuseInstruction()
473 void SetFusionInstruction(HloInstruction* fusion_instruction) { in SetFusionInstruction()
70 HloInstruction* root_instruction, HloInstruction* fusion_instruction) in HloComputation()746 HloInstruction* fusion_instruction) { in FuseInstructionsInto()767 HloInstruction* fusion_instruction = AddInstruction( in CreateFusionInstruction() local
574 HloInstruction* fusion_instruction = comp->AddInstruction( in MakeFusionInstruction() local
5515 HloInstruction* fusion_instruction = in TEST_P() local
134 HloInstruction* fusion_instruction, HloInstruction* producer) { in FuseInstruction()
76 auto fusion_instruction = computation->root_instruction(); in TEST_F() local123 auto fusion_instruction = computation->root_instruction(); in TEST_F() local
206 HloInstruction* fusion_instruction, HloInstruction* producer) { in FuseInstruction()
299 HloInstruction* fusion_instruction = in RunDotOutputFusion() local