Searched defs:worker_env (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_cache.cc | 38 GrpcWorkerEnv* worker_env) in GrpcWorkerCache()
177 GrpcWorkerEnv* worker_env) { in NewGrpcWorkerCache()
183 std::shared_ptr<GrpcChannelCache> cc, GrpcWorkerEnv* worker_env, in NewGrpcWorkerCacheWithLocalWorker()
|
| D | grpc_server_lib.h | 107 WorkerEnv* worker_env() override { return &worker_env_; } in worker_env() function
|
| D | rpc_rendezvous_mgr.cc | 314 const WorkerEnv* worker_env) { in Create()
|
| D | grpc_worker_service.cc | 421 GrpcWorker::GrpcWorker(WorkerEnv* worker_env, const ConfigProto& config) in GrpcWorker()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | c_api_tfrt_distributed_impl.cc | 191 const std::string& service_type, const tensorflow::WorkerEnv* worker_env, in EnableCoordinationService()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | session_mgr.cc | 33 WorkerEnv* worker_env, const string& default_worker_name, in SessionMgr()
|
| D | base_rendezvous_mgr.cc | 46 BaseRendezvousMgr::BaseRendezvousMgr(const WorkerEnv* worker_env) in BaseRendezvousMgr()
|
| D | graph_mgr.cc | 59 GraphMgr::GraphMgr(const WorkerEnv* worker_env, const DeviceMgr* device_mgr) in GraphMgr()
|
| D | collective_param_resolver_distributed_test.cc | 151 auto worker_env = absl::make_unique<WorkerEnv>(); in DefineWorker() local
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 1243 void EagerContext::SetWorkerEnv(WorkerEnv* worker_env, in SetWorkerEnv()
1250 std::unique_ptr<ServerInterface> server, WorkerEnv* worker_env, in InitializeRemoteMaster()
1348 std::unique_ptr<ServerInterface> server, WorkerEnv* worker_env, in SetMasterContextState()
|
| D | context_distributed_manager.cc | 943 const std::string& service_type, const WorkerEnv* worker_env, in EnableCoordinationService()
|