Searched refs:builder_f (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_propagation_test.cc | 437 auto builder_f = HloComputation::Builder("fusion"); in TEST_F() local
439 builder_f.AddInstruction(HloInstruction::CreateParameter(0, shape, "a")); in TEST_F()
441 builder_f.AddInstruction(HloInstruction::CreateParameter(1, shape, "b")); in TEST_F()
442 HloInstruction* add_f = builder_f.AddInstruction( in TEST_F()
445 builder_f.AddInstruction(CreateDot(shape, add_f, add_f)); in TEST_F()
446 auto comp_f = module->AddEmbeddedComputation(builder_f.Build()); in TEST_F()
476 auto builder_f = HloComputation::Builder("fusion0"); in TEST_F() local
478 builder_f.AddInstruction(HloInstruction::CreateParameter(0, shape, "a")); in TEST_F()
480 builder_f.AddInstruction(HloInstruction::CreateParameter(1, shape, "b")); in TEST_F()
481 HloInstruction* add_f = builder_f.AddInstruction( in TEST_F()
[all …]
|