Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/tests/
Dconditional_test.cc503 XlaBuilder true_builder(TestName() + ".true"); in XLA_TEST_F() local
505 Parameter(&true_builder, 0, empty_tuple_, "tuple"); in XLA_TEST_F()
506 auto true_pred = ConstantR0<bool>(&true_builder, true); in XLA_TEST_F()
507 auto true_scalar = ConstantR0<float>(&true_builder, 12.2f); in XLA_TEST_F()
508 auto true_array = ConstantR1<float>(&true_builder, {12.8f, 14.6f}); in XLA_TEST_F()
509 Tuple(&true_builder, {true_pred, true_scalar, true_array}); in XLA_TEST_F()
511 auto true_builder_result = true_builder.Build(); in XLA_TEST_F()
542 XlaBuilder true_builder(TestName() + ".true"); in XLA_TEST_F() local
544 Parameter(&true_builder, 0, empty_tuple_, "tuple"); in XLA_TEST_F()
545 auto true_constant1 = ConstantR0<float>(&true_builder, 12.2f); in XLA_TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment_test.cc770 auto true_builder = HloComputation::Builder(TestName() + "_TrueBranch"); in TEST_F() local
772 auto param = true_builder.AddInstruction( in TEST_F()
774 auto gte0 = true_builder.AddInstruction( in TEST_F()
776 auto gte1 = true_builder.AddInstruction( in TEST_F()
778 auto add = true_builder.AddInstruction( in TEST_F()
780 true_builder.AddInstruction(HloInstruction::CreateTuple({add})); in TEST_F()
783 m->AddEmbeddedComputation(true_builder.Build()); in TEST_F()
Dhlo_dataflow_analysis_test.cc1594 auto true_builder = HloComputation::Builder(TestName() + "_true"); in TEST_P() local
1595 auto true_param = true_builder.AddInstruction( in TEST_P()
1598 module_->AddEmbeddedComputation(true_builder.Build()); in TEST_P()
1674 auto true_builder = HloComputation::Builder(TestName() + "_true"); in TEST_P() local
1675 auto true_param = true_builder.AddInstruction( in TEST_P()
1677 auto true_x = true_builder.AddInstruction( in TEST_P()
1679 auto true_y = true_builder.AddInstruction( in TEST_P()
1681 auto add = true_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_P()
1684 module_->AddEmbeddedComputation(true_builder.Build()); in TEST_P()