jax
840cf3f7
- [sharding_in_types] Add `pad_p` support to sharding_in_types to handle transpose to slice correctly.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
[sharding_in_types] Add `pad_p` support to sharding_in_types to handle transpose to slice correctly. PiperOrigin-RevId: 698573396
References
#25021 - [sharding_in_types] Add `pad_p` support to sharding_in_types to handle transpose to slice correctly.
#148 - CI: 11/22/24 upstream sync
Author
yashk2810
Committer
Google-ML-Automation
Parents
1f6152d1
Loading