Searched refs:outer_body (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | while_test.py | 189 def outer_body(i, x): function
200 outer_body, (0, 1.0),
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | while_test.cc | 1125 XlaComputation outer_body; in XLA_TEST_F() local
1132 outer_body = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
1138 While(outer_condition, outer_body, init); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_alias_analysis_test.cc | 739 HloComputation* outer_body = in TEST_F() local
750 HloInstruction::CreateWhile(tuple_shape, condition2, outer_body, tuple)); in TEST_F()
|
| D | hlo_dataflow_analysis_test.cc | 699 HloComputation* outer_body = in TEST_P() local
710 HloInstruction::CreateWhile(tuple_shape, condition, outer_body, tuple)); in TEST_P()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | control_flow_ops_py_test.py | 1508 def outer_body(i, x): function
1520 outer_body, (0, 1.0),
2932 def outer_body(i, x): function
2941 i, x = control_flow_ops.while_loop(lambda i, x: i < 3, outer_body, [0, 0.0])
|