xla

6e9b663c - Revert "Apply sharding when creating tensors data in data loader (#4995)" (#5012)