Searched refs:flat_new_output (Results 1 – 2 of 2) sorted by relevance
225 def _copy_some_through(flat_new_output, flat_new_state): argument229 flat_new_output = [231 for zero_output, new_output in zip(flat_zero_output, flat_new_output)237 return flat_new_output + flat_new_state247 flat_new_output = nest.flatten(new_output)251 lambda: flat_new_output + flat_new_state,253 lambda: _copy_some_through(flat_new_output, flat_new_state))
4481 flat_new_output = compute_masked_output(mask_t, flat_output,4496 for ta, out in zip(output_ta_t, flat_new_output))4498 tuple(flat_new_output)) + tuple(new_states)