| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | session_mgr.cc | 70 WorkerCacheInterface* worker_cache = nullptr; in CreateSession() local
73 worker_cache = new WorkerCacheWrapper(default_worker_cache_.get()); in CreateSession()
76 TF_RETURN_IF_ERROR(worker_cache_factory_(server_def, &worker_cache)); in CreateSession()
80 if (worker_cache != nullptr && default_worker_cache_ != nullptr) { in CreateSession()
81 worker_cache->SetLogging(this->is_logging_active_); in CreateSession()
120 std::unique_ptr<WorkerCacheInterface>(worker_cache), in CreateSession()
138 std::unique_ptr<WorkerCacheInterface>(worker_cache), in CreateSession()
163 WorkerCacheInterface* worker_cache = nullptr; in UpdateSession() local
165 worker_cache = new WorkerCacheWrapper(default_worker_cache_.get()); in UpdateSession()
167 TF_RETURN_IF_ERROR(worker_cache_factory_(server_def, &worker_cache)); in UpdateSession()
[all …]
|
| D | master.cc | 137 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 …]
|
| D | remote_device.cc | 74 void NewRemoteDevices(Env* env, WorkerCacheInterface* worker_cache, in NewRemoteDevices() argument
76 WorkerInterface* wi = worker_cache->GetOrCreateWorker(worker_name); in NewRemoteDevices()
87 auto cb = [env, worker_cache, worker_name, done, wi, in NewRemoteDevices()
92 [&worker_cache, &worker_name, &wi, &done, &remote_devices, &s, call] { in NewRemoteDevices()
93 worker_cache->ReleaseWorker(worker_name, wi); in NewRemoteDevices()
|
| D | worker_session.h | 49 WorkerCacheInterface* worker_cache() const { return worker_cache_.get(); } in worker_cache() function
57 std::unique_ptr<WorkerCacheInterface> worker_cache,
64 std::unique_ptr<WorkerCacheInterface> worker_cache,
80 std::unique_ptr<WorkerCacheInterface> worker_cache,
|
| D | worker_session.cc | 109 std::unique_ptr<WorkerCacheInterface> worker_cache, in WorkerSession() argument
114 worker_cache_(new WorkerFreeListCache(std::move(worker_cache))), in WorkerSession()
147 std::unique_ptr<WorkerCacheInterface> worker_cache, in CreateWithBorrowedDeviceMgr() argument
151 session_name, worker_name, std::move(worker_cache), borrowed_device_mgr, in CreateWithBorrowedDeviceMgr()
157 std::unique_ptr<WorkerCacheInterface> worker_cache, in WorkerSession() argument
162 worker_cache_(new WorkerFreeListCache(std::move(worker_cache))), in WorkerSession()
|
| D | BUILD | 99 ":worker_cache",
246 ":worker_cache",
272 name = "worker_cache",
273 hdrs = ["worker_cache.h"],
285 deps = [":worker_cache"],
293 ":worker_cache",
324 ":worker_cache",
347 ":worker_cache",
402 ":worker_cache",
418 ":worker_cache",
[all …]
|
| D | rpc_collective_executor_mgr_test.cc | 42 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()
|
| D | collective_rma_distributed.h | 31 WorkerCacheInterface* worker_cache, int64 step_id) in CollectiveRemoteAccessDistributed() argument
33 worker_cache_(worker_cache) {} in CollectiveRemoteAccessDistributed()
|
| D | device_resolver_distributed.cc | 22 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()
|
| D | master_session.cc | 73 bool is_partial, WorkerCacheInterface* worker_cache, in ReffedClientGraph() argument
81 worker_cache_(worker_cache), in ReffedClientGraph()
1107 WorkerCacheInterface* worker_cache = worker_cache_; in DeregisterPartitions() local
1111 auto cb = [worker_cache, c, name, w](const Status& s) { in DeregisterPartitions()
1118 worker_cache->ReleaseWorker(name, w); in DeregisterPartitions()
1228 std::unique_ptr<WorkerCacheInterface> worker_cache, in MasterSession() argument
1236 worker_cache_(std::move(worker_cache)), in MasterSession()
1290 WorkerCacheInterface* worker_cache = get_worker_cache(); in CreateWorkerSessions() local
1308 auto cleanup = gtl::MakeCleanup([&workers, worker_cache] { in CreateWorkerSessions()
1311 worker_cache->ReleaseWorker(*worker_group.name, worker_group.worker); in CreateWorkerSessions()
[all …]
|
| D | rpc_collective_executor_mgr.cc | 32 WorkerCacheInterface* worker_cache, const string& task_name) in RpcCollectiveExecutorMgr() argument
35 worker_cache_(worker_cache), in RpcCollectiveExecutorMgr()
|
| D | remote_device.h | 60 void NewRemoteDevices(Env* env, WorkerCacheInterface* worker_cache,
|
| D | cluster_function_library_runtime.cc | 176 worker_session_->worker_cache()->ReleaseWorker(function_data.target, in ~ClusterFunctionLibraryRuntime()
190 worker_session_->worker_cache()->GetOrCreateWorker(target); in Instantiate()
194 worker_session_->worker_cache()->ListWorkers(&workers); in Instantiate()
|
| D | session_mgr_test.cc | 58 [](const ServerDef& server_def, WorkerCacheInterface** worker_cache) { in __anon18ea8b490102() argument
59 *worker_cache = nullptr; // Set to null to make debugging easier. in __anon18ea8b490102()
|
| D | device_resolver_distributed.h | 32 WorkerCacheInterface* worker_cache,
|
| D | master_env.h | 66 WorkerCacheInterface* worker_cache = nullptr; member
|
| D | rpc_collective_executor_mgr.h | 41 WorkerCacheInterface* worker_cache, const string& task_name);
|
| D | collective_param_resolver_distributed.h | 31 WorkerCacheInterface* worker_cache,
|
| D | cluster_function_library_runtime_test.cc | 39 std::unique_ptr<WorkerCacheInterface> worker_cache( in ClusterFunctionLibraryRuntimeTest() local
45 std::move(worker_cache), std::unique_ptr<DeviceMgr>(), in ClusterFunctionLibraryRuntimeTest()
|
| D | device_resolver_distributed_test.cc | 36 WorkerCacheInterface* worker_cache, in TestableDeviceResolverDistributed() argument
38 : DeviceResolverDistributed(dev_mgr, worker_cache, task) {} in TestableDeviceResolverDistributed()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_server_lib.cc | 248 WorkerCacheInterface* worker_cache; in Init() local
251 WorkerCacheFactory(worker_cache_factory_options, &worker_cache)); in Init()
252 CHECK_NE(nullptr, worker_cache); in Init()
256 opts.collective_mgr_func(config, &worker_env_, worker_cache); in Init()
263 new DeviceResolverDistributed(worker_env_.device_mgr, worker_cache, in Init()
267 dev_resolver.get(), worker_cache, in Init()
271 std::move(param_resolver), worker_cache, default_worker_name); in Init()
277 std::unique_ptr<WorkerCacheInterface>(worker_cache), in Init()
278 [this](const ServerDef& server_def, WorkerCacheInterface** worker_cache) { in Init() argument
280 return WorkerCacheFactory(options, worker_cache); in Init()
[all …]
|
| D | rpc_rendezvous_mgr.cc | 118 void ReleaseWorker(WorkerCacheInterface* worker_cache) { in ReleaseWorker() argument
121 worker_cache->ReleaseWorker(src_worker_, wi_); in ReleaseWorker()
231 sess->worker_cache()->GetOrCreateWorker(call->src_worker_); in RecvFromRemoteAsync()
242 sess->worker_cache()->ReleaseWorker(call->src_worker_, rwi); in RecvFromRemoteAsync()
260 call->ReleaseWorker(sess->worker_cache()); in RecvFromRemoteAsync()
277 call->ReleaseWorker(session()->worker_cache()); in RecvFromRemoteAsync()
|
| D | BUILD | 161 "//tensorflow/core/distributed_runtime:worker_cache",
201 "//tensorflow/core/distributed_runtime:worker_cache",
278 "//tensorflow/core/distributed_runtime:worker_cache",
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 173 worker_session->worker_cache()->ListWorkers(&remote_workers); in CreateContext()
179 TF_RETURN_IF_ERROR(worker_session->worker_cache()->GetEagerClientCache( in CreateContext()
272 worker_session->worker_cache()->ListWorkers(&remote_workers); in UpdateContext()
285 TF_RETURN_IF_ERROR(worker_session->worker_cache()->GetEagerClientCache( in UpdateContext()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 117 tensorflow::WorkerCacheInterface* worker_cache, in AddRemoteDevicesToMgr() argument
126 tensorflow::Env::Default(), worker_cache, added_remote_workers[i], in AddRemoteDevicesToMgr()
151 tensorflow::WorkerCacheInterface* worker_cache, in GetAllRemoteDevices() argument
154 TF_RETURN_IF_ERROR(AddRemoteDevicesToMgr(remote_workers, worker_cache, in GetAllRemoteDevices()
188 grpc_server->master_env()->worker_cache->ListWorkers(remote_workers); in ListRemoteWorkers()
505 grpc_server->master_env()->worker_cache->GetEagerClientCache( in UpdateTFE_ContextWithServerDef()
525 remote_workers, grpc_server->master_env()->worker_cache, in UpdateTFE_ContextWithServerDef()
570 added_workers, grpc_server->master_env()->worker_cache, in UpdateTFE_ContextWithServerDef()
836 status->status = grpc_server->master_env()->worker_cache->GetEagerClientCache( in TFE_ContextCheckAlive()
|