docs: def state_size(self):
      """Tuple of `tf.TensorShape`s indicating the size of state tensors."""
      return LSTMState(tf.TensorShape([self._hidden_state_size]),
                       tf.TensorShape([self._hidden_size]))