diff options
| author |  Priya Gupta <priyag@google.com> | 2018-07-31 18:41:28 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-31 18:45:54 -0700 |
| commit | ffc9be66bc1503ec12f10f43ff622d9193062e97 (patch) | |
| tree | d74f8057b6dd391ceb533db7a8decd8b0d88e86e /tensorflow/contrib/distribute/python/values.py | |
| parent | a738649f8092dda124b6a4f3ccc31bf4159651ea (diff) | |
Support distribution strategies in `Estimator.evaluate`.
PiperOrigin-RevId: 206864512
Diffstat (limited to 'tensorflow/contrib/distribute/python/values.py')
| -rw-r--r-- | tensorflow/contrib/distribute/python/values.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/contrib/distribute/python/values.py b/tensorflow/contrib/distribute/python/values.py index 4018b1e023..f4da91a8ac 100644 --- a/tensorflow/contrib/distribute/python/values.py +++ b/tensorflow/contrib/distribute/python/values.py @@ -294,6 +294,9 @@ class DistributedVariable(DistributedDelegate): self._primary_var.op.type) return self.get().op + def read_value(self): + return distribute_lib.get_distribution_strategy().read_var(self) + def _should_act_as_resource_variable(self): """Pass resource_variable_ops.is_resource_variable check.""" pass |