xla

Apply sharding when creating tensors data in data loader

#4995

Merged