projects / platform / upstream / tensorflow.git / commitdiff
? search:
summary | shortlog | log | commit | commitdiff | tree
raw | patch | inline | side by side (parent: 893a4a5)
Contrib estimator should handle ResourceVariables properly.
authorAlexandre Passos <apassos@google.com>
Tue, 6 Feb 2018 19:51:47 +0000 (11:51 -0800)
committerTensorFlower Gardener <gardener@tensorflow.org>
Tue, 6 Feb 2018 20:30:57 +0000 (12:30 -0800)
PiperOrigin-RevId: 184715896

tensorflow/contrib/learn/python/learn/estimators/estimator.py patch | blob | history

diff --git a/tensorflow/contrib/learn/python/learn/estimators/estimator.py b/tensorflow/contrib/learn/python/learn/estimators/estimator.py
index 63d0f1e1d454354948654e8ad4208a8852d356ca..4b63e08ab3372849309ee5d28d754de82e9632f4 100644 (file)
--- a/tensorflow/contrib/learn/python/learn/estimators/estimator.py
+++ b/tensorflow/contrib/learn/python/learn/estimators/estimator.py
@@ -211,7 +211,7 @@ def _get_replica_device_setter(config):
       'Variable', 'VariableV2', 'AutoReloadVariable', 'MutableHashTable',
       'MutableHashTableV2', 'MutableHashTableOfTensors',
       'MutableHashTableOfTensorsV2', 'MutableDenseHashTable',
-      'MutableDenseHashTableV2'
+      'MutableDenseHashTableV2', 'VarHandleOp'
   ]
 
   if config.task_type:
Domain: Machine Learning / ML Framework;