| /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()
148 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr) { in CreateWithBorrowedDeviceMgr() argument
151 std::move(graph_mgr), std::move(remote_device_mgr))); in CreateWithBorrowedDeviceMgr()
158 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr) in WorkerSession() argument
164 this, !session_name.empty(), remote_device_mgr.get())), in WorkerSession()
167 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
63 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr);
69 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr);
95 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr);
|
| D | cluster_function_library_runtime.h | 33 DeviceMgr* remote_device_mgr) in ClusterFunctionLibraryRuntime() argument
36 remote_device_mgr_(remote_device_mgr) {} in ClusterFunctionLibraryRuntime()
59 DeviceMgr* remote_device_mgr() const override { return remote_device_mgr_; } in remote_device_mgr() function
|
| D | session_mgr.cc | 218 DeviceMgr* remote_device_mgr = worker_session->remote_device_mgr(); in UpdateSession() local
219 std::vector<Device*> curr_remote_devices = remote_device_mgr->ListDevices(); in UpdateSession()
227 !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/common_runtime/eager/ |
| D | context_distributed_manager.cc | 65 tensorflow::DynamicDeviceMgr* remote_device_mgr) { in AddRemoteDevicesToMgr() argument
92 TF_RETURN_IF_ERROR(remote_device_mgr->AddDevices(std::move(remote_devices))); in AddRemoteDevicesToMgr()
100 auto remote_device_mgr = std::make_unique<tensorflow::DynamicDeviceMgr>(); in GetAllRemoteDevices() local
102 remote_device_mgr.get())); in GetAllRemoteDevices()
103 *device_mgr = std::move(remote_device_mgr); in GetAllRemoteDevices()
109 tensorflow::DynamicDeviceMgr* remote_device_mgr) { in RemoveRemoteDevicesFromMgr() argument
111 (remote_device_mgr->ListDevices()); in RemoveRemoteDevicesFromMgr()
122 TF_RETURN_IF_ERROR(remote_device_mgr->RemoveDevices(devices_to_remove)); in RemoveRemoteDevicesFromMgr()
487 tensorflow::DynamicDeviceMgr* remote_device_mgr = nullptr; in UpdateContextWithServerDef() local
492 remote_device_mgr = new_remote_device_mgr.get(); in UpdateContextWithServerDef()
[all …]
|
| D | execute_node_test.cc | 54 auto remote_device_mgr = absl::make_unique<DynamicDeviceMgr>(); in TEST() local
58 TF_ASSERT_OK(remote_device_mgr->AddDevices(std::move(remote_devices))); in TEST()
59 Device* device1 = remote_device_mgr->ListDevices().at(0); in TEST()
79 std::move(remote_device_mgr), /*remote_contexts=*/{}, in TEST()
|
| D | tensor_handle_test.cc | 299 DynamicDeviceMgr remote_device_mgr; in TEST() local
302 TF_ASSERT_OK(remote_device_mgr.AddDevices(std::move(vector_d0))); in TEST()
306 EXPECT_TRUE(remote_device_mgr.ContainsDevice( in TEST()
311 TF_ASSERT_OK(remote_device_mgr.AddDevices(std::move(vector_d1))); in TEST()
312 EXPECT_TRUE(remote_device_mgr.ContainsDevice( in TEST()
317 EXPECT_TRUE(remote_device_mgr.ContainsDevice( in TEST()
321 TF_ASSERT_OK(remote_device_mgr.RemoveDevices(std::move(remove_d1))); in TEST()
322 EXPECT_FALSE(remote_device_mgr.ContainsDevice( in TEST()
324 EXPECT_TRUE(remote_device_mgr.ContainsDevice( in TEST()
|
| D | context.cc | 207 auto remote_device_manager = remote_device_mgr(); in InitPrioritizedDeviceTypeList()
596 if (remote_device_mgr()) { in ListDevices()
597 remote_device_mgr()->ListDeviceAttributes(devices); in ListDevices()
888 if (remote_device_mgr() != nullptr) { in FindDeviceFromName()
889 return remote_device_mgr()->LookupDevice(device_name, device); in FindDeviceFromName()
1421 DynamicDeviceMgr* remote_device_mgr, in InitializeRemoteWorker() argument
1453 remote_device_manager_.Reset(remote_device_mgr); in InitializeRemoteWorker()
|
| D | context.h | 305 const tensorflow::DynamicDeviceMgr* remote_device_mgr() const { in remote_device_mgr() function
392 DynamicDeviceMgr* remote_device_mgr,
513 if (remote_device_mgr() == nullptr) { in CreateRendezvous()
|
| D | placement_utils.cc | 53 if (ctx->remote_device_mgr()->ContainsDevice(device_incarnation)) { in ValidateTensorHandleRemoteDevice()
|
| D | eager_operation.cc | 464 if (ctx_.remote_device_mgr() == nullptr) return true; in IsLocal()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | cluster_function_library_runtime.h | 39 DeviceMgr* remote_device_mgr) in EagerClusterFunctionLibraryRuntime() argument
42 remote_device_mgr_(remote_device_mgr) {} in EagerClusterFunctionLibraryRuntime()
79 DeviceMgr* remote_device_mgr() const override { return remote_device_mgr_; } in remote_device_mgr() function
|
| D | cluster_function_library_runtime.cc | 288 context_id, ctx, worker_session->remote_device_mgr()); in CreateClusterFLR()
|
| D | eager_service_impl.cc | 297 std::move(remote_eager_workers), worker_session->remote_device_mgr(), in CreateContext()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 1007 virtual DeviceMgr* remote_device_mgr() const = 0;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 214 if (parent_ != nullptr && parent_->remote_device_mgr() != nullptr) { in InitializeDeviceAndFlr()
215 all_devices = parent_->remote_device_mgr(); in InitializeDeviceAndFlr()
|
| D | process_function_library_runtime_test.cc | 81 DeviceMgr* remote_device_mgr() const override { return device_mgr_; } in remote_device_mgr() function in tensorflow::__anon135d08d60111::TestClusterFLR
|