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

Try2: 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: 227060221
parent a25801c1
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