Home
last modified time | relevance | path

Searched refs:worker_cache (Results 1 – 25 of 36) sorted by relevance

12

/external/tensorflow/tensorflow/core/distributed_runtime/
Dsession_mgr.cc60 WorkerCacheInterface* worker_cache = nullptr; in CreateSession() local
63 worker_cache = new WorkerCacheWrapper(default_worker_cache_.get()); in CreateSession()
66 TF_RETURN_IF_ERROR(worker_cache_factory_(server_def, &worker_cache)); in CreateSession()
70 if (worker_cache != nullptr && default_worker_cache_ != nullptr) { in CreateSession()
71 worker_cache->SetLogging(this->is_logging_active_); in CreateSession()
91 std::unique_ptr<WorkerCacheInterface>(worker_cache), in CreateSession()
100 std::unique_ptr<WorkerCacheInterface>(worker_cache), in CreateSession()
150 auto* worker_cache = legacy_session_->worker_cache.get(); in SetLogging() local
151 if (worker_cache) { in SetLogging()
152 worker_cache->SetLogging(active); in SetLogging()
[all …]
Dworker_session.cc97 std::unique_ptr<WorkerCacheInterface> worker_cache, in WorkerSession() argument
102 worker_cache(new WorkerFreeListCache(std::move(worker_cache))), in WorkerSession()
112 std::unique_ptr<WorkerCacheInterface> worker_cache, in CreateWithBorrowedDeviceMgr() argument
115 new WorkerSession(session_name, worker_name, std::move(worker_cache), in CreateWithBorrowedDeviceMgr()
121 std::unique_ptr<WorkerCacheInterface> worker_cache, in WorkerSession() argument
126 worker_cache(new WorkerFreeListCache(std::move(worker_cache))), in WorkerSession()
Dmaster.cc137 WorkerCacheInterface* worker_cache, in GetRemoteDevices() argument
139 DeviceFinder finder(device_filters, env, worker_cache); in GetRemoteDevices()
148 WorkerCacheInterface* worker_cache, std::vector<string>* workers) { in GetRemoteWorkers() argument
149 DeviceFinder finder(device_filters, env, worker_cache); in GetRemoteWorkers()
156 WorkerCacheInterface* worker_cache) in DeviceFinder() argument
157 : env_(env), worker_cache_(worker_cache) { in DeviceFinder()
158 CHECK(worker_cache) << "Worker cache was null!"; in DeviceFinder()
176 worker_cache->ListWorkers(&workers); in DeviceFinder()
203 worker_cache->ListWorkersInJob(job_name, &workers_in_job); in DeviceFinder()
219 worker_cache->ListWorkers(&workers); in DeviceFinder()
[all …]
Dworker_session.h41 const std::unique_ptr<WorkerCacheInterface> worker_cache; member
61 std::unique_ptr<WorkerCacheInterface> worker_cache,
67 std::unique_ptr<WorkerCacheInterface> worker_cache,
74 std::unique_ptr<WorkerCacheInterface> worker_cache,
Dremote_device.cc57 void NewRemoteDevices(Env* env, WorkerCacheInterface* worker_cache, in NewRemoteDevices() argument
59 WorkerInterface* wi = worker_cache->CreateWorker(worker_name); in NewRemoteDevices()
70 auto cb = [env, worker_cache, worker_name, done, wi, in NewRemoteDevices()
75 [&worker_cache, &worker_name, &wi, &done, &remote_devices, &s, call] { in NewRemoteDevices()
76 worker_cache->ReleaseWorker(worker_name, wi); in NewRemoteDevices()
DBUILD97 ":worker_cache",
236 ":worker_cache",
262 name = "worker_cache",
263 hdrs = ["worker_cache.h"],
275 ":worker_cache",
286 ":worker_cache",
316 ":worker_cache",
338 ":worker_cache",
393 ":worker_cache",
409 ":worker_cache",
[all …]
Drpc_collective_executor_mgr_test.cc42 WorkerCacheInterface* worker_cache = nullptr; in RpcCollectiveExecutorMgrTest() local
49 device_mgr_.get(), worker_cache, task_name)); in RpcCollectiveExecutorMgrTest()
53 worker_cache, task_name)); in RpcCollectiveExecutorMgrTest()
57 worker_cache, task_name)); in RpcCollectiveExecutorMgrTest()
Dcollective_rma_distributed.h28 WorkerCacheInterface* worker_cache, in CollectiveRemoteAccessDistributed() argument
31 worker_cache_(worker_cache) {} in CollectiveRemoteAccessDistributed()
Dmaster_session.cc71 bool is_partial, WorkerCacheInterface* worker_cache, in ReffedClientGraph() argument
79 worker_cache_(worker_cache), in ReffedClientGraph()
1052 WorkerCacheInterface* worker_cache = worker_cache_; in DeregisterPartitions() local
1056 auto cb = [worker_cache, c, name, w](const Status& s) { in DeregisterPartitions()
1063 worker_cache->ReleaseWorker(name, w); in DeregisterPartitions()
1173 std::unique_ptr<WorkerCacheInterface> worker_cache, in MasterSession() argument
1181 worker_cache_(std::move(worker_cache)), in MasterSession()
1236 WorkerCacheInterface* worker_cache = get_worker_cache(); in CreateWorkerSessions() local
1254 auto cleanup = gtl::MakeCleanup([&workers, worker_cache] { in CreateWorkerSessions()
1257 worker_cache->ReleaseWorker(*worker_group.name, worker_group.worker); in CreateWorkerSessions()
[all …]
Ddevice_resolver_distributed.cc22 const DeviceMgr* dev_mgr, WorkerCacheInterface* worker_cache, in DeviceResolverDistributed() argument
24 : dev_mgr_(dev_mgr), worker_cache_(worker_cache), task_name_(task_name) {} in DeviceResolverDistributed()
Dcluster_function_library_runtime.cc115 worker_session_->worker_cache->ReleaseWorker(function_data.target, in ~ClusterFunctionLibraryRuntime()
127 worker_session_->worker_cache->CreateWorker(options.target); in Instantiate()
131 worker_session_->worker_cache->ListWorkers(&workers); in Instantiate()
Drpc_collective_executor_mgr.cc32 WorkerCacheInterface* worker_cache, const string& task_name) in RpcCollectiveExecutorMgr() argument
35 worker_cache_(worker_cache), in RpcCollectiveExecutorMgr()
Dremote_device.h43 void NewRemoteDevices(Env* env, WorkerCacheInterface* worker_cache,
Dsession_mgr_test.cc58 [](const ServerDef& server_def, WorkerCacheInterface** worker_cache) { in __anon395f77470102() argument
59 *worker_cache = nullptr; // Set to null to make debugging easier. in __anon395f77470102()
Ddevice_resolver_distributed_test.cc36 WorkerCacheInterface* worker_cache, in TestableDeviceResolverDistributed() argument
38 : DeviceResolverDistributed(dev_mgr, worker_cache, task) {} in TestableDeviceResolverDistributed()
Dcluster_function_library_runtime_test.cc38 std::unique_ptr<WorkerCacheInterface> worker_cache( in ClusterFunctionLibraryRuntimeTest() local
44 std::move(worker_cache), std::unique_ptr<DeviceMgr>(), in ClusterFunctionLibraryRuntimeTest()
Ddevice_resolver_distributed.h32 WorkerCacheInterface* worker_cache,
Dmaster_env.h66 WorkerCacheInterface* worker_cache = nullptr; member
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_server_lib.cc215 WorkerCacheInterface* worker_cache; in Init() local
218 WorkerCacheFactory(worker_cache_factory_options, &worker_cache)); in Init()
219 CHECK_NE(nullptr, worker_cache); in Init()
223 opts.collective_mgr_func(config, &worker_env_, worker_cache); in Init()
230 new DeviceResolverDistributed(worker_env_.device_mgr, worker_cache, in Init()
234 dev_resolver.get(), worker_cache, in Init()
238 std::move(param_resolver), worker_cache, default_worker_name); in Init()
244 std::unique_ptr<WorkerCacheInterface>(worker_cache), in Init()
245 [this](const ServerDef& server_def, WorkerCacheInterface** worker_cache) { in Init() argument
247 return WorkerCacheFactory(options, worker_cache); in Init()
[all …]
Drpc_rendezvous_mgr.cc118 void ReleaseWorker(WorkerCacheInterface* worker_cache) { in ReleaseWorker() argument
121 worker_cache->ReleaseWorker(src_worker_, wi_); in ReleaseWorker()
235 WorkerInterface* rwi = sess->worker_cache->CreateWorker(call->src_worker_); in RecvFromRemoteAsync()
246 sess->worker_cache->ReleaseWorker(call->src_worker_, rwi); in RecvFromRemoteAsync()
264 call->ReleaseWorker(sess->worker_cache.get()); in RecvFromRemoteAsync()
281 call->ReleaseWorker(session()->worker_cache.get()); in RecvFromRemoteAsync()
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_server_lib.cc64 WorkerCacheInterface* worker_cache) { in Init() argument
72 new DeviceResolverDistributed(env->device_mgr, worker_cache, in Init()
76 config, env->device_mgr, dev_resolver.get(), worker_cache, in Init()
80 std::move(param_resolver), worker_cache, default_worker_name, in Init()
Dgdr_rendezvous_mgr.cc145 WorkerInterface* rwi = sess->worker_cache->CreateWorker(src_worker); in RecvFromRemoteAsync()
155 sess->worker_cache->ReleaseWorker(src_worker, rwi); in RecvFromRemoteAsync()
173 session()->worker_cache->ReleaseWorker(src_worker, rwi); in RecvFromRemoteAsync()
190 session()->worker_cache->ReleaseWorker(src_worker, rwi); in RecvFromRemoteAsync()
Dgdr_collective_executor_mgr.h39 WorkerCacheInterface* worker_cache, const string& task_name, in GdrCollectiveExecutorMgr() argument
42 std::move(param_resolver), worker_cache, in GdrCollectiveExecutorMgr()
DBUILD70 "//tensorflow/core/distributed_runtime:worker_cache",
91 "//tensorflow/core/distributed_runtime:worker_cache",
111 "//tensorflow/core/distributed_runtime:worker_cache",
Dgdr_collective_executor_mgr.cc69 WorkerCacheInterface* worker_cache, in CollectiveRemoteAccessDistributed() argument
73 worker_cache_(worker_cache), in CollectiveRemoteAccessDistributed()

12