jax
c0811c9d - Adds coverage for spmd-axisname-filtering in shard_map transpose.

Commit

1 year ago

Adds coverage for spmd-axisname-filtering in shard_map transpose. PiperOrigin-RevId: 699193349

References

Author

jkr26

Committer

Google-ML-Automation

Parents