xla

e3cf356a - Raise error when minibatch is used in SPMD dataloading and per host b… (#8205)