flax
707a569c - _apply_sharding disallow mixed Explicit/Auto mesh

Commit

18 days ago

_apply_sharding disallow mixed Explicit/Auto mesh PiperOrigin-RevId: 859762099

References

#5199 - _apply_sharding disallow mixed Explicit/Auto mesh

Author

Cristian Garcia

Committer

a-googler

a-googler

Parents

Loading