Prevent UnrollStore, etc. from being assigned to GPU. Bug introduced after multi host patch.
Prevent UnrollStore, etc. from being assigned to GPU. Bug introduced after multi host patch.

PiperOrigin-RevId: 332199015

google-research/seed_rl,google-research/seed_rl,google-research/seed_rl