xla

[Distributed] Move the cached all reduce token to C++

#4912

Merged