Commit d75dde91 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Change implementation of DistributedValues to be a device map and 

tuple of values instead of a dict mapping devices to values.
Values in a DistributedValues object are now indexed by replica id.
Lays the groundwork for supporting multiple logical devices
(model parallelism).

PiperOrigin-RevId: 226108301
parent f73a4f99
Expand all 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