Searched refs:batch_size_tensor (Results 1 – 1 of 1) sorted by relevance
3005 def _generate_zero_filled_state(batch_size_tensor, state_size, dtype): argument3007 if batch_size_tensor is None or dtype is None:3010 'batch_size={}, dtype={}'.format(batch_size_tensor, dtype))3014 init_state_size = [batch_size_tensor] + flat_dims