xla

b79fac57 - Fix the test by avoiding the current (will be fixed soon) XLA AllReduce bug on tuple inputs with single element inside.