Searched refs:elements_finished (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn.py | 1100 (elements_finished, next_input,
1162 def condition(unused_time, elements_finished, *_): argument
1163 return math_ops.logical_not(math_ops.reduce_all(elements_finished))
1165 def body(time, elements_finished, current_input, emit_ta, state, argument
1209 return array_ops.where(elements_finished, cur_i, cand_i)
1219 elements_finished = math_ops.logical_or(elements_finished, next_finished)
1221 return (next_time, elements_finished, next_input, emit_ta, next_state,
1228 time, elements_finished, next_input, emit_ta, state, loop_state
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_cell_test.py | 2082 elements_finished = (time_ >= sequence_length)
2083 finished = math_ops.reduce_all(elements_finished)
2089 return (elements_finished, next_input, next_state, emit_output, None)
2193 elements_finished = array_ops.tile([time_ >= max_time], [batch_size])
2194 finished = math_ops.reduce_all(elements_finished)
2200 return (elements_finished, next_input, next_state, emit_output,
2236 elements_finished = array_ops.tile([time_ >= max_time], [batch_size])
2237 finished = math_ops.reduce_all(elements_finished)
2243 return (elements_finished, next_input, next_state, emit_output,
2279 elements_finished = array_ops.tile([time_ >= max_time], [batch_size])
[all …]
|