Tensorforest: adding resource to the list of ops to be created on PS

PiperOrigin-RevId: 265982902
2019-08-28 13:40:59 -07:00 · 2019-08-28 13:40:59 -07:00 · 5756b52f14
commit 5756b52f14
parent f3ceeb53eb
1 changed files with 2 additions and 1 deletions
--- a/tensorflow/python/training/device_setter.py
+++ b/tensorflow/python/training/device_setter.py
@ -33,7 +33,8 @@ STANDARD_PS_OPS = ("Variable", "VariableV2", "AutoReloadVariable",
                   "MutableDenseHashTable", "MutableDenseHashTableV2",
                   "VarHandleOp", "BoostedTreesEnsembleResourceHandleOp",
                   "BoostedTreesQuantileStreamResourceHandleOp",
-                   "ResourceConditionalAccumulator")
+                   "ResourceConditionalAccumulator",
+                   "DecisionTreeResource")


 class _RoundRobinStrategy(object):