Searched refs:get_state_shape (Results 1 – 1 of 1) sorted by relevance
173 def get_state_shape(s): function181 return nest.map_structure(get_state_shape, state_size)