Searched refs:flat_new_output (Results 1 – 2 of 2) sorted by relevance
228 def _copy_some_through(flat_new_output, flat_new_state): argument232 flat_new_output = [234 for zero_output, new_output in zip(flat_zero_output, flat_new_output)240 return flat_new_output + flat_new_state250 flat_new_output = nest.flatten(new_output)254 lambda: flat_new_output + flat_new_state,256 lambda: _copy_some_through(flat_new_output, flat_new_state))
4554 flat_new_output = compute_masked_output(mask_t, flat_output,4569 for ta, out in zip(output_ta_t, flat_new_output))4571 tuple(flat_new_output)) + tuple(new_states)