xla

d7b80a4e - Add ShardingSpec to XLATensor when it is created with a PJRTShardedData