| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_cache.cc | 41 GrpcWorkerEnv* worker_env) in GrpcWorkerCache() argument
45 worker_env_(worker_env), in GrpcWorkerCache()
163 const string& local_target, GrpcWorkerEnv* worker_env) { in NewGrpcWorkerCacheWithLocalWorker() argument
164 return new GrpcWorkerCache(cc, local_worker, local_target, worker_env); in NewGrpcWorkerCacheWithLocalWorker()
|
| D | rpc_rendezvous_mgr.h | 50 BaseRemoteRendezvous* Create(int64 step_id, const WorkerEnv* worker_env);
|
| D | grpc_worker_cache.h | 70 const string& local_target, GrpcWorkerEnv* worker_env);
|
| D | rpc_rendezvous_mgr.cc | 290 const WorkerEnv* worker_env) { in Create() argument
291 return new RpcRemoteRendezvous(worker_env, step_id); in Create()
|
| D | grpc_worker_service.h | 70 std::unique_ptr<GrpcWorker> NewGrpcWorker(WorkerEnv* worker_env,
|
| D | BUILD | 202 "//tensorflow/core/distributed_runtime:worker_env",
279 "//tensorflow/core/distributed_runtime:worker_env",
316 "//tensorflow/core/distributed_runtime:worker_env",
|
| D | grpc_server_lib.h | 96 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/c/experimental/ |
| D | BUILD | 33 "//tensorflow/core/distributed_runtime:worker_env",
70 "//tensorflow/core/distributed_runtime:worker_env",
117 "//tensorflow/core/distributed_runtime:worker_env",
|
| D | rendezvous_internal.h | 116 const WorkerEnv* worker_env) override { in Create() argument
118 worker_env, step_id, in Create()
|
| D | network_test.cc | 171 TF_EXPECT_OK(grpc_server->worker_env()->session_mgr->CreateSession( in InitializeRendezvous()
175 TF_EXPECT_OK(grpc_server->worker_env()->session_mgr->WorkerSessionForSession( in InitializeRendezvous()
226 auto* rendezvous_mgr = grpc_server->worker_env()->rendezvous_mgr; in TEST_F()
|
| /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 | BUILD | 155 ":worker_env",
166 name = "worker_env",
167 hdrs = ["worker_env.h"],
377 ":worker_env",
403 ":worker_env",
431 ":worker_env",
798 "worker_env.h",
|
| D | base_rendezvous_mgr.h | 62 explicit BaseRendezvousMgr(const WorkerEnv* worker_env);
96 const WorkerEnv* worker_env) = 0;
|
| 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);
|
| D | base_rendezvous_mgr.cc | 45 BaseRendezvousMgr::BaseRendezvousMgr(const WorkerEnv* worker_env) in BaseRendezvousMgr() argument
46 : worker_env_(worker_env) {} in BaseRendezvousMgr()
|
| D | graph_mgr.cc | 58 GraphMgr::GraphMgr(const WorkerEnv* worker_env, DeviceMgr* device_mgr) in GraphMgr() argument
59 : worker_env_(worker_env), device_mgr_(device_mgr), table_(5) { in GraphMgr()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | BUILD | 119 "//tensorflow/core/distributed_runtime:worker_env",
149 "//tensorflow/core/distributed_runtime:worker_env",
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.h | 316 std::unique_ptr<ServerInterface> server, WorkerEnv* worker_env,
333 WorkerEnv* worker_env,
556 std::unique_ptr<ServerInterface> server, WorkerEnv* worker_env,
|
| D | context.cc | 914 std::unique_ptr<ServerInterface> server, WorkerEnv* worker_env, in InitializeRemoteMaster() argument
935 std::move(server), worker_env, std::move(worker_session), in InitializeRemoteMaster()
942 WorkerEnv* worker_env, in UpdateRemoteMaster() argument
983 worker_env_ = worker_env; in UpdateRemoteMaster()
1018 std::unique_ptr<ServerInterface> server, WorkerEnv* worker_env, in SetMasterContextState() argument
1049 worker_env_ = worker_env; in SetMasterContextState()
|
| D | BUILD | 60 "//tensorflow/core/distributed_runtime:worker_env",
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 578 grpc_server->worker_env()->device_mgr->ListDeviceAttributes( in UpdateTFE_ContextWithServerDef()
622 grpc_server->worker_env()->rendezvous_mgr->Find(context_id); in UpdateTFE_ContextWithServerDef()
624 auto* device_mgr = grpc_server->worker_env()->device_mgr; in UpdateTFE_ContextWithServerDef()
628 TF_RETURN_IF_ERROR(grpc_server->worker_env()->session_mgr->CreateSession( in UpdateTFE_ContextWithServerDef()
632 grpc_server->worker_env()->session_mgr->WorkerSessionForSession( in UpdateTFE_ContextWithServerDef()
645 std::move(new_server), grpc_server->worker_env(), worker_session, in UpdateTFE_ContextWithServerDef()
655 grpc_server->worker_env()->session_mgr->UpdateSession( in UpdateTFE_ContextWithServerDef()
659 grpc_server->worker_env()->session_mgr->WorkerSessionForSession( in UpdateTFE_ContextWithServerDef()
665 grpc_server->worker_env(), std::move(remote_eager_workers), in UpdateTFE_ContextWithServerDef()
|
| D | BUILD | 83 "//tensorflow/core/distributed_runtime:worker_env",
260 "//tensorflow/core/distributed_runtime:worker_env",
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental.cc | 785 std::move(new_server), grpc_server->worker_env()->device_mgr, in EnableCollectiveOps()
786 grpc_server->worker_env()->collective_executor_mgr)); in EnableCollectiveOps()
790 /*new_server=*/nullptr, grpc_server->worker_env()->device_mgr, in EnableCollectiveOps()
791 grpc_server->worker_env()->collective_executor_mgr)); in EnableCollectiveOps()
|