jax
037bc5ed - [Pallas] Add support for trivial vmap of scalar prefetch

Commit

2 years ago

[Pallas] Add support for trivial vmap of scalar prefetch PiperOrigin-RevId: 600898742

References

Author

sharadmv

Committer

a-googler

Parents