jax
9bb6f185 - Summarize equations in sharding mismatch errors lazily.

Commit

2 years ago

Summarize equations in sharding mismatch errors lazily. Don't spend time building error output unless there is an error. PiperOrigin-RevId: 606812191

Author

hawkinsp

Committer

a-googler

Parents