Commit dbd92888 authored by Chris Leary's avatar Chris Leary Committed by TensorFlower Gardener
Browse files

[XLA] Add CrossReplicaSum to Python API bindings. 

Add support to CPU for the trivial replica_count=1 case.

PiperOrigin-RevId: 179756956
parent ff44881e
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment