| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | session_mgr.cc | 106 WorkerCacheInterface* worker_cache = nullptr; in CreateSession() local
187 void SessionMgr::ResetDefaultWorkerCache(WorkerCacheInterface* worker_cache) { in ResetDefaultWorkerCache()
207 WorkerCacheInterface* worker_cache = nullptr; in UpdateSession() local
297 auto* worker_cache = legacy_session_->worker_cache(); in SetLogging() local
306 auto* worker_cache = session->worker_cache(); in SetLogging() local
318 auto* worker_cache = legacy_session_->worker_cache(); in RetrieveLogs() local
331 auto* worker_cache = session->worker_cache(); in RetrieveLogs() local
348 auto* worker_cache = legacy_session_->worker_cache(); in ClearLogs() local
357 auto* worker_cache = session->worker_cache(); in ClearLogs() local
|
| D | worker_session.cc | 114 std::unique_ptr<WorkerCacheInterface> worker_cache, in WorkerSession()
151 std::unique_ptr<WorkerCacheInterface> worker_cache, in CreateWithBorrowedDeviceMgr()
161 std::unique_ptr<WorkerCacheInterface> worker_cache, in WorkerSession()
|
| D | rpc_collective_executor_mgr.cc | 33 WorkerCacheInterface* worker_cache, const string& task_name) in RpcCollectiveExecutorMgr()
175 WorkerCacheInterface* worker_cache, const string& default_worker_name) { in CreateProdRpcCollectiveExecutorMgr()
|
| D | collective_rma_distributed.h | 32 WorkerCacheInterface* worker_cache, int64_t step_id, string task_name) in CollectiveRemoteAccessDistributed()
|
| D | worker_session.h | 49 WorkerCacheInterface* worker_cache() const { in worker_cache() function
|
| D | master.cc | 138 WorkerCacheInterface* worker_cache, in GetRemoteDevices()
149 WorkerCacheInterface* worker_cache, std::vector<string>* workers) { in GetRemoteWorkers()
157 WorkerCacheInterface* worker_cache) in DeviceFinder()
369 WorkerCacheInterface* worker_cache = nullptr; in CreateSession() local
|
| D | cluster_function_library_runtime.h | 80 const std::shared_ptr<WorkerCacheInterface> worker_cache; member
|
| D | master_env.h | 69 WorkerCacheInterface* worker_cache = nullptr; member
|
| D | remote_device.cc | 79 void NewRemoteDevices(Env* env, WorkerCacheInterface* worker_cache, in NewRemoteDevices()
|
| D | rpc_collective_executor_mgr_test.cc | 45 WorkerCacheInterface* worker_cache = nullptr; in RpcCollectiveExecutorMgrTest() local
|
| D | cluster_function_library_runtime_test.cc | 42 std::unique_ptr<WorkerCacheInterface> worker_cache( in ClusterFunctionLibraryRuntimeTest() local
|
| D | master_session.cc | 75 bool is_partial, WorkerCacheInterface* worker_cache, in ReffedClientGraph()
1103 WorkerCacheInterface* worker_cache = worker_cache_; in DeregisterPartitions() local
1224 std::unique_ptr<WorkerCacheInterface> worker_cache, in MasterSession()
1286 WorkerCacheInterface* worker_cache = get_worker_cache(); in CreateWorkerSessions() local
1401 WorkerCacheInterface* worker_cache = get_worker_cache(); in DeleteWorkerSessions() local
1535 WorkerCacheInterface* worker_cache = get_worker_cache(); in StartStep() local
|
| D | session_mgr_test.cc | 58 [](const ServerDef& server_def, WorkerCacheInterface** worker_cache) { in __anon3309bf4a0102()
|
| D | cluster_function_library_runtime.cc | 189 std::shared_ptr<WorkerCacheInterface> worker_cache = in Instantiate() local
|
| D | collective_param_resolver_distributed.cc | 93 WorkerCacheInterface* worker_cache, const string& task_name) in CollectiveParamResolverDistributed()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_server_lib.cc | 275 WorkerCacheInterface* worker_cache; in Init() local
298 [this](const ServerDef& server_def, WorkerCacheInterface** worker_cache) { in Init()
316 std::vector<string> filtered_worker_list) { in Init()
325 WorkerCacheInterface** worker_cache) { in Init()
360 WorkerCacheInterface** worker_cache) { in WorkerCacheFactory()
450 WorkerCacheInterface* worker_cache; in UpdateServerDef() local
|
| D | grpc_worker_cache_test.cc | 40 std::unique_ptr<WorkerCacheInterface> worker_cache( in TEST() local
76 WorkerCacheInterface* worker_cache = in TEST() local
|
| D | rpc_rendezvous_mgr.cc | 119 void ReleaseWorker(WorkerCacheInterface* worker_cache) { in ReleaseWorker()
249 std::shared_ptr<WorkerCacheInterface> worker_cache = in RecvFromRemoteAsync() local
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context_distributed_manager.cc | 235 WorkerCacheInterface* worker_cache, in AddRemoteDevicesToMgr()
267 WorkerCacheInterface* worker_cache, in GetAllRemoteDevices()
858 auto worker_cache = in EnableCollectiveOps() local
944 const ServerDef& server_def, WorkerCacheInterface* worker_cache) { in EnableCoordinationService()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | c_api_tfrt_distributed_impl.cc | 193 tensorflow::WorkerCacheInterface* worker_cache) { in EnableCoordinationService()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl_test.cc | 149 WorkerCacheInterface** worker_cache) { in EagerServiceImplTest()
|