Searched refs:loop_outputs (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | while_test.py | 56 loop_outputs = xla.while_loop([init_index], loop_cond, loop_body)
58 result = sess.run(loop_outputs, {init_index: 0})
79 loop_outputs = xla.while_loop([init_index, init_sum], loop_cond,
82 result = sess.run(loop_outputs, {init_index: 0, init_sum: 0.0})
84 no_iters_result = sess.run(loop_outputs, {init_index: 10, init_sum: 0.0})
105 loop_outputs = xla.while_loop([init_index, init_sum], loop_cond,
108 result = sess.run(loop_outputs, {init_index: 0, init_sum: 0.0})
110 no_iters_result = sess.run(loop_outputs, {init_index: 10, init_sum: 0.0})
130 loop_outputs = xla.while_loop([init_index, 42], loop_cond, loop_body)
132 result = sess.run(loop_outputs, {init_index: 0})
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 2034 tensorflow::OutputList loop_outputs; in TF_FinishWhileHelper() local
2037 body_fn, params->name, &loop_outputs); in TF_FinishWhileHelper()
2050 DCHECK_LE(loop_outputs.size(), num_loop_vars); in TF_FinishWhileHelper()
2051 for (int i = 0; i < loop_outputs.size(); ++i) { in TF_FinishWhileHelper()
2052 outputs[i] = {ToOperation(loop_outputs[i].node()), loop_outputs[i].index()}; in TF_FinishWhileHelper()
|