Allows the Keras learning phase to work correctly when layers are called...
Allows the Keras learning phase to work correctly when layers are called inside of functions (e.g. in custom function-wrapped training loops). PiperOrigin-RevId: 230978736
Loading
Please sign in to comment