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

index 63d0f1e1d454354948654e8ad4208a8852d356ca..4b63e08ab3372849309ee5d28d754de82e9632f4 100644 (file)
@@ -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: