View source on GitHub |
Connects to a single machine to enable remote execution on it.
tf.config.experimental_connect_to_host( remote_host=None, job_name='worker' )
Will make devices on the remote host available to use. Note that calling this more than once will work, but will invalidate any tensor handles on the old remote devices.
Using the default job_name of worker, you can schedule ops to run remotely as follows:
# 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 # will also execute remotely. x1 = array_ops.ones([2, 2]) x2 = array_ops.ones([2, 2]) y = math_ops.matmul(x1, x2)
Args | |
---|---|
remote_host | a single or a list the remote server addr in host-port format. |
job_name | The job name under which the new server will be accessible. |
Raises | |
---|---|
ValueError | if remote_host is None. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/config/experimental_connect_to_host