xla

59570c77 - Revert "Allow MpDeviceLoader to shard dictionaries of tensors" (#7964)