Searched refs:fusion_builder (Results 1 – 3 of 3) sorted by relevance
140 auto fusion_builder = HloComputation::Builder("fusion_computation"); in TEST_F() local142 HloInstruction* constant = fusion_builder.AddInstruction( in TEST_F()145 HloInstruction* param = fusion_builder.AddInstruction( in TEST_F()148 fusion_builder.AddInstruction(HloInstruction::CreateReduce( in TEST_F()152 fusion_builder.AddInstruction(HloInstruction::CreateTuple(outputs)); in TEST_F()153 fusion_computation = module->AddEmbeddedComputation(fusion_builder.Build()); in TEST_F()
1065 HloComputation::Builder fusion_builder("fusion"); in TEST_P() local1066 HloInstruction* fusion_param = fusion_builder.AddInstruction( in TEST_P()1068 HloInstruction* fusion_element0 = fusion_builder.AddInstruction( in TEST_P()1070 HloInstruction* fusion_element1 = fusion_builder.AddInstruction( in TEST_P()1072 fusion_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_P()1075 module->AddEmbeddedComputation(fusion_builder.Build()); in TEST_P()1216 HloComputation::Builder fusion_builder("fusion"); in TEST_P() local1217 HloInstruction* fusion_param = fusion_builder.AddInstruction( in TEST_P()1219 HloInstruction* fusion_element0 = fusion_builder.AddInstruction( in TEST_P()1221 HloInstruction* fusion_element1 = fusion_builder.AddInstruction( in TEST_P()[all …]
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 …]