diff options
Diffstat (limited to 'tensorflow/python/training/device_setter.py')
| -rw-r--r-- | tensorflow/python/training/device_setter.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/training/device_setter.py b/tensorflow/python/training/device_setter.py index 8a1dadccf4..7f403f4927 100644 --- a/tensorflow/python/training/device_setter.py +++ b/tensorflow/python/training/device_setter.py @@ -148,7 +148,7 @@ def replica_device_setter(ps_tasks=0, ps_device="/job:ps", cluster_spec = { "ps": ["ps0:2222", "ps1:2222"], "worker": ["worker0:2222", "worker1:2222", "worker2:2222"]} - with tf.device(tf.replica_device_setter(cluster=cluster_spec)): + with tf.device(tf.train.replica_device_setter(cluster=cluster_spec)): # Build your graph v1 = tf.Variable(...) # assigned to /job:ps/task:0 v2 = tf.Variable(...) # assigned to /job:ps/task:1 |