Home
last modified time | relevance | path

Searched defs:worker_cache (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/
Dsession_mgr.cc60 WorkerCacheInterface* worker_cache = nullptr; in CreateSession() local
150 auto* worker_cache = legacy_session_->worker_cache.get(); in SetLogging() local
159 auto* worker_cache = session->worker_cache.get(); in SetLogging() local
172 auto* worker_cache = legacy_session_->worker_cache.get(); in RetrieveLogs() local
185 auto* worker_cache = session->worker_cache.get(); in RetrieveLogs() local
202 auto* worker_cache = legacy_session_->worker_cache.get(); in ClearLogs() local
211 auto* worker_cache = session->worker_cache.get(); in ClearLogs() local
Dworker_session.cc97 std::unique_ptr<WorkerCacheInterface> worker_cache, in WorkerSession()
112 std::unique_ptr<WorkerCacheInterface> worker_cache, in CreateWithBorrowedDeviceMgr()
121 std::unique_ptr<WorkerCacheInterface> worker_cache, in WorkerSession()
Dcollective_rma_distributed.h28 WorkerCacheInterface* worker_cache, in CollectiveRemoteAccessDistributed()
Dworker_session.h41 const std::unique_ptr<WorkerCacheInterface> worker_cache; member
Dmaster.cc137 WorkerCacheInterface* worker_cache, in GetRemoteDevices()
148 WorkerCacheInterface* worker_cache, std::vector<string>* workers) { in GetRemoteWorkers()
156 WorkerCacheInterface* worker_cache) in DeviceFinder()
368 WorkerCacheInterface* worker_cache = nullptr; in CreateSession() local
Dmaster_env.h66 WorkerCacheInterface* worker_cache = nullptr; member
Dremote_device.cc57 void NewRemoteDevices(Env* env, WorkerCacheInterface* worker_cache, in NewRemoteDevices()
Ddevice_resolver_distributed.cc22 const DeviceMgr* dev_mgr, WorkerCacheInterface* worker_cache, in DeviceResolverDistributed()
Drpc_collective_executor_mgr.cc32 WorkerCacheInterface* worker_cache, const string& task_name) in RpcCollectiveExecutorMgr()
Drpc_collective_executor_mgr_test.cc42 WorkerCacheInterface* worker_cache = nullptr; in RpcCollectiveExecutorMgrTest() local
Dcluster_function_library_runtime_test.cc38 std::unique_ptr<WorkerCacheInterface> worker_cache( in ClusterFunctionLibraryRuntimeTest() local
Dsession_mgr_test.cc58 [](const ServerDef& server_def, WorkerCacheInterface** worker_cache) { in __anon395f77470102()
Ddevice_resolver_distributed_test.cc36 WorkerCacheInterface* worker_cache, in TestableDeviceResolverDistributed()
Dmaster_session.cc71 bool is_partial, WorkerCacheInterface* worker_cache, in ReffedClientGraph()
1052 WorkerCacheInterface* worker_cache = worker_cache_; in DeregisterPartitions() local
1173 std::unique_ptr<WorkerCacheInterface> worker_cache, in MasterSession()
1236 WorkerCacheInterface* worker_cache = get_worker_cache(); in CreateWorkerSessions() local
1315 WorkerCacheInterface* worker_cache = get_worker_cache(); in DeleteWorkerSessions() local
1449 WorkerCacheInterface* worker_cache = get_worker_cache(); in StartStep() local
Dcollective_param_resolver_distributed.cc83 DeviceResolverDistributed* dev_resolver, WorkerCacheInterface* worker_cache, in CollectiveParamResolverDistributed()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_server_lib.cc215 WorkerCacheInterface* worker_cache; in Init() local
245 [this](const ServerDef& server_def, WorkerCacheInterface** worker_cache) { in Init()
262 std::vector<string> filtered_worker_list) { in Init()
271 WorkerCacheInterface** worker_cache) { in Init()
306 WorkerCacheInterface** worker_cache) { in WorkerCacheFactory()
Drpc_rendezvous_mgr.cc118 void ReleaseWorker(WorkerCacheInterface* worker_cache) { in ReleaseWorker()
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_collective_executor_mgr.h39 WorkerCacheInterface* worker_cache, const string& task_name, in GdrCollectiveExecutorMgr()
Dgdr_server_lib.cc64 WorkerCacheInterface* worker_cache) { in Init()
Dgdr_collective_executor_mgr.cc69 WorkerCacheInterface* worker_cache, in CollectiveRemoteAccessDistributed()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl_test.cc62 WorkerCacheInterface** worker_cache) { in EagerServiceImplTest()
/external/tensorflow/tensorflow/c/eager/
Dc_api.cc82 tensorflow::WorkerCacheInterface* worker_cache, in GetAllRemoteDevices()