xla
[Distributed] Switch all_reduce to use the new functional collective op
#6887
Merged
Merged