Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_conversion_folding_test.cc220 HloComputation::Builder sum_builder("add"); in TEST_F() local
221 auto x = sum_builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
223 auto y = sum_builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
225 sum_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
227 HloComputation* sum = module->AddEmbeddedComputation(sum_builder.Build()); in TEST_F()
Dall_reduce_combiner_test.cc87 HloComputation::Builder sum_builder(HloOpcodeString(type)); in MakeReduction() local
88 auto x = sum_builder.AddInstruction(HloInstruction::CreateParameter( in MakeReduction()
90 auto y = sum_builder.AddInstruction(HloInstruction::CreateParameter( in MakeReduction()
92 sum_builder.AddInstruction( in MakeReduction()
95 module->AddEmbeddedComputation(sum_builder.Build()); in MakeReduction()
Dbfloat16_normalization_test.cc240 HloComputation::Builder sum_builder("sum"); in TEST_F() local
241 auto x = sum_builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
243 auto y = sum_builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
245 sum_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
248 module->AddEmbeddedComputation(sum_builder.Build()); in TEST_F()