Home
last modified time | relevance | path

Searched refs:fusion_builder (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dparallel_reduction_test.cc99 auto fusion_builder = HloComputation::Builder("fusion_computation"); in TEST_F() local
101 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()
/external/tensorflow/tensorflow/compiler/xla/service/
Dheap_simulator_test.cc520 auto fusion_builder = HloComputation::Builder("simple_two_way_forwarding"); in TEST_F() local
522 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() local
596 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() local
640 HloComputation::Builder& builder = fusion_builder; in TEST_F()
[all …]
Dmemory_space_assignment_test.cc1030 HloComputation::Builder fusion_builder("fusion"); in TEST_P() local
1031 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() local
1182 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 …]