xla
[SPMD] Preserve parameter sharding with output data sharding
#4721

Merged

Commits

[SPMD] Persist tensor sharding with XLA sharding propagation.

yeounoh committed 3 years ago
Add tests

yeounoh committed 3 years ago
Add more comments

yeounoh committed 3 years ago
Refactoring

yeounoh committed 3 years ago

Loading