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

Add a protected getter for host_name field in GrpcServer

Browse Source 
PiperOrigin-RevId: 356420826
Change-Id: I461be80631878ace47ca107fee304957df12837e
This commit is contained in:
Isha Arkatkar 2021-02-08 21:18:28 -08:00 committed by TensorFlower Gardener
parent a0b5d0bacb
commit 6fd9628434
2 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/core/distributed_runtime/rpc/grpc_server_lib.cc
View File

@ -134,9 +134,9 @@ Status GrpcServer::GetHostAndPort(const ServerDef& server_def,
if (job.name() == server_def.job_name()) {
auto iter = job.tasks().find(server_def.task_index());
if (iter == job.tasks().end()) {
return errors::InvalidArgument("Task ", server_def.task_index(),
" was not defined in job \"",
server_def.job_name(), "\"");
return errors::Internal("Task ", server_def.task_index(),
" was not defined in job \"",
server_def.job_name(), "\"");
}
if (server_def.port() != 0) {

3
tensorflow/core/distributed_runtime/rpc/grpc_server_lib.h
View File

@ -136,6 +136,9 @@ class GrpcServer : public ServerInterface {
// This method may only be called after `this->Init()` returns successfully.
int bound_port() const { return bound_port_; }
// Returns hostname.
const string& host_name() const { return host_name_; }
const ServerDef& server_def() const { return server_def_; }
GrpcWorker* worker_impl() const { return worker_impl_.get(); }
GrpcWorkerEnv* grpc_worker_env() const { return grpc_worker_env_.get(); }