def decoder_hidden_units(self): # @TODO: is this correct for LSTMStateTuple? return self.decoder_cell.output_size