/external/tensorflow/tensorflow/core/distributed_runtime/ |
D | session_mgr.cc | 60 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 …]
|
D | worker_session.cc | 97 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()
|
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 | worker_session.h | 41 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,
|
D | remote_device.cc | 57 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()
|
D | BUILD | 97 ":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 …]
|
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 | 28 WorkerCacheInterface* worker_cache, in CollectiveRemoteAccessDistributed() argument 31 worker_cache_(worker_cache) {} in CollectiveRemoteAccessDistributed()
|
D | master_session.cc | 71 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 …]
|
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 | cluster_function_library_runtime.cc | 115 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()
|
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 | 43 void NewRemoteDevices(Env* env, WorkerCacheInterface* worker_cache,
|
D | session_mgr_test.cc | 58 [](const ServerDef& server_def, WorkerCacheInterface** worker_cache) { in __anon395f77470102() argument 59 *worker_cache = nullptr; // Set to null to make debugging easier. in __anon395f77470102()
|
D | device_resolver_distributed_test.cc | 36 WorkerCacheInterface* worker_cache, in TestableDeviceResolverDistributed() argument 38 : DeviceResolverDistributed(dev_mgr, worker_cache, task) {} in TestableDeviceResolverDistributed()
|
D | cluster_function_library_runtime_test.cc | 38 std::unique_ptr<WorkerCacheInterface> worker_cache( in ClusterFunctionLibraryRuntimeTest() local 44 std::move(worker_cache), std::unique_ptr<DeviceMgr>(), in ClusterFunctionLibraryRuntimeTest()
|
D | device_resolver_distributed.h | 32 WorkerCacheInterface* worker_cache,
|
D | master_env.h | 66 WorkerCacheInterface* worker_cache = nullptr; member
|
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
D | grpc_server_lib.cc | 215 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 …]
|
D | rpc_rendezvous_mgr.cc | 118 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/ |
D | gdr_server_lib.cc | 64 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()
|
D | gdr_rendezvous_mgr.cc | 145 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()
|
D | gdr_collective_executor_mgr.h | 39 WorkerCacheInterface* worker_cache, const string& task_name, in GdrCollectiveExecutorMgr() argument 42 std::move(param_resolver), worker_cache, in GdrCollectiveExecutorMgr()
|
D | BUILD | 70 "//tensorflow/core/distributed_runtime:worker_cache", 91 "//tensorflow/core/distributed_runtime:worker_cache", 111 "//tensorflow/core/distributed_runtime:worker_cache",
|
D | gdr_collective_executor_mgr.cc | 69 WorkerCacheInterface* worker_cache, in CollectiveRemoteAccessDistributed() argument 73 worker_cache_(worker_cache), in CollectiveRemoteAccessDistributed()
|