Searched refs:init_state_specs (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))629 flat_state_specs = nest.flatten(init_state_specs)