Specify a device for a SavedModelPolicy.
Specify a device for a SavedModelPolicy.

PiperOrigin-RevId: 450025488
Change-Id: Ic52ef52f2e1c87a1f9afa1c43c7a1d5e9c283739

deepmind/meltingpot,deepmind/meltingpot