31b3add957269c206b7862b44117da895f5be781,open_seq2seq/parts/rnns/weight_drop.py,WeightDropLayerNormBasicLSTMCell,call,#WeightDropLayerNormBasicLSTMCell#Any#Any#,98

Before Change

    LSTM cell with layer normalization and recurrent dropout.
    c, h = state
    args = tf.concat([inputs, h], 1)
    print("inputs shape:", inputs)
    concat = self._linear(args)
    dtype = args.dtype

    i, j, f, o = tf.split(value=concat, num_or_size_splits=4, axis=1)

After Change

    LSTM cell with layer normalization and recurrent dropout.
    c, h = state
    args = tf.concat([inputs, h], 1)
    concat = self._linear(args, inputs.get_shape().as_list()[-1], h.get_shape().as_list()[-1])
    dtype = args.dtype

    i, j, f, o = tf.split(value=concat, num_or_size_splits=4, axis=1)

Instances