experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix documentation for connect_to_remote_host

Browse Source 
PiperOrigin-RevId: 293279392
Change-Id: I8915b62683b74673c8d39d2c3c54f30ba4021c9c
This commit is contained in:
Hongkun Yu 2020-02-04 19:15:30 -08:00 committed by TensorFlower Gardener
parent 645b10fbde
commit de36b1aac7
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/python/eager/remote.py
View File

@ -49,9 +49,8 @@ def connect_to_remote_host(remote_host=None, job_name="worker"):
Using the default job_name of worker, you can schedule ops to run remotely as
follows:
```python
# Enable eager execution, and connect to the remote host.
tf.compat.v1.enable_eager_execution()
tf.contrib.eager.connect_to_remote_host("exampleaddr.com:9876")
# When eager execution is enabled, connect to the remote host.
tf.config.experimental_connect_to_host("exampleaddr.com:9876")
with ops.device("job:worker/replica:0/task:1/device:CPU:0"):
# The following tensors should be resident on the remote device, and the op