Searched refs:WorkerEnv (Results 1 – 23 of 23) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.h | 35 struct WorkerEnv;
41 GrpcWorker(WorkerEnv* env, const ConfigProto& config);
59 WorkerEnv* env();
70 std::unique_ptr<GrpcWorker> NewGrpcWorker(WorkerEnv* worker_env,
|
| D | grpc_server_lib.h | 47 typedef std::function<RendezvousMgrInterface*(const WorkerEnv*)>
52 const ConfigProto&, const WorkerEnv*, WorkerCacheInterface*)>
57 typedef std::function<void(const WorkerEnv*, ::grpc::ServerBuilder*)>
61 typedef std::function<std::unique_ptr<GrpcWorker>(WorkerEnv*,
104 WorkerEnv* worker_env() { return &worker_env_; } in worker_env()
177 WorkerEnv worker_env_;
|
| D | rpc_rendezvous_mgr.h | 47 explicit RpcRendezvousMgr(const WorkerEnv* env);
50 BaseRemoteRendezvous* Create(int64 step_id, const WorkerEnv* worker_env);
|
| D | rpc_rendezvous_mgr.cc | 43 RpcRemoteRendezvous(const WorkerEnv* env, int64 step_id) in RpcRemoteRendezvous()
310 RpcRendezvousMgr::RpcRendezvousMgr(const WorkerEnv* env) in RpcRendezvousMgr()
314 const WorkerEnv* worker_env) { in Create()
|
| D | grpc_worker_service.cc | 421 GrpcWorker::GrpcWorker(WorkerEnv* worker_env, const ConfigProto& config) in GrpcWorker()
741 WorkerEnv* GrpcWorker::env() { return env_; } in env()
749 std::unique_ptr<GrpcWorker> NewGrpcWorker(WorkerEnv* env, in NewGrpcWorker()
|
| D | rpc_rendezvous_mgr_test.cc | 133 WorkerEnv env;
|
| D | grpc_server_lib.cc | 88 RendezvousMgrInterface* NewRpcRendezvousMgr(const WorkerEnv* env) { in NewRpcRendezvousMgr()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | base_rendezvous_mgr.h | 62 explicit BaseRendezvousMgr(const WorkerEnv* worker_env);
93 const WorkerEnv* worker_env) = 0;
100 const WorkerEnv* const worker_env_;
118 BaseRemoteRendezvous(const WorkerEnv* env, int64 step_id);
171 const WorkerEnv* const env_; // Not owned.
|
| D | session_mgr.h | 31 struct WorkerEnv;
44 WorkerEnv* worker_env, const string& default_worker_name,
94 WorkerEnv* const worker_env_; // Not owned.
|
| D | worker.h | 31 struct WorkerEnv;
45 Worker(WorkerEnv* env);
112 WorkerEnv* const env_; // Not owned.
|
| D | graph_mgr.h | 72 explicit GraphMgr(const WorkerEnv* worker_env, const DeviceMgr* device_mgr);
147 const WorkerEnv* worker_env_; // Not owned.
|
| D | worker_env.h | 39 struct WorkerEnv { struct
|
| D | base_rendezvous_mgr.cc | 46 BaseRendezvousMgr::BaseRendezvousMgr(const WorkerEnv* worker_env) in BaseRendezvousMgr()
119 BaseRemoteRendezvous::BaseRemoteRendezvous(const WorkerEnv* env, int64 step_id) in BaseRemoteRendezvous()
|
| D | session_mgr_test.cc | 56 WorkerEnv env_;
|
| D | session_mgr.cc | 33 WorkerEnv* worker_env, const string& default_worker_name, in SessionMgr()
|
| D | worker.cc | 31 Worker::Worker(WorkerEnv* env) : env_(env), recent_request_ids_(100000) { in Worker()
|
| D | graph_mgr.cc | 59 GraphMgr::GraphMgr(const WorkerEnv* worker_env, const DeviceMgr* device_mgr) in GraphMgr()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.h | 40 explicit EagerServiceImpl(const WorkerEnv* env) : env_(env) { in EagerServiceImpl()
119 const WorkerEnv* env) { in CreateMasterContext()
124 int64 destroy_after_secs, const WorkerEnv* env,
160 const WorkerEnv* const env_; // Not owned.
221 const WorkerEnv* const env_; // Not owned.
|
| D | eager_service_impl_test.cc | 49 explicit TestEagerServiceImpl(const WorkerEnv* env) : EagerServiceImpl(env) {} in TestEagerServiceImpl()
165 WorkerEnv worker_env_;
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/eager/ |
| D | grpc_eager_service_impl.h | 43 GrpcEagerServiceImpl(const WorkerEnv* env,
153 const WorkerEnv* const env_; // Not owned.
|
| D | grpc_eager_service_impl.cc | 29 const WorkerEnv* env, ::grpc::ServerBuilder* server_builder) in GrpcEagerServiceImpl()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.h | 366 std::unique_ptr<ServerInterface> server, WorkerEnv* worker_env,
668 std::unique_ptr<ServerInterface> server, WorkerEnv* worker_env,
682 WorkerEnv* worker_env_ = nullptr;
|
| D | context.cc | 1201 std::unique_ptr<ServerInterface> server, WorkerEnv* worker_env, in InitializeRemoteMaster()
1299 std::unique_ptr<ServerInterface> server, WorkerEnv* worker_env, in SetMasterContextState()
|