From a479cf46c336bf63ea1f9524772190493820dfd0 Mon Sep 17 00:00:00 2001 From: Sourabh Bajaj Date: Wed, 27 Feb 2019 14:35:10 -0800 Subject: [PATCH] Support hosts passed with grpc:// PiperOrigin-RevId: 235994227 --- tensorflow/python/eager/remote.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tensorflow/python/eager/remote.py b/tensorflow/python/eager/remote.py index fdea95fa803..5a01e3da1ae 100644 --- a/tensorflow/python/eager/remote.py +++ b/tensorflow/python/eager/remote.py @@ -58,6 +58,11 @@ def connect_to_remote_host(remote_host=None, job_name="worker"): """ if remote_host is None: raise ValueError("Must provide an remote_host") + + grpc_prefix = "grpc://" + if remote_host.startswith(grpc_prefix): + remote_host = remote_host[len(grpc_prefix):] + cluster_def = ClusterDef() job_def = cluster_def.job.add() job_def.name = job_name