Home
last modified time | relevance | path

Searched refs:sum_builder (Results 1 – 2 of 2) 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()
Dbfloat16_normalization_test.cc238 HloComputation::Builder sum_builder("sum"); in TEST_F() local
239 auto x = sum_builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
241 auto y = sum_builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
243 sum_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
246 module->AddEmbeddedComputation(sum_builder.Build()); in TEST_F()