xla

7e3efc58 - Allgather coalescee: Check tuple shape only if return shape is tuple. (#9403)