Searched refs:inner_builder (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | outfeed_in_nested_computation_test.cc | 128 XlaBuilder inner_builder("true_computation"); in XLA_TEST_F() local 129 XlaOp param = Parameter(&inner_builder, 0, result_shape, "param"); in XLA_TEST_F() 132 return inner_builder.Build(); in XLA_TEST_F() 136 XlaBuilder inner_builder("false_computation"); in XLA_TEST_F() local 137 Parameter(&inner_builder, 0, result_shape, "param"); in XLA_TEST_F() 138 return inner_builder.Build(); in XLA_TEST_F()
|
D | conditional_test.cc | 344 XlaBuilder inner_builder(TestName() + ".inner_conditional"); in XLA_TEST_F() local 345 auto pred_cond = Parameter(&inner_builder, 0, r0bool, "param0"); in XLA_TEST_F() 346 auto true_operand = Parameter(&inner_builder, 1, r0f32_, "param1"); in XLA_TEST_F() 347 auto false_operand = Parameter(&inner_builder, 2, r0f32_, "param2"); in XLA_TEST_F() 350 auto inner_builder_result = inner_builder.Build(); in XLA_TEST_F() 632 XlaBuilder inner_builder(TestName() + ".inner_conditional"); in XLA_TEST_F() local 636 auto param0 = Parameter(&inner_builder, 0, tuple_shape, "param0"); in XLA_TEST_F() 643 auto inner_builder_result = inner_builder.Build(); in XLA_TEST_F() 662 XlaBuilder inner_builder(TestName() + ".inner_conditional"); in XLA_TEST_F() local 666 auto param0 = Parameter(&inner_builder, 0, tuple_shape, "param0"); in XLA_TEST_F() [all …]
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_dataflow_analysis_test.cc | 413 auto inner_builder = HloComputation::Builder("InnerComputation"); in TEST_P() local 414 auto inner_param0 = inner_builder.AddInstruction( in TEST_P() 416 auto inner_param1 = inner_builder.AddInstruction( in TEST_P() 418 auto add = inner_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_P() 421 module_->AddEmbeddedComputation(inner_builder.Build()); in TEST_P() 671 auto inner_builder = HloComputation::Builder("inner_body"); in TEST_P() local 672 auto inner_param = inner_builder.AddInstruction( in TEST_P() 674 auto inner_element_0 = inner_builder.AddInstruction( in TEST_P() 676 auto inner_element_1 = inner_builder.AddInstruction( in TEST_P() 678 auto add = inner_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_P() [all …]
|
D | hlo_alias_analysis_test.cc | 711 auto inner_builder = HloComputation::Builder("inner_body"); in TEST_F() local 712 auto inner_param = inner_builder.AddInstruction( in TEST_F() 714 auto inner_element_0 = inner_builder.AddInstruction( in TEST_F() 716 auto inner_element_1 = inner_builder.AddInstruction( in TEST_F() 718 auto add = inner_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F() 720 inner_builder.AddInstruction( in TEST_F() 723 module_->AddEmbeddedComputation(inner_builder.Build()); in TEST_F()
|