jax
d9797a48 - Make the device check works when the number of devices is larger than one.

Commit

2 years ago

Make the device check works when the number of devices is larger than one. PiperOrigin-RevId: 538760855

Author

a-googler

Committer

a-googler

Parents