jax
c057c7cd - Remove `DeviceLocalLayout` and `.device_local_layout` aliases.

Commit

1 year ago

Remove `DeviceLocalLayout` and `.device_local_layout` aliases. PiperOrigin-RevId: 774604723

References

Author

yashk2810

Committer

Google-ML-Automation

Parents