Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/tests/
Dconditional_test.cc514 XlaBuilder false_builder(TestName() + ".false"); in XLA_TEST_F() local
516 Parameter(&false_builder, 0, empty_tuple_, "tuple"); in XLA_TEST_F()
517 auto false_pred = ConstantR0<bool>(&false_builder, false); in XLA_TEST_F()
518 auto false_scalar = ConstantR0<float>(&false_builder, 25.6f); in XLA_TEST_F()
519 auto false_array = ConstantR1<float>(&false_builder, {26.4f, 32.6f}); in XLA_TEST_F()
520 Tuple(&false_builder, {false_pred, false_scalar, false_array}); in XLA_TEST_F()
522 auto false_builder_result = false_builder.Build(); in XLA_TEST_F()
556 XlaBuilder false_builder(TestName() + ".false"); in XLA_TEST_F() local
558 Parameter(&false_builder, 0, empty_tuple_, "tuple"); in XLA_TEST_F()
559 auto false_constant1 = ConstantR0<float>(&false_builder, 46.6f); in XLA_TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment_test.cc785 auto false_builder = HloComputation::Builder(TestName() + "_FalseBranch"); in TEST_F() local
788 false_builder.AddInstruction( in TEST_F()
791 auto token = false_builder.AddInstruction(HloInstruction::CreateToken()); in TEST_F()
792 auto infeed = false_builder.AddInstruction( in TEST_F()
794 auto infeed_data = false_builder.AddInstruction( in TEST_F()
796 false_builder.AddInstruction(HloInstruction::CreateTuple({infeed_data})); in TEST_F()
799 m->AddEmbeddedComputation(false_builder.Build()); in TEST_F()
Dhlo_dataflow_analysis_test.cc1600 auto false_builder = HloComputation::Builder(TestName() + "_false"); in TEST_P() local
1601 auto false_param = false_builder.AddInstruction( in TEST_P()
1604 module_->AddEmbeddedComputation(false_builder.Build()); in TEST_P()
1686 auto false_builder = HloComputation::Builder(TestName() + "_false"); in TEST_P() local
1687 auto false_param = false_builder.AddInstruction( in TEST_P()
1689 auto false_x = false_builder.AddInstruction( in TEST_P()
1691 auto false_y = false_builder.AddInstruction( in TEST_P()
1693 auto sub = false_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_P()
1696 module_->AddEmbeddedComputation(false_builder.Build()); in TEST_P()