Searched refs:flat_state (Results 1 – 3 of 3) sorted by relevance
212 flat_state = nest.flatten(state)238 for state, new_state in zip(flat_state, flat_new_state)272 empty_update = lambda: flat_zero_output + flat_state280 if len(final_output_and_state) != len(flat_zero_output) + len(flat_state):288 for substate, flat_substate in zip(final_state, flat_state):1125 flat_state = nest.flatten(state)1126 flat_state = [ops.convert_to_tensor(s) for s in flat_state]1127 state = nest.pack_sequence_as(structure=state, flat_sequence=flat_state)1139 flat_emit_dtypes = [flat_state[0].dtype] * len(flat_emit_size)1487 flat_state = nest.flatten(state)[all …]
4558 flat_state = nest.flatten(states)4560 for state, new_state in zip(flat_state, flat_new_state):4564 flat_state)4595 flat_state = nest.flatten(states)4597 for state, new_state in zip(flat_state, flat_new_state):
516 def _get_state_shape(flat_state): argument517 state_shape = [batch] + tensor_shape.TensorShape(flat_state).as_list()