bfd0068e7a0922c06cdeff346a6c6339f98ae256,python/eight_mile/tf/layers.py,BiLSTMEncoder1,call,#BiLSTMEncoder1#Any#,809

Before Change

        with tf.name_scope(self.name), tf.variable_scope(self.name):
            rnnout, (fwd_state, backward_state) = tf.nn.bidirectional_dynamic_rnn(self.fwd_rnn, self.bwd_rnn, inputs, sequence_length=lengths, dtype=tf.float32)
        rnnout = tf.concat(axis=2, values=rnnout)
        return self.output_fn(rnnout, ((fwd_state[-1].h, fwd_state[-1].c), (backward_state[-1].h, backward_state[-1].c)))

    @property
    def requires_length(self):
        return self._requires_length

After Change

            h = tf.concat([fw_state[i].h, bw_state[i].h], -1)
            c = tf.concat([fw_state[i].c, bw_state[i].c], -1)
            encoder_state.append(tf.contrib.rnn.LSTMStateTuple(h=h, c=c))
        encoder_state = tuple(encoder_state)
        return self.output_fn(rnnout, encoder_state)

    def output_fn(self, out, state):
        return out, state

Instances