This is to get compatible for an incoming Keras optimizer migration. PiperOrigin-RevId: 469561702
Attach a file by drag & drop or click to upload
