jax
c9dbcaa9 - Increase shard count for the newly added scaled_dot_test on GPU and TPU.

Commit

239 days ago

Increase shard count for the newly added scaled_dot_test on GPU and TPU. The test times out otherwise. PiperOrigin-RevId: 864996769

References

Author

belitskiy

Committer

Google-ML-Automation

Parents