Searched refs:init_state_specs (Results 1 – 1 of 1) sorted by relevance
608 def _validate_state_spec(cell_state_sizes, init_state_specs): argument623 '{}'.format(init_state_specs, cell_state_sizes))625 flat_state_specs = nest.flatten(init_state_specs)