jax
0ec9f3c2 - Fix the bug when lax.gather uses dynamic_gather.

Commit
2 years ago
Fix the bug when lax.gather uses dynamic_gather. PiperOrigin-RevId: 538693533
Author
Committer
Parents
Loading