jax
1194174f - [pltpu.emit_pipeline] Support TransformedRefs in async prefetch.

Commit

10 days ago

[pltpu.emit_pipeline] Support TransformedRefs in async prefetch. PiperOrigin-RevId: 981524078

References

Author

rdyro

Committer

Google-ML-Automation

Parents