| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker_session.h | 40 const DeviceMgr* device_mgr() { in device_mgr()
61 std::unique_ptr<DeviceMgr> device_mgr,
68 const DeviceMgr* borrowed_device_mgr, std::unique_ptr<GraphMgr> graph_mgr,
93 const DeviceMgr* borrowed_device_mgr,
116 const std::unique_ptr<const DeviceMgr> device_mgr_;
117 const DeviceMgr* const borrowed_device_mgr_; // Not owned.
|
| D | worker_env.h | 30 class DeviceMgr; variable
56 const DeviceMgr* device_mgr = nullptr;
|
| D | cluster_function_library_runtime.h | 33 DeviceMgr* remote_device_mgr) in ClusterFunctionLibraryRuntime()
59 DeviceMgr* remote_device_mgr() const override { return remote_device_mgr_; } in remote_device_mgr()
73 DeviceMgr* remote_device_mgr_; // not owned.
|
| D | device_resolver_distributed.h | 27 class DeviceMgr; variable
32 explicit DeviceResolverDistributed(const DeviceMgr* dev_mgr);
|
| D | graph_mgr.h | 45 class DeviceMgr; variable
72 explicit GraphMgr(const WorkerEnv* worker_env, const DeviceMgr* device_mgr);
148 const DeviceMgr* device_mgr_;
|
| D | server_lib.h | 27 class DeviceMgr; variable
70 const tensorflow::DeviceMgr* local_device_mgr;
|
| D | rpc_collective_executor_mgr.h | 24 class DeviceMgr; variable
38 const ConfigProto& config, const DeviceMgr* dev_mgr,
|
| D | collective_param_resolver_distributed.h | 27 class DeviceMgr; variable
32 const DeviceMgr* dev_mgr,
|
| D | worker_session.cc | 110 std::unique_ptr<DeviceMgr> device_mgr, std::unique_ptr<GraphMgr> graph_mgr, in WorkerSession()
147 const DeviceMgr* borrowed_device_mgr, std::unique_ptr<GraphMgr> graph_mgr, in CreateWithBorrowedDeviceMgr()
157 const DeviceMgr* borrowed_device_mgr, std::unique_ptr<GraphMgr> graph_mgr, in WorkerSession()
|
| D | collective_rma_distributed_test.cc | 65 FakeWorker(const string& name, DeviceMgr* dev_mgr, in FakeWorker()
135 DeviceMgr* device_mgr_;
189 for (DeviceMgr* dm : device_mgrs_) { in ~CollRMADistTest()
248 DeviceMgr* dev_mgr = new StaticDeviceMgr(std::move(devices)); in DefineWorker()
285 std::vector<DeviceMgr*> device_mgrs_;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | rendezvous_mgr.h | 45 explicit RefCountedIntraProcessRendezvous(const DeviceMgr* device_mgr);
55 const DeviceMgr* device_mgr_;
73 explicit PrivateIntraProcessRendezvous(const DeviceMgr* device_mgr);
84 const DeviceMgr* device_mgr_;
|
| D | device_mgr.h | 38 class DeviceMgr {
40 DeviceMgr() = default;
41 virtual ~DeviceMgr();
75 TF_DISALLOW_COPY_AND_ASSIGN(DeviceMgr);
79 class StaticDeviceMgr : public DeviceMgr {
118 class DynamicDeviceMgr : public DeviceMgr {
|
| D | device_resolver_local.h | 26 class DeviceMgr; variable
31 explicit DeviceResolverLocal(const DeviceMgr* dev_mgr) : dev_mgr_(dev_mgr) {} in DeviceResolverLocal()
43 const DeviceMgr* dev_mgr_;
|
| D | collective_executor_mgr.h | 24 class DeviceMgr; variable
30 const ConfigProto& config, const DeviceMgr* dev_mgr,
70 const DeviceMgr* dev_mgr_;
|
| D | buf_rendezvous.h | 31 class DeviceMgr; variable
44 explicit BufRendezvous(uint64 step_id, const DeviceMgr* dev_mgr) in BufRendezvous()
124 const DeviceMgr* const dev_mgr_; // Not owned.
|
| D | base_collective_executor.h | 28 class DeviceMgr; variable
100 const DeviceMgr* dev_mgr, const string* gpu_ring_order, in BaseCollectiveExecutor()
140 const DeviceMgr* dev_mgr_; // Not owned.
|
| D | rendezvous_mgr.cc | 36 void SameWorkerRecvDone(const DeviceMgr* device_mgr, in SameWorkerRecvDone()
115 void IntraProcessRecvAsyncImpl(const DeviceMgr* device_mgr, in IntraProcessRecvAsyncImpl()
153 const DeviceMgr* device_mgr) in RefCountedIntraProcessRendezvous()
178 const DeviceMgr* device_mgr) in PrivateIntraProcessRendezvous()
|
| D | collective_param_resolver_local.h | 36 class DeviceMgr; variable
44 const DeviceMgr* dev_mgr,
189 const DeviceMgr* dev_mgr_;
|
| D | collective_rma_local.h | 28 CollectiveRemoteAccessLocal(const DeviceMgr* dev_mgr, in CollectiveRemoteAccessLocal()
75 const DeviceMgr* dev_mgr_; // not owned
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | rendezvous.h | 31 class DeviceMgr; variable
137 Factory(std::function<Status(const int64, const DeviceMgr*, Rendezvous**)> in Factory() argument
147 std::function<Status(const int64, const DeviceMgr*, Rendezvous**)> in Factory() argument
155 Status operator()(const int64 step_id, const DeviceMgr* device_mgr, in operator()
164 std::function<Status(const int64, const DeviceMgr*, Rendezvous**)>
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | cluster_function_library_runtime.h | 39 DeviceMgr* remote_device_mgr) in EagerClusterFunctionLibraryRuntime()
79 DeviceMgr* remote_device_mgr() const override { return remote_device_mgr_; } in remote_device_mgr()
84 DeviceMgr* remote_device_mgr_; // not owned.
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_server_lib.h | 72 const DeviceMgr* local_device_mgr = nullptr;
78 GrpcServer(const ServerDef& server_def, DeviceMgr* local_device_mgr,
91 const DeviceMgr* local_device_mgr,
178 std::unique_ptr<const DeviceMgr> owned_device_manager_;
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.h | 100 bool async, const DeviceMgr* device_mgr, bool device_mgr_owned,
302 const tensorflow::DeviceMgr* local_device_mgr() const { in local_device_mgr()
371 Rendezvous* r, const DeviceMgr* local_device_mgr, int keep_alive_secs,
408 std::unique_ptr<ServerInterface> new_server, const DeviceMgr* device_mgr,
526 void ResetPFLR(const DeviceMgr* device_mgr, Env* env,
580 OwnedOrUnownedHelper<const DeviceMgr> local_device_manager_;
582 std::vector<std::unique_ptr<const DeviceMgr>> old_local_device_managers_;
673 const DeviceMgr* local_device_mgr, int keep_alive_secs,
|
| /external/tensorflow/tensorflow/core/data/ |
| D | standalone.h | 111 Dataset(DatasetBase* dataset, DeviceMgr* device_mgr,
116 std::unique_ptr<DeviceMgr> device_mgr_;
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.h | 37 std::unique_ptr<DeviceMgr> device_mgr,
133 const std::unique_ptr<DeviceMgr> device_mgr_ TF_GUARDED_BY(mu_);
160 OpKernelContext* ctx, std::unique_ptr<DeviceMgr>* device_mgr,
|