Commit 2696015d authored by Igor Ganichev's avatar Igor Ganichev Committed by TensorFlower Gardener
Browse files

Colocate slot variables with their base variable in keras OptimizerV2 

Not doing so results in errors because ops such as ResourceApplyRMSProp
take base as well as slot variables. Such ops cannot be placed when the
variable devices are different.

PiperOrigin-RevId: 226924297
parent 2fc6decf
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment