Searched refs:loop_outputs (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | while_test.py | 52 loop_outputs = xla.while_loop([init_index], loop_cond, loop_body)
54 result = sess.run(loop_outputs, {init_index: 0})
75 loop_outputs = xla.while_loop([init_index, init_sum], loop_cond,
78 result = sess.run(loop_outputs, {init_index: 0, init_sum: 0.0})
80 no_iters_result = sess.run(loop_outputs, {init_index: 10, init_sum: 0.0})
101 loop_outputs = xla.while_loop([init_index, init_sum], loop_cond,
104 result = sess.run(loop_outputs, {init_index: 0, init_sum: 0.0})
106 no_iters_result = sess.run(loop_outputs, {init_index: 10, init_sum: 0.0})
126 loop_outputs = xla.while_loop([init_index, 42], loop_cond, loop_body)
128 result = sess.run(loop_outputs, {init_index: 0})
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 2059 tensorflow::OutputList loop_outputs; in TF_FinishWhileHelper() local
2062 body_fn, params->name, &loop_outputs); in TF_FinishWhileHelper()
2075 DCHECK_LE(loop_outputs.size(), num_loop_vars); in TF_FinishWhileHelper()
2076 for (int i = 0; i < loop_outputs.size(); ++i) { in TF_FinishWhileHelper()
2077 outputs[i] = {ToOperation(loop_outputs[i].node()), loop_outputs[i].index()}; in TF_FinishWhileHelper()
|