Home
last modified time | relevance | path

Searched refs:false_builder (Results 1 – 5 of 5) 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()
Ddynamic_dimension_inference_test.cc852 auto false_builder = HloComputation::Builder("false"); in TEST_F() local
854 auto false_param = false_builder.AddInstruction( in TEST_F()
856 auto gte_0 = false_builder.AddInstruction( in TEST_F()
858 auto gte_1 = false_builder.AddInstruction( in TEST_F()
860 auto add = false_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
862 false_builder.AddInstruction(HloInstruction::CreateTuple({add})); in TEST_F()
865 module_->AddEmbeddedComputation(false_builder.Build()); in TEST_F()
Dhlo_dataflow_analysis_test.cc1727 auto false_builder = HloComputation::Builder(TestName() + "_false"); in TEST_P() local
1728 auto false_param = false_builder.AddInstruction( in TEST_P()
1731 module_->AddEmbeddedComputation(false_builder.Build()); in TEST_P()
1813 auto false_builder = HloComputation::Builder(TestName() + "_false"); in TEST_P() local
1814 auto false_param = false_builder.AddInstruction( in TEST_P()
1816 auto false_x = false_builder.AddInstruction( in TEST_P()
1818 auto false_y = false_builder.AddInstruction( in TEST_P()
1820 auto sub = false_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_P()
1823 module_->AddEmbeddedComputation(false_builder.Build()); in TEST_P()
Dmemory_space_assignment_test.cc2879 auto false_builder = HloComputation::Builder("False"); in TEST_P() local
2880 HloInstruction* false_param = false_builder.AddInstruction( in TEST_P()
2883 module->AddEmbeddedComputation(false_builder.Build()); in TEST_P()