xla
18ed9dd8 - Add shard_map wrapper (#7670)

Commit
1 year ago
Add shard_map wrapper (#7670) Co-authored-by: Siyuan Liu <lsiyuan@google.coim>
Author
Parents
Loading