| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | session_mgr.cc | 70 WorkerCacheInterface* worker_cache = nullptr; in CreateSession() local
163 WorkerCacheInterface* worker_cache = nullptr; in UpdateSession() local
253 auto* worker_cache = legacy_session_->worker_cache(); in SetLogging() local
262 auto* worker_cache = session->worker_cache(); in SetLogging() local
275 auto* worker_cache = legacy_session_->worker_cache(); in RetrieveLogs() local
288 auto* worker_cache = session->worker_cache(); in RetrieveLogs() local
305 auto* worker_cache = legacy_session_->worker_cache(); in ClearLogs() local
314 auto* worker_cache = session->worker_cache(); in ClearLogs() local
|
| D | worker_session.cc | 109 std::unique_ptr<WorkerCacheInterface> worker_cache, in WorkerSession()
147 std::unique_ptr<WorkerCacheInterface> worker_cache, in CreateWithBorrowedDeviceMgr()
157 std::unique_ptr<WorkerCacheInterface> worker_cache, in WorkerSession()
|
| D | collective_rma_distributed.h | 31 WorkerCacheInterface* worker_cache, int64 step_id) in CollectiveRemoteAccessDistributed()
|
| D | worker_session.h | 49 WorkerCacheInterface* worker_cache() const { return worker_cache_.get(); } in worker_cache() function
|
| D | master.cc | 137 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
|
| D | master_env.h | 66 WorkerCacheInterface* worker_cache = nullptr; member
|
| D | remote_device.cc | 74 void NewRemoteDevices(Env* env, WorkerCacheInterface* worker_cache, in NewRemoteDevices()
|
| D | device_resolver_distributed.cc | 22 const DeviceMgr* dev_mgr, WorkerCacheInterface* worker_cache, in DeviceResolverDistributed()
|
| D | rpc_collective_executor_mgr_test.cc | 42 WorkerCacheInterface* worker_cache = nullptr; in RpcCollectiveExecutorMgrTest() local
|
| D | rpc_collective_executor_mgr.cc | 32 WorkerCacheInterface* worker_cache, const string& task_name) in RpcCollectiveExecutorMgr()
|
| D | cluster_function_library_runtime_test.cc | 39 std::unique_ptr<WorkerCacheInterface> worker_cache( in ClusterFunctionLibraryRuntimeTest() local
|
| D | session_mgr_test.cc | 58 [](const ServerDef& server_def, WorkerCacheInterface** worker_cache) { in __anon18ea8b490102()
|
| D | master_session.cc | 73 bool is_partial, WorkerCacheInterface* worker_cache, in ReffedClientGraph()
1107 WorkerCacheInterface* worker_cache = worker_cache_; in DeregisterPartitions() local
1228 std::unique_ptr<WorkerCacheInterface> worker_cache, in MasterSession()
1290 WorkerCacheInterface* worker_cache = get_worker_cache(); in CreateWorkerSessions() local
1385 WorkerCacheInterface* worker_cache = get_worker_cache(); in DeleteWorkerSessions() local
1519 WorkerCacheInterface* worker_cache = get_worker_cache(); in StartStep() local
|
| D | collective_param_resolver_distributed.cc | 86 DeviceResolverDistributed* dev_resolver, WorkerCacheInterface* worker_cache, in CollectiveParamResolverDistributed()
|
| D | device_resolver_distributed_test.cc | 36 WorkerCacheInterface* worker_cache, in TestableDeviceResolverDistributed()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_server_lib.cc | 248 WorkerCacheInterface* worker_cache; in Init() local
278 [this](const ServerDef& server_def, WorkerCacheInterface** worker_cache) { in Init()
295 std::vector<string> filtered_worker_list) { in Init()
304 WorkerCacheInterface** worker_cache) { in Init()
339 WorkerCacheInterface** worker_cache) { in WorkerCacheFactory()
413 WorkerCacheInterface* worker_cache; in UpdateServerDef() local
|
| D | rpc_rendezvous_mgr.cc | 118 void ReleaseWorker(WorkerCacheInterface* worker_cache) { in ReleaseWorker()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl_test.cc | 137 WorkerCacheInterface** worker_cache) { in EagerServiceImplTest()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 117 tensorflow::WorkerCacheInterface* worker_cache, in AddRemoteDevicesToMgr()
151 tensorflow::WorkerCacheInterface* worker_cache, in GetAllRemoteDevices()
|