xla
Fix bug in ShardTensor to generate correct shards for each device
#4549
Merged

Loading