Fix checkpointable dependencies in recurrent layers.
1. Add dependency between RNN layer and RNN Cell. 2. Add dependency between Bidirectional layer and the inner forward and backward layers. PiperOrigin-RevId: 204833646
Loading
Please sign in to comment