jax
853c4702 - Improve the repr of NamedSharding and error message of device_put

Commit

2 years ago

Improve the repr of NamedSharding and error message of device_put PiperOrigin-RevId: 552841710

Author

yashk2810

Committer

a-googler

Parents