xla

9a5fdf39 - Enabling sharding device data IR (#5475)

Enabling sharding device data IR (#5475) * Allow shard device data IR * Handle XLATensor that is DeviceData IR and does not have XLAData * fix typo