| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker_session.cc | 111 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr) in WorkerSession() argument
118 remote_device_mgr ? remote_device_mgr.get() : nullptr)), in WorkerSession()
121 remote_device_mgr_(std::move(remote_device_mgr)) { in WorkerSession()
140 remote_device_mgr())); in UpdateWorkerCacheAndDevices()
149 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr) { in CreateWithBorrowedDeviceMgr() argument
152 std::move(graph_mgr), std::move(remote_device_mgr))); in CreateWithBorrowedDeviceMgr()
159 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr) in WorkerSession() argument
165 this, !session_name.empty(), remote_device_mgr.get())), in WorkerSession()
168 remote_device_mgr_(std::move(remote_device_mgr)) { in WorkerSession()
|
| D | worker_session.h | 44 DynamicDeviceMgr* remote_device_mgr() { return remote_device_mgr_.get(); } in remote_device_mgr() function
60 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr);
66 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr);
83 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr);
|
| D | cluster_function_library_runtime.h | 31 DeviceMgr* remote_device_mgr) in ClusterFunctionLibraryRuntime() argument
34 remote_device_mgr_(remote_device_mgr) {} in ClusterFunctionLibraryRuntime()
52 DeviceMgr* remote_device_mgr() const override { return remote_device_mgr_; } in remote_device_mgr() function
|
| D | session_mgr.cc | 175 DeviceMgr* remote_device_mgr = worker_session->remote_device_mgr(); in UpdateSession() local
176 std::vector<Device*> curr_remote_devices = remote_device_mgr->ListDevices(); in UpdateSession()
184 !remote_device_mgr->LookupDevice(da.name(), &device).ok()) { in UpdateSession()
|
| D | session_mgr_test.cc | 101 session->remote_device_mgr()->LookupDevice(local_cpu->name(), &device)); in TEST_F()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | cluster_function_library_runtime.h | 38 DeviceMgr* remote_device_mgr) in EagerClusterFunctionLibraryRuntime() argument
41 remote_device_mgr_(remote_device_mgr) {} in EagerClusterFunctionLibraryRuntime()
64 DeviceMgr* remote_device_mgr() const override { return remote_device_mgr_; } in remote_device_mgr() function
|
| D | cluster_function_library_runtime.cc | 212 context_id, ctx, worker_session->remote_device_mgr()); in CreateClusterFLR()
|
| D | eager_service_impl.cc | 187 std::move(remote_eager_workers), worker_session->remote_device_mgr(), in CreateContext()
294 worker_session->remote_device_mgr(), remote_workers, in UpdateContext()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 118 tensorflow::DynamicDeviceMgr* remote_device_mgr) { in AddRemoteDevicesToMgr() argument
145 TF_RETURN_IF_ERROR(remote_device_mgr->AddDevices(std::move(remote_devices))); in AddRemoteDevicesToMgr()
153 auto remote_device_mgr = absl::make_unique<tensorflow::DynamicDeviceMgr>(); in GetAllRemoteDevices() local
155 remote_device_mgr.get())); in GetAllRemoteDevices()
156 *device_mgr = std::move(remote_device_mgr); in GetAllRemoteDevices()
162 tensorflow::DynamicDeviceMgr* remote_device_mgr) { in RemoveRemoteDevicesFromMgr() argument
164 (remote_device_mgr->ListDevices()); in RemoveRemoteDevicesFromMgr()
175 TF_RETURN_IF_ERROR(remote_device_mgr->RemoveDevices(devices_to_remove)); in RemoveRemoteDevicesFromMgr()
522 tensorflow::DynamicDeviceMgr* remote_device_mgr = nullptr; in UpdateTFE_ContextWithServerDef() local
527 remote_device_mgr = new_remote_device_mgr.get(); in UpdateTFE_ContextWithServerDef()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 149 auto remote_device_manager = remote_device_mgr(); in InitPrioritizedDeviceTypeList()
475 if (remote_device_mgr()) { in ListDevices()
476 remote_device_mgr()->ListDeviceAttributes(devices); in ListDevices()
688 if (remote_device_mgr() != nullptr) { in FindDeviceFromName()
689 return remote_device_mgr()->LookupDevice(device_name, device); in FindDeviceFromName()
1134 DynamicDeviceMgr* remote_device_mgr, in InitializeRemoteWorker() argument
1166 remote_device_manager_.Reset(remote_device_mgr); in InitializeRemoteWorker()
1191 DynamicDeviceMgr* remote_device_mgr, in UpdateRemoteWorker() argument
1210 remote_device_manager_.Reset(remote_device_mgr); in UpdateRemoteWorker()
|
| D | context.h | 241 if (remote_device_mgr() == nullptr) { in CreateRendezvous()
260 const tensorflow::DynamicDeviceMgr* remote_device_mgr() const { in remote_device_mgr() function
344 DynamicDeviceMgr* remote_device_mgr,
358 DynamicDeviceMgr* remote_device_mgr,
|
| D | eager_operation.cc | 150 if (ctx_.remote_device_mgr() == nullptr) return true; in IsLocal()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 863 virtual DeviceMgr* remote_device_mgr() const = 0;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 111 if (parent_ != nullptr && parent_->remote_device_mgr() != nullptr) { in ProcessFunctionLibraryRuntime()
112 all_devices = parent_->remote_device_mgr(); in ProcessFunctionLibraryRuntime()
|
| D | process_function_library_runtime_test.cc | 73 DeviceMgr* remote_device_mgr() const override { return device_mgr_; } in remote_device_mgr() function in tensorflow::__anon7df932810111::TestClusterFLR
|