Searched refs:unnested_state_size (Results 1 – 1 of 1) sorted by relevance
3012 def create_zeros(unnested_state_size): argument3013 flat_dims = tensor_shape.TensorShape(unnested_state_size).as_list()