Searched refs:loop_outputs (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | while_test.py | 55 loop_outputs = xla.while_loop([init_index], loop_cond, loop_body)
57 result = sess.run(loop_outputs, {init_index: 0})
78 loop_outputs = xla.while_loop([init_index, init_sum], loop_cond,
81 result = sess.run(loop_outputs, {init_index: 0, init_sum: 0.0})
83 no_iters_result = sess.run(loop_outputs, {init_index: 10, init_sum: 0.0})
104 loop_outputs = xla.while_loop([init_index, init_sum], loop_cond,
107 result = sess.run(loop_outputs, {init_index: 0, init_sum: 0.0})
109 no_iters_result = sess.run(loop_outputs, {init_index: 10, init_sum: 0.0})
129 loop_outputs = xla.while_loop([init_index, 42], loop_cond, loop_body)
131 result = sess.run(loop_outputs, {init_index: 0})
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | hierarchical_controller.py | 938 loop_outputs = control_flow_ops.while_loop(condition, body, loop_vars)
940 last_c = loop_outputs[-4]
941 last_h = loop_outputs[-3]
942 actions = loop_outputs[-2].stack()
944 log_probs = loop_outputs[-1]
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 2441 tensorflow::OutputList loop_outputs; in TF_FinishWhileHelper() local
2444 body_fn, params->name, &loop_outputs); in TF_FinishWhileHelper()
2457 DCHECK_LE(loop_outputs.size(), num_loop_vars); in TF_FinishWhileHelper()
2458 for (int i = 0; i < loop_outputs.size(); ++i) { in TF_FinishWhileHelper()
2459 outputs[i] = {ToOperation(loop_outputs[i].node()), loop_outputs[i].index()}; in TF_FinishWhileHelper()
|