Searched refs:unnested_state_size (Results 1 – 1 of 1) sorted by relevance
2762 def create_zeros(unnested_state_size): argument2763 flat_dims = tensor_shape.as_shape(unnested_state_size).as_list()