| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker_session.h | 30 class WorkerCacheInterface; variable
49 WorkerCacheInterface* worker_cache() const { return worker_cache_.get(); } in worker_cache()
57 std::unique_ptr<WorkerCacheInterface> worker_cache,
64 std::unique_ptr<WorkerCacheInterface> worker_cache,
72 std::unique_ptr<WorkerCacheInterface> new_worker_cache,
80 std::unique_ptr<WorkerCacheInterface> worker_cache,
92 std::unique_ptr<WorkerCacheInterface> worker_cache_;
|
| D | worker_session.cc | 30 class WorkerFreeListCache : public WorkerCacheInterface {
32 explicit WorkerFreeListCache(std::unique_ptr<WorkerCacheInterface> w) in WorkerFreeListCache()
92 std::unique_ptr<WorkerCacheInterface> wrapped_;
109 std::unique_ptr<WorkerCacheInterface> worker_cache, in WorkerSession()
130 std::unique_ptr<WorkerCacheInterface> new_worker_cache, in UpdateWorkerCacheAndDevices()
133 worker_cache_ = std::unique_ptr<WorkerCacheInterface>( in UpdateWorkerCacheAndDevices()
147 std::unique_ptr<WorkerCacheInterface> worker_cache, in CreateWithBorrowedDeviceMgr()
157 std::unique_ptr<WorkerCacheInterface> worker_cache, in WorkerSession()
|
| D | session_mgr.h | 29 class WorkerCacheInterface; variable
39 typedef std::function<Status(const ServerDef&, WorkerCacheInterface**)>
44 std::unique_ptr<WorkerCacheInterface> default_worker_cache,
94 std::unique_ptr<WorkerCacheInterface> default_worker_cache_;
|
| D | collective_rma_distributed.h | 23 class WorkerCacheInterface; variable
31 WorkerCacheInterface* worker_cache, int64 step_id) in CollectiveRemoteAccessDistributed()
48 WorkerCacheInterface* worker_cache_; // Not owned
|
| D | device_resolver_distributed.h | 27 class WorkerCacheInterface; variable
32 WorkerCacheInterface* worker_cache,
63 WorkerCacheInterface* worker_cache_; // Not owned
|
| D | master_env.h | 66 WorkerCacheInterface* worker_cache = nullptr;
86 std::unique_ptr<WorkerCacheInterface>,
92 WorkerCacheInterface**)>
|
| D | rpc_collective_executor_mgr.h | 26 class WorkerCacheInterface; variable
41 WorkerCacheInterface* worker_cache, const string& task_name);
61 WorkerCacheInterface* const worker_cache_; // Not owned.
|
| D | worker_cache_wrapper.h | 26 class WorkerCacheWrapper : public WorkerCacheInterface {
28 WorkerCacheWrapper(WorkerCacheInterface* wrapped) : wrapped_(wrapped) {} in WorkerCacheWrapper()
92 WorkerCacheInterface* wrapped_; // Not owned.
|
| D | collective_param_resolver_distributed.h | 22 class WorkerCacheInterface; variable
31 WorkerCacheInterface* worker_cache,
85 WorkerCacheInterface* worker_cache_; // Not owned
|
| D | session_mgr.cc | 34 std::unique_ptr<WorkerCacheInterface> default_worker_cache, in SessionMgr()
40 std::unique_ptr<WorkerCacheInterface>( in SessionMgr()
70 WorkerCacheInterface* worker_cache = nullptr; in CreateSession()
120 std::unique_ptr<WorkerCacheInterface>(worker_cache), in CreateSession()
138 std::unique_ptr<WorkerCacheInterface>(worker_cache), in CreateSession()
163 WorkerCacheInterface* worker_cache = nullptr; in UpdateSession()
206 std::unique_ptr<WorkerCacheInterface>(worker_cache), in UpdateSession()
|
| D | remote_device.h | 30 class WorkerCacheInterface; variable
60 void NewRemoteDevices(Env* env, WorkerCacheInterface* worker_cache,
|
| D | worker_cache.h | 33 class WorkerCacheInterface {
35 virtual ~WorkerCacheInterface() {} in ~WorkerCacheInterface()
|
| D | cancellable_call.h | 31 WorkerCacheInterface* wc) in CancellableCall()
59 WorkerCacheInterface* const wc_; // Not owned
|
| 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 | master.cc | 137 WorkerCacheInterface* worker_cache, in GetRemoteDevices()
148 WorkerCacheInterface* worker_cache, std::vector<string>* workers) { in GetRemoteWorkers()
156 WorkerCacheInterface* worker_cache) in DeviceFinder()
297 WorkerCacheInterface* worker_cache_;
368 WorkerCacheInterface* worker_cache = nullptr; in CreateSession()
371 std::unique_ptr<WorkerCacheInterface> worker_cache_ptr; in CreateSession()
422 worker_cache_ptr = std::unique_ptr<WorkerCacheInterface>(worker_cache); in CreateSession()
|
| D | worker_cache_partial.h | 32 class WorkerCachePartial : public WorkerCacheInterface {
|
| D | session_mgr_test.cc | 48 std::unique_ptr<WorkerCacheInterface>(), factory_) { in SessionMgrTest()
58 [](const ServerDef& server_def, WorkerCacheInterface** worker_cache) { in __anon18ea8b490102()
|
| D | remote_device_test.cc | 38 std::unique_ptr<WorkerCacheInterface> worker_cache_;
|
| D | collective_param_resolver_distributed.cc | 31 const string& remote_worker, WorkerCacheInterface* wc) in CompleteGroupCall()
55 const string& remote_worker, WorkerCacheInterface* wc) in CompleteInstanceCall()
86 DeviceResolverDistributed* dev_resolver, WorkerCacheInterface* worker_cache, in CollectiveParamResolverDistributed()
|
| D | remote_device.cc | 74 void NewRemoteDevices(Env* env, WorkerCacheInterface* worker_cache, in NewRemoteDevices()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_cache.h | 66 WorkerCacheInterface* NewGrpcWorkerCache(std::shared_ptr<GrpcChannelCache> cc);
68 WorkerCacheInterface* NewGrpcWorkerCacheWithLocalWorker(
|
| D | grpc_server_lib.cc | 248 WorkerCacheInterface* worker_cache; in Init()
277 std::unique_ptr<WorkerCacheInterface>(worker_cache), in Init()
278 [this](const ServerDef& server_def, WorkerCacheInterface** worker_cache) { in Init()
293 std::unique_ptr<WorkerCacheInterface> worker_cache, in Init()
304 WorkerCacheInterface** worker_cache) { in Init()
339 WorkerCacheInterface** worker_cache) { in WorkerCacheFactory()
413 WorkerCacheInterface* worker_cache; in UpdateServerDef()
|
| D | grpc_worker_cache.cc | 83 WorkerCacheInterface::ReleaseWorker(target, worker); in ReleaseWorker()
157 WorkerCacheInterface* NewGrpcWorkerCache(std::shared_ptr<GrpcChannelCache> cc) { in NewGrpcWorkerCache()
161 WorkerCacheInterface* NewGrpcWorkerCacheWithLocalWorker( in NewGrpcWorkerCacheWithLocalWorker()
|
| D | grpc_server_lib.h | 51 const ConfigProto&, const WorkerEnv*, WorkerCacheInterface*)>
120 WorkerCacheInterface** worker_cache);
|
| D | rpc_rendezvous_mgr_test.cc | 52 class DummyWorkerCache : public WorkerCacheInterface {
77 std::unique_ptr<WorkerCacheInterface>(cache_), in RpcRendezvousMgrTest()
|