Searched refs:flat_new_state (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn.py | 228 def _copy_some_through(flat_new_output, flat_new_state): argument
236 flat_new_state = [
238 for state, new_state in zip(flat_state, flat_new_state)
240 return flat_new_output + flat_new_state
249 flat_new_state = nest.flatten(new_state)
254 lambda: flat_new_output + flat_new_state,
256 lambda: _copy_some_through(flat_new_output, flat_new_state))
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 4559 flat_new_state = nest.flatten(new_states)
4560 for state, new_state in zip(flat_state, flat_new_state):
4563 flat_final_state = compute_masked_output(mask_t, flat_new_state,
4596 flat_new_state = nest.flatten(new_states)
4597 for state, new_state in zip(flat_state, flat_new_state):
4604 new_states = nest.pack_sequence_as(initial_states, flat_new_state)
|