diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-08-24 09:12:59 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-24 09:18:02 -0700 |
| commit | 6f00ab7b8f16f450c00375df271c45da4dc72be5 (patch) | |
| tree | 25163f5e9d441de60df20f1e990758271cbcf316 /tensorflow/contrib/distribute/python/values.py | |
| parent | 247b81a7c47fe52a383c86a9a32efa536ead6fa6 (diff) | |
For ParameterServerStrategy, make sure to include the AggregatingVariable
wrapper for variables in collections instead of what it wraps.
PiperOrigin-RevId: 210107528
Diffstat (limited to 'tensorflow/contrib/distribute/python/values.py')
| -rw-r--r-- | tensorflow/contrib/distribute/python/values.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/contrib/distribute/python/values.py b/tensorflow/contrib/distribute/python/values.py index a58bb3a849..1b9fdef5b0 100644 --- a/tensorflow/contrib/distribute/python/values.py +++ b/tensorflow/contrib/distribute/python/values.py @@ -1180,6 +1180,10 @@ class AggregatingVariable(checkpointable.CheckpointableBase): def __repr__(self): return repr(self._v) + def _should_act_as_resource_variable(self): + """Pass resource_variable_ops.is_resource_variable check.""" + pass + # Register a conversion function which reads the value of the variable, # allowing instances of the class to be used as tensors. |