Home
last modified time | relevance | path

Searched refs:inner_builder (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Doutfeed_in_nested_computation_test.cc128 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()
Dconditional_test.cc344 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/
Dhlo_dataflow_analysis_test.cc413 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 …]
Dhlo_alias_analysis_test.cc711 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()