1c499b27e225d0c3b8634347c1bb924e26842cd1,txtgen/modules/decoders/rnn_decoder_base.py,RNNDecoderBase,_build,#RNNDecoderBase#Any#Any#,123

Before Change

        self._add_internal_trainable_variables()
        // Add trainable variables of `self._cell` which may be constructed
        // externally
        self._add_trainable_variable(self._cell.trainable_variables())
        self._built = True

        return outputs, final_state, sequence_lengths

After Change

        max_decoding_length_train = self._hparams.max_decoding_length_train
        if max_decoding_length_train is None:
            max_decoding_length_train = utils.MAX_SEQ_LENGTH
        max_decoding_length_infer = self._hparams.max_decoding_length_infer
        if max_decoding_length_infer is None:
            max_decoding_length_infer = utils.MAX_SEQ_LENGTH
        max_decoding_length = tf.cond(
            context.is_train(),
            lambda: max_decoding_length_train,
            lambda: max_decoding_length_infer)

Instances