jax
496cde6e
- [jax2tf] Add special case for translation of lax.gather to tf.gather. (#3486)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
5 years ago
[jax2tf] Add special case for translation of lax.gather to tf.gather. (#3486) Also adds more tests for conversion for gather.
References
#3486 - [jax2tf] Add special case for translation of lax.gather to tf.gather.
Author
gnecula
Parents
ccb640af
Loading