Code changes to get ready for an incoming Keras optimizer migration.
Code changes to get ready for an incoming Keras optimizer migration.

PiperOrigin-RevId: 475429355

keras-team/keras,keras-team/keras