xla
[SPMD] Set replicated output false
#5139
Merged

Commits
  • * [SPMD] index local shards correctly in the execution arguments for replication.
    yeounoh committed 3 years ago
  • linter fix
    yeounoh committed 3 years ago
  • Extend partial replication test
    yeounoh committed 3 years ago
  • [SPMD] output sharding by default
    yeounoh committed 3 years ago
Loading