jax
8f706931
- Add trivial batching rule for tpu custom call
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
192 days ago
Add trivial batching rule for tpu custom call PiperOrigin-RevId: 794354666
References
#30970 - Add trivial batching rule for tpu custom call
Author
sharadmv
Committer
Google-ML-Automation
Parents
e00df525
Loading
