Replace `tensorflow.python.keras` with `keras`. `tensorflow.python.keras` is an old copy and is deprecated.
PiperOrigin-RevId: 485551171
A
A. Unique TensorFlower committed
c2c88d46525ad1b6330593701dfe2fe4b52ba5f3
Parent: abce63c