jax
e459d3bb - Deprecate jax.lax.infeed and jax.lax.outfeed.

Commit
1 year ago
Deprecate jax.lax.infeed and jax.lax.outfeed. These APIs are already broken on GPU and TPU by virtue of not being implemented in the PJRT C API, so it seems unlikely that they have any users. PiperOrigin-RevId: 746595857
Author
Committer
Parents
Loading