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

Correctly use the aggregation mode set for variables in 

ParameterServerStrategy when using >1 device per machine. This means
wrapping the variable instances returned in that case in a class
that intercepts assign_*() method calls.

PiperOrigin-RevId: 209533673
parent 3f78692c
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