2d5a9971e9da7e28a4466ba614070838e6f55879,tf_agents/networks/sequential_test.py,SequentialTest,testMixOfNonRecurrentAndRecurrent,#SequentialTest#,85

Before Change

                return_state=True,
                return_sequences=True,
            ),
            tf.keras.layers.Reshape((-1, 4, 4, 2)),
            tf.keras.layers.Conv2D(2, 3),
            tf.keras.layers.TimeDistributed(tf.keras.layers.Flatten()),
            tf.keras.layers.GRU(

After Change

        ),
        tf.keras.layers.Lambda(lambda t: reshape_inner_dims(t, 1, [4, 4, 2])),
        tf.keras.layers.Conv2D(2, 3),
        tf.keras.layers.Lambda(lambda t: reshape_inner_dims(t, 3, [8])),
        tf.keras.layers.GRU(2, return_state=True, return_sequences=True),
        dynamic_unroll_layer.DynamicUnroll(tf.keras.layers.LSTMCell(2)),
    ],

Instances