Searched refs:_get_state_shape (Results 1 – 1 of 1) sorted by relevance
470 def _get_state_shape(flat_state): function473 state_shape = nest.map_structure(_get_state_shape, state_size)