Searched refs:inner_params (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/c/ |
| D | while_loop_test.cc | 236 TF_WhileParams inner_params = in TEST_F() local
239 inner_params.name = "inner_loop"; in TEST_F()
242 TF_Operation* three = ScalarConst(3, inner_params.cond_graph, s_); in TEST_F()
245 inner_params.cond_inputs[0], {three, 0}, inner_params.cond_graph, s_); in TEST_F()
247 inner_params.cond_output = {inner_less_than, 0}; in TEST_F()
250 TF_Operation* one = ScalarConst(1, inner_params.body_graph, s_, "one"); in TEST_F()
252 TF_Operation* two = ScalarConst(2, inner_params.body_graph, s_, "two"); in TEST_F()
256 Add(inner_params.body_inputs[1].oper, one, inner_params.body_graph, s_); in TEST_F()
258 inner_params.body_outputs[1] = {input2_add, 0}; in TEST_F()
260 TF_Operation* inner_input1_add = Add(inner_params.body_inputs[0].oper, two, in TEST_F()
[all …]
|