Return optimizer slot variables for the caller to restore when appropriate.
Return optimizer slot variables for the caller to restore when appropriate.

Part of a TensorFlow change to collect slot variable restorations in a queue and restore them as a batch.

PiperOrigin-RevId: 402936796

keras-team/keras,keras-team/keras