Force-place embedding variables on CPUs ein eager mode.
This avoids problems which happen because most optimizers do not have sparse updating gpu kernels implemented. Fixes #22042 PiperOrigin-RevId: 213654354
Loading
Please sign in to comment
This avoids problems which happen because most optimizers do not have sparse updating gpu kernels implemented. Fixes #22042 PiperOrigin-RevId: 213654354