| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| rpc_rendezvous_mgr.h | 20 #include "tensorflow/core/distributed_runtime/worker_env.h"
34 // Rendezvous* rendez = worker_env->rendezvous_mgr->Find(0x8935);
50 BaseRemoteRendezvous* Create(int64 step_id, const WorkerEnv* worker_env);
|
| grpc_worker_service.h | 52 std::unique_ptr<GrpcWorker> NewGrpcWorker(WorkerEnv* worker_env);
|
| grpc_server_lib.h | 30 #include "tensorflow/core/distributed_runtime/worker_env.h"
100 WorkerEnv* worker_env() { return &worker_env_; } function in class:tensorflow::GrpcServer
|
| rpc_rendezvous_mgr.cc | 269 const WorkerEnv* worker_env) {
270 return new RpcRemoteRendezvous(worker_env, step_id);
|
| grpc_worker_service.cc | 357 GrpcWorker::GrpcWorker(WorkerEnv* worker_env)
358 : Worker(worker_env), recv_tensor_recent_request_ids_(100000) {}
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| gdr_rendezvous_mgr.h | 21 #include "tensorflow/core/distributed_runtime/worker_env.h"
32 BaseRemoteRendezvous* Create(int64 step_id, const WorkerEnv* worker_env);
|
| gdr_worker.cc | 42 GdrWorker::GdrWorker(WorkerEnv* worker_env,
44 : GrpcWorker(worker_env),
|
| gdr_server_lib.cc | 68 gdr_thread_.reset(worker_env()->env->StartThread(
|
| gdr_rendezvous_mgr.cc | 203 const WorkerEnv* worker_env) {
204 return new GdrRemoteRendezvous(worker_env, step_id, remote_memory_manager_);
|
| /external/tensorflow/tensorflow/contrib/verbs/ |
| rdma_rendezvous_mgr.h | 23 #include "tensorflow/core/distributed_runtime/worker_env.h"
35 // Rendezvous* rendez = worker_env->rendezvous_mgr->Find(0x8935);
53 const WorkerEnv* worker_env) override;
|
| grpc_verbs_service.h | 38 GrpcVerbsService(const WorkerEnv* worker_env, ::grpc::ServerBuilder* builder);
66 void SetNewVerbsService(GrpcVerbsService** handle, const WorkerEnv* worker_env,
|
| rdma_mgr.h | 26 #include "tensorflow/core/distributed_runtime/worker_env.h"
35 explicit RdmaMgr(const WorkerEnv* const worker_env,
|
| verbs_server_lib.cc | 86 rdma_mgr_ = new RdmaMgr(worker_env(), channel_cache_);
89 dynamic_cast<RdmaRendezvousMgr*>(worker_env()->rendezvous_mgr)
102 verbs_thread_.reset(worker_env()->env->StartThread(
130 ServiceInitFunction service_func = [&ret](const WorkerEnv* worker_env,
132 return SetNewVerbsService(&ret->verbs_service_, worker_env, builder);
|
| grpc_verbs_service.cc | 28 GrpcVerbsService::GrpcVerbsService(const WorkerEnv* worker_env,
30 : is_shutdown_(false), worker_env_(worker_env) {
160 void SetNewVerbsService(GrpcVerbsService** handle, const WorkerEnv* worker_env,
162 *handle = new GrpcVerbsService(worker_env, builder);
|
| rdma_rendezvous_mgr.cc | 86 const WorkerEnv* worker_env) {
87 return new RdmaRemoteRendezvous(worker_env, step_id, rdma_mgr_);
|
| rdma_mgr.cc | 33 RdmaMgr::RdmaMgr(const WorkerEnv* const worker_env,
35 : worker_env_(worker_env), channel_cache_(channel_cache) {
|
| rdma.h | 31 #include "tensorflow/core/distributed_runtime/worker_env.h"
379 RdmaAdapter(const WorkerEnv* worker_env);
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| session_mgr.cc | 31 WorkerEnv* worker_env, const string& default_worker_name,
34 : worker_env_(worker_env),
40 std::unique_ptr<DeviceMgr>(worker_env->device_mgr),
42 new GraphMgr(worker_env, worker_env->device_mgr)))),
|
| base_rendezvous_mgr.h | 23 #include "tensorflow/core/distributed_runtime/worker_env.h"
49 // Rendezvous* rendez = worker_env->rendezvous_mgr->Find(0x8935);
62 explicit BaseRendezvousMgr(const WorkerEnv* worker_env);
96 const WorkerEnv* worker_env) = 0;
|
| session_mgr.h | 43 WorkerEnv* worker_env, const string& default_worker_name,
|
| graph_mgr.h | 26 #include "tensorflow/core/distributed_runtime/worker_env.h"
60 // GraphMgr gmgr(worker_env);
71 explicit GraphMgr(const WorkerEnv* worker_env, DeviceMgr* device_mgr);
|
| base_rendezvous_mgr.cc | 43 BaseRendezvousMgr::BaseRendezvousMgr(const WorkerEnv* worker_env)
44 : worker_env_(worker_env) {}
|
| graph_mgr.cc | 52 GraphMgr::GraphMgr(const WorkerEnv* worker_env, DeviceMgr* device_mgr)
53 : worker_env_(worker_env), device_mgr_(device_mgr), table_(5) {
|
| /external/tensorflow/tensorflow/contrib/mpi/ |
| mpi_rendezvous_mgr.cc | 58 const WorkerEnv* worker_env) {
59 return new MPIRemoteRendezvous(worker_env, step_id, mpiutils_, this);
|
| mpi_rendezvous_mgr.h | 38 #include "tensorflow/core/distributed_runtime/worker_env.h"
159 const WorkerEnv* worker_env) override;
|
