| /external/tensorflow/tensorflow/contrib/verbs/ |
| D | verbs_server_lib.cc | 92 rdma_mgr_ = new RdmaMgr(worker_env(), channel_cache_); in Init()
95 dynamic_cast<RdmaRendezvousMgr*>(worker_env()->rendezvous_mgr) in Init()
108 verbs_thread_.reset(worker_env()->env->StartThread( in Start()
136 ServiceInitFunction service_func = [&ret](const WorkerEnv* worker_env, in Create()
138 return SetNewVerbsService(&ret->verbs_service_, worker_env, builder); in Create()
|
| D | grpc_verbs_service.cc | 28 GrpcVerbsService::GrpcVerbsService(const WorkerEnv* worker_env, in GrpcVerbsService() argument
30 : is_shutdown_(false), worker_env_(worker_env) { in GrpcVerbsService()
160 void SetNewVerbsService(GrpcVerbsService** handle, const WorkerEnv* worker_env, in SetNewVerbsService() argument
162 *handle = new GrpcVerbsService(worker_env, builder); in SetNewVerbsService()
|
| D | grpc_verbs_service.h | 32 GrpcVerbsService(const WorkerEnv* worker_env, ::grpc::ServerBuilder* builder);
60 void SetNewVerbsService(GrpcVerbsService** handle, const WorkerEnv* worker_env,
|
| D | BUILD | 101 "//tensorflow/core/distributed_runtime:worker_env",
117 "//tensorflow/core/distributed_runtime:worker_env",
143 "//tensorflow/core/distributed_runtime:worker_env",
|
| D | rdma_rendezvous_mgr.cc | 86 const WorkerEnv* worker_env) { in Create() argument
87 return new RdmaRemoteRendezvous(worker_env, step_id, rdma_mgr_); in Create()
|
| D | rdma_rendezvous_mgr.h | 53 const WorkerEnv* worker_env) override;
|
| D | rdma_mgr.h | 35 explicit RdmaMgr(const WorkerEnv* const worker_env,
|
| D | rdma_mgr.cc | 35 RdmaMgr::RdmaMgr(const WorkerEnv* const worker_env, in RdmaMgr() argument
37 : worker_env_(worker_env), channel_cache_(channel_cache) { in RdmaMgr()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | session_mgr.cc | 32 WorkerEnv* worker_env, const string& default_worker_name, in SessionMgr() argument
35 : worker_env_(worker_env), in SessionMgr()
41 worker_env->device_mgr, in SessionMgr()
43 new GraphMgr(worker_env, worker_env->device_mgr)))), in SessionMgr()
|
| D | base_rendezvous_mgr.h | 62 explicit BaseRendezvousMgr(const WorkerEnv* worker_env);
96 const WorkerEnv* worker_env) = 0;
|
| D | BUILD | 146 ":worker_env",
157 name = "worker_env",
158 hdrs = ["worker_env.h"],
368 ":worker_env",
394 ":worker_env",
423 ":worker_env",
|
| D | session_mgr.h | 43 WorkerEnv* worker_env, const string& default_worker_name,
|
| D | graph_mgr.h | 72 explicit GraphMgr(const WorkerEnv* worker_env, DeviceMgr* device_mgr);
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | BUILD | 69 "//tensorflow/core/distributed_runtime:worker_env",
92 "//tensorflow/core/distributed_runtime:worker_env",
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | BUILD | 72 "//tensorflow/core/distributed_runtime:worker_env",
104 "//tensorflow/core/distributed_runtime:worker_env",
213 "//tensorflow/core/distributed_runtime:worker_env",
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_worker.cc | 41 GdrWorker::GdrWorker(WorkerEnv* worker_env, const ConfigProto& config, in GdrWorker() argument
43 : GrpcWorker(worker_env, config), in GdrWorker()
|
| D | gdr_rendezvous_mgr.cc | 212 const WorkerEnv* worker_env) { in Create() argument
213 return new GdrRemoteRendezvous(worker_env, step_id, remote_memory_manager_); in Create()
|
| D | gdr_rendezvous_mgr.h | 32 BaseRemoteRendezvous* Create(int64 step_id, const WorkerEnv* worker_env);
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr.h | 50 BaseRemoteRendezvous* Create(int64 step_id, const WorkerEnv* worker_env);
|
| D | rpc_rendezvous_mgr.cc | 294 const WorkerEnv* worker_env) { in Create() argument
295 return new RpcRemoteRendezvous(worker_env, step_id); in Create()
|
| D | grpc_worker_service.h | 61 std::unique_ptr<GrpcWorker> NewGrpcWorker(WorkerEnv* worker_env,
|
| D | BUILD | 196 "//tensorflow/core/distributed_runtime:worker_env",
270 "//tensorflow/core/distributed_runtime:worker_env",
305 "//tensorflow/core/distributed_runtime:worker_env",
|
| D | grpc_server_lib.h | 95 WorkerEnv* worker_env() { return &worker_env_; } in worker_env() function
|
| /external/tensorflow/tensorflow/contrib/mpi/ |
| D | mpi_rendezvous_mgr.cc | 58 const WorkerEnv* worker_env) { in Create() argument
59 return new MPIRemoteRendezvous(worker_env, step_id, mpiutils_, this); in Create()
|
| D | BUILD | 78 "//tensorflow/core/distributed_runtime:worker_env",
|