jax
eb194e9a
- Replace `jax.stages.OutInfo` with `jax.ShapeDtypeStruct`
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
Replace `jax.stages.OutInfo` with `jax.ShapeDtypeStruct` PiperOrigin-RevId: 784729142
References
#30318 - Replace `jax.stages.OutInfo` with `jax.ShapeDtypeStruct`
Author
yashk2810
Committer
Google-ML-Automation
Parents
c5c89953
Loading
