Home
last modified time | relevance | path

Searched refs:true_builder (Results 1 – 5 of 5) 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.cc767 auto true_builder = HloComputation::Builder(TestName() + "_TrueBranch"); in TEST_F() local
769 auto param = true_builder.AddInstruction( in TEST_F()
771 auto gte0 = true_builder.AddInstruction( in TEST_F()
773 auto gte1 = true_builder.AddInstruction( in TEST_F()
775 auto add = true_builder.AddInstruction( in TEST_F()
777 true_builder.AddInstruction(HloInstruction::CreateTuple({add})); in TEST_F()
780 m->AddEmbeddedComputation(true_builder.Build()); in TEST_F()
Ddynamic_dimension_inference_test.cc831 auto true_builder = HloComputation::Builder("true"); in TEST_F() local
833 auto true_param = true_builder.AddInstruction( in TEST_F()
835 auto gte_0 = true_builder.AddInstruction( in TEST_F()
837 auto gte_1 = true_builder.AddInstruction( in TEST_F()
839 auto add = true_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
841 true_builder.AddInstruction(HloInstruction::CreateTuple({add})); in TEST_F()
844 module_->AddEmbeddedComputation(true_builder.Build()); in TEST_F()
Dhlo_dataflow_analysis_test.cc1721 auto true_builder = HloComputation::Builder(TestName() + "_true"); in TEST_P() local
1722 auto true_param = true_builder.AddInstruction( in TEST_P()
1725 module_->AddEmbeddedComputation(true_builder.Build()); in TEST_P()
1801 auto true_builder = HloComputation::Builder(TestName() + "_true"); in TEST_P() local
1802 auto true_param = true_builder.AddInstruction( in TEST_P()
1804 auto true_x = true_builder.AddInstruction( in TEST_P()
1806 auto true_y = true_builder.AddInstruction( in TEST_P()
1808 auto add = true_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_P()
1811 module_->AddEmbeddedComputation(true_builder.Build()); in TEST_P()
Dmemory_space_assignment_test.cc2847 auto true_builder = HloComputation::Builder("True"); in TEST_P() local
2848 HloInstruction* true_param = true_builder.AddInstruction( in TEST_P()
2851 true_builder.AddInstruction(HloInstruction::CreateIota(shape2, 0)); in TEST_P()
2852 HloInstruction* slice = true_builder.AddInstruction( in TEST_P()
2855 true_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_P()
2857 HloInstruction* negate0 = true_builder.AddInstruction( in TEST_P()
2859 HloInstruction* negate1 = true_builder.AddInstruction( in TEST_P()
2861 HloInstruction* negate2 = true_builder.AddInstruction( in TEST_P()
2863 HloInstruction* negate3 = true_builder.AddInstruction( in TEST_P()
2865 HloInstruction* negate4 = true_builder.AddInstruction( in TEST_P()
[all …]