diff options
author | 2018-03-29 09:43:19 -0700 | |
---|---|---|
committer | 2018-03-29 09:48:47 -0700 | |
commit | a2b6c3c124664d682094a1ecfa9cc00cca8ada85 (patch) | |
tree | 1f72bb1a6e6542bfd5e215b535e46b9e782459d5 /tensorflow/python/training/device_setter.py | |
parent | ae3d20f9aef78554f0d0f5eec13982e9802a45d2 (diff) |
Added kernels and estimators for Gradient Boosting Trees algorithm.
BoostedTreesClassifier and BoostedTreesRegressor are added to tf.estimator.
Also some training utility functions are added to tf.contrib.estimator.
PiperOrigin-RevId: 190942599
Diffstat (limited to 'tensorflow/python/training/device_setter.py')
-rw-r--r-- | tensorflow/python/training/device_setter.py | 13 |
1 files changed, 6 insertions, 7 deletions
diff --git a/tensorflow/python/training/device_setter.py b/tensorflow/python/training/device_setter.py index d31c375b4c..be80c36571 100644 --- a/tensorflow/python/training/device_setter.py +++ b/tensorflow/python/training/device_setter.py @@ -25,14 +25,13 @@ from tensorflow.python.platform import tf_logging as logging from tensorflow.python.training import server_lib from tensorflow.python.util.tf_export import tf_export -# This is a tuple of PS ops used by tf.estimator.Esitmator which should work in +# This is a tuple of PS ops used by tf.estimator.Estimator which should work in # almost all of cases. -STANDARD_PS_OPS = ( - "Variable", "VariableV2", "AutoReloadVariable", "MutableHashTable", - "MutableHashTableV2", "MutableHashTableOfTensors", - "MutableHashTableOfTensorsV2", "MutableDenseHashTable", - "MutableDenseHashTableV2", "VarHandleOp" -) +STANDARD_PS_OPS = ("Variable", "VariableV2", "AutoReloadVariable", + "MutableHashTable", "MutableHashTableV2", + "MutableHashTableOfTensors", "MutableHashTableOfTensorsV2", + "MutableDenseHashTable", "MutableDenseHashTableV2", + "VarHandleOp", "BoostedTreesEnsembleResourceHandleOp") class _RoundRobinStrategy(object): |