| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker_session.h | 30 class WorkerCacheInterface; variable
49 WorkerCacheInterface* worker_cache() const { in worker_cache()
60 std::unique_ptr<WorkerCacheInterface> worker_cache,
67 std::unique_ptr<WorkerCacheInterface> worker_cache,
75 std::shared_ptr<WorkerCacheInterface> GetSharedWorkerCache() { in GetSharedWorkerCache()
84 std::unique_ptr<WorkerCacheInterface> new_worker_cache,
92 std::unique_ptr<WorkerCacheInterface> worker_cache,
105 std::shared_ptr<WorkerCacheInterface> worker_cache_
|
| D | worker_session.cc | 30 class WorkerFreeListCache : public WorkerCacheInterface {
32 explicit WorkerFreeListCache(std::unique_ptr<WorkerCacheInterface> w) in WorkerFreeListCache()
97 std::unique_ptr<WorkerCacheInterface> wrapped_;
114 std::unique_ptr<WorkerCacheInterface> worker_cache, in WorkerSession()
134 std::unique_ptr<WorkerCacheInterface> new_worker_cache, in UpdateWorkerCacheAndDevices()
139 worker_cache_ = std::shared_ptr<WorkerCacheInterface>( in UpdateWorkerCacheAndDevices()
151 std::unique_ptr<WorkerCacheInterface> worker_cache, in CreateWithBorrowedDeviceMgr()
161 std::unique_ptr<WorkerCacheInterface> worker_cache, in WorkerSession()
|
| D | session_mgr.h | 30 class WorkerCacheInterface; variable
40 typedef std::function<Status(const ServerDef&, WorkerCacheInterface**)>
45 std::unique_ptr<WorkerCacheInterface> default_worker_cache,
70 void ResetDefaultWorkerCache(WorkerCacheInterface* worker_cache);
110 std::unique_ptr<WorkerCacheInterface> default_worker_cache_;
|
| D | rpc_collective_executor_mgr.h | 26 class WorkerCacheInterface; variable
42 WorkerCacheInterface* worker_cache, const string& task_name);
62 WorkerCacheInterface* const worker_cache_; // Not owned.
91 WorkerCacheInterface* worker_cache, const string& default_worker_name);
|
| D | session_mgr.cc | 34 std::unique_ptr<WorkerCacheInterface> default_worker_cache, in SessionMgr()
40 std::unique_ptr<WorkerCacheInterface>( in SessionMgr()
106 WorkerCacheInterface* worker_cache = nullptr; in CreateSession()
156 std::unique_ptr<WorkerCacheInterface>(worker_cache), in CreateSession()
174 std::unique_ptr<WorkerCacheInterface>(worker_cache), in CreateSession()
187 void SessionMgr::ResetDefaultWorkerCache(WorkerCacheInterface* worker_cache) { in ResetDefaultWorkerCache()
207 WorkerCacheInterface* worker_cache = nullptr; in UpdateSession()
250 std::unique_ptr<WorkerCacheInterface>(worker_cache), in UpdateSession()
|
| D | collective_rma_distributed.h | 24 class WorkerCacheInterface; variable
32 WorkerCacheInterface* worker_cache, int64_t step_id, string task_name) in CollectiveRemoteAccessDistributed()
55 WorkerCacheInterface* worker_cache_; // Not owned
|
| D | master_env.h | 69 WorkerCacheInterface* worker_cache = nullptr;
89 std::unique_ptr<WorkerCacheInterface>,
95 WorkerCacheInterface**)>
|
| D | collective_param_resolver_distributed.h | 25 class WorkerCacheInterface; variable
35 WorkerCacheInterface* worker_cache, const string& task_name);
90 WorkerCacheInterface* worker_cache_; // Not owned
|
| D | worker_cache_wrapper.h | 26 class WorkerCacheWrapper : public WorkerCacheInterface {
28 WorkerCacheWrapper(WorkerCacheInterface* wrapped) : wrapped_(wrapped) {} in WorkerCacheWrapper()
97 WorkerCacheInterface* wrapped_; // Not owned.
|
| D | cancellable_call.h | 31 WorkerCacheInterface* wc) in CancellableCall()
55 WorkerCacheInterface* const wc_; // Not owned
|
| D | remote_device.h | 30 class WorkerCacheInterface; variable
60 void NewRemoteDevices(Env* env, WorkerCacheInterface* worker_cache,
|
| D | worker_cache.h | 34 class WorkerCacheInterface {
36 virtual ~WorkerCacheInterface() {} in ~WorkerCacheInterface()
|
| D | master_session.h | 53 std::unique_ptr<WorkerCacheInterface> worker_cache,
125 const std::unique_ptr<WorkerCacheInterface> worker_cache_;
127 WorkerCacheInterface* get_worker_cache() const;
|
| D | cluster_function_library_runtime.h | 80 const std::shared_ptr<WorkerCacheInterface> worker_cache;
86 std::shared_ptr<WorkerCacheInterface> worker_cache, in FunctionData()
|
| 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()
298 WorkerCacheInterface* worker_cache_;
369 WorkerCacheInterface* worker_cache = nullptr; in CreateSession()
372 std::unique_ptr<WorkerCacheInterface> worker_cache_ptr; in CreateSession()
423 worker_cache_ptr = std::unique_ptr<WorkerCacheInterface>(worker_cache); in CreateSession()
|
| D | worker_cache_partial.h | 32 class WorkerCachePartial : public WorkerCacheInterface {
|
| 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 | device_resolver_distributed.h | 28 class WorkerCacheInterface; variable
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | immediate_execution_distributed_manager.h | 26 class WorkerCacheInterface; variable
53 const ServerDef& server_def, WorkerCacheInterface* worker_cache) = 0;
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_cache.h | 68 WorkerCacheInterface* NewGrpcWorkerCache(std::shared_ptr<GrpcChannelCache> cc,
71 WorkerCacheInterface* NewGrpcWorkerCacheWithLocalWorker(
|
| D | grpc_server_lib.cc | 275 WorkerCacheInterface* worker_cache; in Init()
297 std::unique_ptr<WorkerCacheInterface>(worker_cache), in Init()
298 [this](const ServerDef& server_def, WorkerCacheInterface** worker_cache) { in Init()
314 std::unique_ptr<WorkerCacheInterface> worker_cache, in Init()
325 WorkerCacheInterface** worker_cache) { in Init()
360 WorkerCacheInterface** worker_cache) { in WorkerCacheFactory()
450 WorkerCacheInterface* worker_cache; in UpdateServerDef()
|
| D | grpc_worker_cache.cc | 74 WorkerCacheInterface::ReleaseWorker(target, worker); in ReleaseWorker()
176 WorkerCacheInterface* NewGrpcWorkerCache(std::shared_ptr<GrpcChannelCache> cc, in NewGrpcWorkerCache()
182 WorkerCacheInterface* NewGrpcWorkerCacheWithLocalWorker( in NewGrpcWorkerCacheWithLocalWorker()
|
| D | grpc_worker_cache_test.cc | 40 std::unique_ptr<WorkerCacheInterface> worker_cache( in TEST()
76 WorkerCacheInterface* worker_cache = in TEST()
|
| D | grpc_server_lib.h | 52 const ConfigProto&, const WorkerEnv*, WorkerCacheInterface*)>
136 WorkerCacheInterface** worker_cache);
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context_distributed_manager.h | 49 WorkerCacheInterface* worker_cache) override;
|