xla
[SPMD] Support partial replication.
#4788
Merged

Commits
  • [SPMD] Support partial replication.
    yeounoh committed 3 years ago
  • * [SPMD] index local shards correctly in the execution arguments for replication.
    yeounoh committed 3 years ago
  • Update partial replication tests & bug fix
    yeounoh committed 3 years ago
  • linter fix
    yeounoh committed 3 years ago
Loading