| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_cache.cc | 37 GrpcWorkerEnv* worker_env) in GrpcWorkerCache() argument
41 worker_env_(worker_env), in GrpcWorkerCache()
164 GrpcWorkerEnv* worker_env) { in NewGrpcWorkerCache() argument
166 worker_env); in NewGrpcWorkerCache()
170 std::shared_ptr<GrpcChannelCache> cc, GrpcWorkerEnv* worker_env, in NewGrpcWorkerCacheWithLocalWorker() argument
172 return new GrpcWorkerCache(cc, local_worker, local_target, worker_env); in NewGrpcWorkerCacheWithLocalWorker()
|
| D | grpc_worker_cache.h | 69 GrpcWorkerEnv* worker_env);
72 std::shared_ptr<GrpcChannelCache> cc, GrpcWorkerEnv* worker_env,
|
| D | rpc_rendezvous_mgr.h | 50 BaseRemoteRendezvous* Create(int64 step_id, const WorkerEnv* worker_env);
|
| D | grpc_worker_service.h | 70 std::unique_ptr<GrpcWorker> NewGrpcWorker(WorkerEnv* worker_env,
|
| D | rpc_rendezvous_mgr.cc | 314 const WorkerEnv* worker_env) { in Create() argument
315 return new RpcRemoteRendezvous(worker_env, step_id); in Create()
|
| D | BUILD | 224 "//tensorflow/core/distributed_runtime:worker_env",
303 "//tensorflow/core/distributed_runtime:worker_env",
339 "//tensorflow/core/distributed_runtime:worker_env",
|
| D | grpc_server_lib.h | 104 WorkerEnv* worker_env() { return &worker_env_; } in worker_env() function
|
| D | grpc_worker_service.cc | 421 GrpcWorker::GrpcWorker(WorkerEnv* worker_env, const ConfigProto& config) in GrpcWorker() argument
422 : Worker(worker_env), in GrpcWorker()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context_distributed_manager.cc | 542 grpc_server->worker_env()->device_mgr->ListDeviceAttributes( in UpdateContextWithServerDef()
607 grpc_server->worker_env()->rendezvous_mgr->Find(context_id); in UpdateContextWithServerDef()
608 auto* device_mgr = grpc_server->worker_env()->device_mgr; in UpdateContextWithServerDef()
611 grpc_server->worker_env()->session_mgr->CreateSession( in UpdateContextWithServerDef()
615 grpc_server->worker_env()->session_mgr->WorkerSessionForSession( in UpdateContextWithServerDef()
628 std::move(new_server), grpc_server->worker_env(), worker_session, in UpdateContextWithServerDef()
637 sg.Update(grpc_server->worker_env()->session_mgr->UpdateSession( in UpdateContextWithServerDef()
708 std::move(new_server), grpc_server->worker_env()->device_mgr, in EnableCollectiveOps()
709 grpc_server->worker_env()->collective_executor_mgr.get())); in EnableCollectiveOps()
713 /*new_server=*/nullptr, grpc_server->worker_env()->device_mgr, in EnableCollectiveOps()
[all …]
|
| D | context.h | 366 std::unique_ptr<ServerInterface> server, WorkerEnv* worker_env,
668 std::unique_ptr<ServerInterface> server, WorkerEnv* worker_env,
|
| D | context.cc | 1201 std::unique_ptr<ServerInterface> server, WorkerEnv* worker_env, in InitializeRemoteMaster() argument
1225 std::move(server), worker_env, std::move(worker_session), in InitializeRemoteMaster()
1299 std::unique_ptr<ServerInterface> server, WorkerEnv* worker_env, in SetMasterContextState() argument
1336 worker_env_ = worker_env; in SetMasterContextState()
|
| D | BUILD | 99 "//tensorflow/core/distributed_runtime:worker_env",
200 "//tensorflow/core/distributed_runtime:worker_env",
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | session_mgr.cc | 33 WorkerEnv* worker_env, const string& default_worker_name, in SessionMgr() argument
36 : worker_env_(worker_env), in SessionMgr()
42 worker_env->device_mgr, in SessionMgr()
44 new GraphMgr(worker_env, worker_env->device_mgr)), in SessionMgr()
|
| D | base_rendezvous_mgr.h | 62 explicit BaseRendezvousMgr(const WorkerEnv* worker_env);
93 const WorkerEnv* worker_env) = 0;
|
| D | BUILD | 157 ":worker_env",
168 name = "worker_env",
169 hdrs = ["worker_env.h"],
380 ":worker_env",
406 ":worker_env",
435 ":worker_env",
812 "worker_env.h",
|
| D | session_mgr.h | 44 WorkerEnv* worker_env, const string& default_worker_name,
|
| D | graph_mgr.h | 72 explicit GraphMgr(const WorkerEnv* worker_env, const DeviceMgr* device_mgr);
|
| D | base_rendezvous_mgr.cc | 46 BaseRendezvousMgr::BaseRendezvousMgr(const WorkerEnv* worker_env) in BaseRendezvousMgr() argument
47 : worker_env_(worker_env) {} in BaseRendezvousMgr()
|
| D | graph_mgr.cc | 59 GraphMgr::GraphMgr(const WorkerEnv* worker_env, const DeviceMgr* device_mgr) in GraphMgr() argument
60 : worker_env_(worker_env), device_mgr_(device_mgr), table_(5) { in GraphMgr()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | BUILD | 127 "//tensorflow/core/distributed_runtime:worker_env",
157 "//tensorflow/core/distributed_runtime:worker_env",
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | BUILD | 104 "//tensorflow/core/distributed_runtime:worker_env",
1002 "//tensorflow/core/distributed_runtime:worker_env",
|