Searched refs:fusion_computation (Results 1 – 7 of 7) sorted by relevance
633 auto fusion_computation = hlo->fused_instructions_computation(); in ResolveInconsistentFusions() local634 auto fusion_root = fusion_computation->root_instruction(); in ResolveInconsistentFusions()643 fusion_computation->DeepCopyInstructionWithCustomCopier( in ResolveInconsistentFusions()655 fusion_computation->set_root_instruction(copy); in ResolveInconsistentFusions()
460 auto fusion_computation = in TEST_F() local468 HloInstruction::FusionKind::kLoop, {neg}, fusion_computation)); in TEST_F()
1147 HloComputation* fusion_computation) in HloFusionInstruction() argument1153 AppendComputation(fusion_computation); in HloFusionInstruction()1154 fusion_computation->SetFusionInstruction(this); in HloFusionInstruction()
754 HloComputation* fusion_computation);
1375 optional<HloComputation*> fusion_computation; in ParseInstructionRhs() local1377 &fusion_computation}; in ParseInstructionRhs()1384 shape, *fusion_kind, operands, *fusion_computation)); in ParseInstructionRhs()
712 HloComputation* fusion_computation);
1209 HloComputation* fusion_computation) { in CreateFusion() argument1211 fusion_computation); in CreateFusion()