diff options
Diffstat (limited to 'tensorflow/contrib/training/python/training/device_setter.py')
-rw-r--r-- | tensorflow/contrib/training/python/training/device_setter.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/training/python/training/device_setter.py b/tensorflow/contrib/training/python/training/device_setter.py index ae6ffb8f28..53f6fa5436 100644 --- a/tensorflow/contrib/training/python/training/device_setter.py +++ b/tensorflow/contrib/training/python/training/device_setter.py @@ -42,7 +42,7 @@ class GreedyLoadBalancingStrategy(object): off CPU-intensive ops with RAM-intensive ops with network bandwidth. This class is intended to be used as a `ps_strategy` in - `tf.replica_device_setter`. + `tf.train.replica_device_setter`. """ def __init__(self, num_tasks, load_fn): |