xla
Apply sharding when creating tensors data in data loader
#4995
Merged

Loading