Searched defs:worker_env (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| H A D | grpc_worker_cache.cc | 38 GrpcWorkerEnv* worker_env) in GrpcWorkerCache()
184 GrpcWorkerEnv* worker_env) { in NewGrpcWorkerCache()
190 std::shared_ptr<GrpcChannelCache> cc, GrpcWorkerEnv* worker_env, in NewGrpcWorkerCacheWithLocalWorker()
|
| H A D | grpc_server_lib.h | 107 WorkerEnv* worker_env() override { return &worker_env_; } in worker_env() function
|
| H A D | rpc_rendezvous_mgr.cc | 312 const WorkerEnv* worker_env) { in Create()
|
| H A D | grpc_worker_service.cc | 422 GrpcWorker::GrpcWorker(WorkerEnv* worker_env, const ConfigProto& config) in GrpcWorker()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
| H A D | context.cc | 1451 void EagerContext::SetWorkerEnv(WorkerEnv* worker_env, in SetWorkerEnv()
1458 std::unique_ptr<ServerInterface> server, WorkerEnv* worker_env, in InitializeRemoteMaster()
1556 std::unique_ptr<ServerInterface> server, WorkerEnv* worker_env, in SetMasterContextState()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | session_mgr.cc | 42 WorkerEnv* worker_env, const std::string& default_worker_name, in SessionMgr()
|
| H A D | base_rendezvous_mgr.cc | 48 BaseRendezvousMgr::BaseRendezvousMgr(const WorkerEnv* worker_env) in BaseRendezvousMgr()
|
| H A D | collective_param_resolver_distributed_test.cc | 151 auto worker_env = std::make_unique<WorkerEnv>(); in DefineWorker() local
|
| H A D | graph_mgr.cc | 59 GraphMgr::GraphMgr(const WorkerEnv* worker_env, const DeviceMgr* device_mgr) in GraphMgr()
|