jax
90401d51 - Accept layout on `ShapeDtypeStruct` on the `sharding` argument. `DeviceLocalLayout.AUTO` is not allowed on SDS.
Commit
View On GitHub
Commit
2 years ago
Accept layout on `ShapeDtypeStruct` on the `sharding` argument. `DeviceLocalLayout.AUTO` is not allowed on SDS. PiperOrigin-RevId: 624982814
Author
yashk2810 yashk2810
Committer
a-googler a-googler
Parents
b7293005
FAQ Terms Privacy Refunds Impressum
Loading