xla

8ff43a9b - Allow MpDeviceLoader to shard dictionaries of tensor for 2.5 release (#8212)