jax
19494137 - Increase sharding of checkify_test on TPU to fix CI flakes.

Commit

1 year ago

Increase sharding of checkify_test on TPU to fix CI flakes. PiperOrigin-RevId: 678720498

Author

hawkinsp

Committer

Google-ML-Automation

Parents