Home
last modified time | relevance | path

Searched refs:false_builder (Results 1 – 4 of 4) 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.cc782 auto false_builder = HloComputation::Builder(TestName() + "_FalseBranch"); in TEST_F() local
785 false_builder.AddInstruction( in TEST_F()
788 auto token = false_builder.AddInstruction(HloInstruction::CreateToken()); in TEST_F()
789 auto infeed = false_builder.AddInstruction( in TEST_F()
791 auto infeed_data = false_builder.AddInstruction( in TEST_F()
793 false_builder.AddInstruction(HloInstruction::CreateTuple({infeed_data})); in TEST_F()
796 m->AddEmbeddedComputation(false_builder.Build()); in TEST_F()
Dhlo_dataflow_analysis_test.cc1674 auto false_builder = HloComputation::Builder(TestName() + "_false"); in TEST_P() local
1675 auto false_param = false_builder.AddInstruction( in TEST_P()
1678 module_->AddEmbeddedComputation(false_builder.Build()); in TEST_P()
1760 auto false_builder = HloComputation::Builder(TestName() + "_false"); in TEST_P() local
1761 auto false_param = false_builder.AddInstruction( in TEST_P()
1763 auto false_x = false_builder.AddInstruction( in TEST_P()
1765 auto false_y = false_builder.AddInstruction( in TEST_P()
1767 auto sub = false_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_P()
1770 module_->AddEmbeddedComputation(false_builder.Build()); in TEST_P()
Dmemory_space_assignment_test.cc1756 auto false_builder = HloComputation::Builder("False"); in TEST_P() local
1757 HloInstruction* false_param = false_builder.AddInstruction( in TEST_P()
1760 module->AddEmbeddedComputation(false_builder.Build()); in TEST_P()