Searched refs:fusion_builder (Results 1 – 3 of 3) sorted by relevance
99 auto fusion_builder = HloComputation::Builder("fusion_computation"); in TEST_F() local101 HloInstruction* constant = fusion_builder.AddInstruction( in TEST_F()104 HloInstruction* param = fusion_builder.AddInstruction( in TEST_F()107 fusion_builder.AddInstruction(HloInstruction::CreateReduce( in TEST_F()111 fusion_builder.AddInstruction(HloInstruction::CreateTuple(outputs)); in TEST_F()112 fusion_computation = module->AddEmbeddedComputation(fusion_builder.Build()); in TEST_F()
520 auto fusion_builder = HloComputation::Builder("simple_two_way_forwarding"); in TEST_F() local522 auto param = fusion_builder.AddInstruction( in TEST_F()524 fusion_builder.AddInstruction(HloInstruction::CreateTuple({param, param})); in TEST_F()527 module->AddEmbeddedComputation(fusion_builder.Build()); in TEST_F()594 auto fusion_builder = HloComputation::Builder("simple_two_way_forwarding"); in TEST_F() local596 auto param = fusion_builder.AddInstruction( in TEST_F()598 fusion_builder.AddInstruction(HloInstruction::CreateTuple({param, param})); in TEST_F()601 module->AddEmbeddedComputation(fusion_builder.Build()); in TEST_F()638 HloComputation::Builder fusion_builder("fusion"); in TEST_F() local640 HloComputation::Builder& builder = fusion_builder; in TEST_F()[all …]
1030 HloComputation::Builder fusion_builder("fusion"); in TEST_P() local1031 HloInstruction* fusion_param = fusion_builder.AddInstruction( in TEST_P()1033 HloInstruction* fusion_element0 = fusion_builder.AddInstruction( in TEST_P()1035 HloInstruction* fusion_element1 = fusion_builder.AddInstruction( in TEST_P()1037 fusion_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_P()1040 module->AddEmbeddedComputation(fusion_builder.Build()); in TEST_P()1181 HloComputation::Builder fusion_builder("fusion"); in TEST_P() local1182 HloInstruction* fusion_param = fusion_builder.AddInstruction( in TEST_P()1184 HloInstruction* fusion_element0 = fusion_builder.AddInstruction( in TEST_P()1186 HloInstruction* fusion_element1 = fusion_builder.AddInstruction( in TEST_P()[all …]