Commit 55a4b150 authored by Ayush Dubey's avatar Ayush Dubey Committed by TensorFlower Gardener
Browse files

Enable user to choose all-reduce algorithm for multi-worker MirroredStrategy. 

Currently, MirroredStrategy defaults to "pscpu/pscpu" when run for multiple
workers.  This change enables passing in a MultiWorkerAllReduce object to
MirroredStrategy, which would allow the user to specify one of the many
all-reduce specs that MirroredStrategy supports.

PiperOrigin-RevId: 217785932
parent 76e78044
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