diff --git a/tensorflow/python/training/device_setter.py b/tensorflow/python/training/device_setter.py index 6bf5d8aca1f..e3e94a6b6a1 100644 --- 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):