def build(self, inputs):
    self._transformer_decoder = nlp_transformer.DecoderStack(self._hparams)
def build(self, inputs):
    self._transformer_decoder = nlp_transformer.DecoderStack(self._hparams)