xla
e2abcaf4 - [SPMD] Preserve parameter sharding with output data sharding (#4721)

Commit

2 years ago

[SPMD] Preserve parameter sharding with output data sharding (#4721) [SPMD] Persist tensor sharding with XLA sharding propagation

References

Author

yeounoh

Parents