SdyDim` since these are not `Sharding` from a JAX POV.

Commit

1 year ago

Rename `SdyArraySharding -> SdyArray` and `SdyDimSharding -> SdyDim` since these are not `Sharding` from a JAX POV. PiperOrigin-RevId: 758679923

References

Author

yashk2810

Committer

Google-ML-Automation

Parents