Searched refs:CreateFusion (Results 1 – 12 of 12) sorted by relevance
409 HloInstruction* z = computation->AddInstruction(HloInstruction::CreateFusion( in TEST_F()445 HloInstruction* z = computation->AddInstruction(HloInstruction::CreateFusion( in TEST_F()492 HloInstruction* z = computation->AddInstruction(HloInstruction::CreateFusion( in TEST_F()
397 auto fusion0 = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_F()410 auto fusion1 = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_F()447 auto fusion = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_F()486 auto fusion = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_F()661 auto body_fusion = builder_body.AddInstruction(HloInstruction::CreateFusion( in TEST_F()
132 HloInstruction::CreateFusion(r0f32_, HloInstruction::FusionKind::kInput, in TEST_F()
546 HloInstruction::CreateFusion(consumer->shape(), kind, consumer)); in AddFusionInstruction()
748 static std::unique_ptr<HloInstruction> CreateFusion(751 static std::unique_ptr<HloInstruction> CreateFusion(
619 HloInstruction::CreateFusion(root->shape(), fusion_kind, root)); in CreateFusionInstruction()
336 CreateFusion(shape, fusion_kind, all_operands(), fused_computation); in CreateFromProto()1200 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateFusion( in CreateFusion() function in xla::HloInstruction1206 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateFusion( in CreateFusion() function in xla::HloInstruction
466 auto fusion = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_F()
1383 instruction = builder->AddInstruction(HloInstruction::CreateFusion( in ParseInstructionRhs()
215 computation()->AddInstruction(HloInstruction::CreateFusion( in DoProducerConsumerMultiOutputFusion()
604 return HloInstruction::CreateFusion( in TEST_F()
294 module->entry_computation()->AddInstruction(HloInstruction::CreateFusion( in RunDotOutputFusion()