jax
ef680634 - Merge pull request #27809 from mattjj:26621

Commit

1 year ago

Merge pull request #27809 from mattjj:26621 PiperOrigin-RevId: 745212009

References

#27809 - [shard-map] canonicalize rep=None to be rep={all possible axes}

Author

Google-ML-Automation

Google-ML-Automation

Parents

Loading