jax
bf0e7358
- Merge pull request #21980 from vfdev-5:add-device-kwarg-to-jnp-ops
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
Merge pull request #21980 from vfdev-5:add-device-kwarg-to-jnp-ops PiperOrigin-RevId: 647091110
References
#21980 - Added device to `jnp.arange`, `jnp.eye` and tests
Author
a-googler
Parents
43c9fe89
78ee8a52
Loading
