Searched refs:flat_new_output (Results 1 – 2 of 2) sorted by relevance
243 def _copy_some_through(flat_new_output, flat_new_state): argument247 flat_new_output = [249 for zero_output, new_output in zip(flat_zero_output, flat_new_output)]253 return flat_new_output + flat_new_state262 flat_new_output = nest.flatten(new_output)265 time < min_sequence_length, lambda: flat_new_output + flat_new_state,267 lambda: _copy_some_through(flat_new_output, flat_new_state))
3571 flat_new_output = tuple(3588 for ta, out in zip(output_ta_t, flat_new_output))3590 tuple(flat_new_output)) + tuple(new_states)