jax
382b3e02 - fix-forward for pallas tpu memory spaces test

Commit

1 year ago

fix-forward for pallas tpu memory spaces test PiperOrigin-RevId: 770937094

References

ci-upstream-sync-215_1

#29450 - fix-forward for pallas tpu memory spaces test

Author

sharadmv

sharadmv

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading