Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_ordering_test.cc461 auto subbuilder = HloComputation::Builder(TestName() + ".sub"); in TEST_F() local
462 HloInstruction* param = subbuilder.AddInstruction( in TEST_F()
464 HloInstruction* root = subbuilder.AddInstruction( in TEST_F()
466 HloInstruction* dead = subbuilder.AddInstruction( in TEST_F()
469 subbuilder.Build(/*root_instruction=*/root)); in TEST_F()
Dhlo_alias_analysis_test.cc413 auto subbuilder = HloComputation::Builder("Subcomputation"); in TEST_F() local
414 auto subparam0 = subbuilder.AddInstruction( in TEST_F()
416 auto subparam1 = subbuilder.AddInstruction( in TEST_F()
418 auto add = subbuilder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
421 module_->AddEmbeddedComputation(subbuilder.Build()); in TEST_F()
453 auto subbuilder = HloComputation::Builder("Subcomputation"); in TEST_F() local
454 auto subparam0 = subbuilder.AddInstruction( in TEST_F()
456 auto subparam1 = subbuilder.AddInstruction( in TEST_F()
458 auto add = subbuilder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
461 module_->AddEmbeddedComputation(subbuilder.Build()); in TEST_F()
Dhlo_dataflow_analysis_test.cc254 auto subbuilder = HloComputation::Builder("Subcomputation"); in TEST_P() local
255 auto subparam0 = subbuilder.AddInstruction( in TEST_P()
257 auto subparam1 = subbuilder.AddInstruction( in TEST_P()
259 auto add = subbuilder.AddInstruction(HloInstruction::CreateBinary( in TEST_P()
262 module_->AddEmbeddedComputation(subbuilder.Build()); in TEST_P()
304 auto subbuilder = HloComputation::Builder("Subcomputation"); in TEST_P() local
305 auto subparam0 = subbuilder.AddInstruction( in TEST_P()
307 auto subparam1 = subbuilder.AddInstruction( in TEST_P()
309 auto add = subbuilder.AddInstruction(HloInstruction::CreateBinary( in TEST_P()
312 module_->AddEmbeddedComputation(subbuilder.Build()); in TEST_P()
[all …]