Searched refs:cell_state_sizes (Results 1 – 1 of 1) sorted by relevance
612 def _validate_state_spec(cell_state_sizes, init_state_specs): argument627 '{}'.format(init_state_specs, cell_state_sizes))628 flat_cell_state_sizes = nest.flatten(cell_state_sizes)