Searched refs:init_state_specs (Results 1 – 1 of 1) sorted by relevance
578 def _validate_state_spec(cell_state_sizes, init_state_specs): argument593 '{}'.format(init_state_specs, cell_state_sizes))594 if len(cell_state_sizes) == len(init_state_specs):598 init_state_specs[i].shape[1:]).is_compatible_with(