xla

df0a2776 - Add all_to_all back to our doc (#2472)