From 5756b52f1446652c8fb8c6fe954eac554ddceea4 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Wed, 28 Aug 2019 13:40:59 -0700
Subject: [PATCH] Tensorforest: adding resource to the list of ops to be
 created on PS PiperOrigin-RevId: 265982902

---
 tensorflow/python/training/device_setter.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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):