Commit 34024edf authored by Ayush Dubey's avatar Ayush Dubey Committed by TensorFlower Gardener
Browse files

Enable user to choose between all-reduce implementations in 

MultiWorkerMirroredStrategy.

Possible choices: AUTO, RING - which uses `common_runtime/ring_reducer.{cc,h}`,
and NCCL - which uses Nvidia NCCL for all-reduce.

PiperOrigin-RevId: 236000699
parent 4c6563e4
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