jax
80f18ded
- [sharding_in_types] Make slice and ellipsis work with `.at[...].get(out_sharding=P(...))`
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
319 days ago
[sharding_in_types] Make slice and ellipsis work with `.at[...].get(out_sharding=P(...))` PiperOrigin-RevId: 729723470
References
#26677 - [sharding_in_types] Make slice and ellipsis work with `.at[...].get(out_sharding=P(...))`
#240 - CI: 02/25/25 upstream sync
Author
yashk2810
Committer
Google-ML-Automation
Parents
629426f8
Loading