jax
5cef547e - Added support for `lax.cond_p` to Pallas Mosaic GPU lowering

Commit

1 year ago

Added support for `lax.cond_p` to Pallas Mosaic GPU lowering PiperOrigin-RevId: 679156819

Author

superbobry

Committer

Google-ML-Automation

Parents