Searched refs:remote_devices (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | remote_device.cc | 65 std::vector<std::unique_ptr<Device>>* remote_devices) { in AsRemoteDevices() argument
70 remote_devices->emplace_back(RenamedDevice::NewRenamedDevice( in AsRemoteDevices()
74 remote_devices->emplace_back(d); in AsRemoteDevices()
95 std::vector<Device*> remote_devices; in NewRemoteDevices() local
97 [&worker_cache, &worker_name, &wi, &done, &remote_devices, &s, call] { in NewRemoteDevices()
99 done(s, &remote_devices); in NewRemoteDevices()
112 remote_devices.reserve(call->resp.device_attributes_size()); in NewRemoteDevices()
124 remote_devices.push_back(d); in NewRemoteDevices()
139 remote_devices.push_back(d); in NewRemoteDevices()
142 remote_devices.push_back(d); in NewRemoteDevices()
|
| D | session_mgr.cc | 145 std::unique_ptr<DynamicDeviceMgr> remote_devices; in CreateSession() local
147 remote_devices = MakeUnique<DynamicDeviceMgr>(); in CreateSession()
149 remote_devices->AddDevices(std::move(cluster_devices))); in CreateSession()
157 std::move(remote_devices))); in CreateSession()
161 std::unique_ptr<DynamicDeviceMgr> remote_devices; in CreateSession() local
163 remote_devices = MakeUnique<DynamicDeviceMgr>(); in CreateSession()
165 remote_devices->AddDevices(std::move(cluster_devices))); in CreateSession()
175 std::move(remote_devices)); in CreateSession()
|
| D | master.cc | 374 std::unique_ptr<std::vector<std::unique_ptr<Device>>> remote_devices( in CreateSession() local
427 worker_cache, remote_devices.get()); in CreateSession()
430 for (auto&& d : *remote_devices) { in CreateSession()
445 worker_cache, remote_devices.get()); in CreateSession()
448 for (auto&& d : *remote_devices) { in CreateSession()
473 options, env_, std::move(remote_devices), std::move(worker_cache_ptr), in CreateSession()
604 std::vector<std::unique_ptr<Device>> remote_devices; in ListDevices() local
606 &remote_devices); in ListDevices()
611 for (auto&& dev : remote_devices) { in ListDevices()
|
| D | remote_device.h | 44 std::vector<std::unique_ptr<Device>>* remote_devices);
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute_node_test.cc | 55 std::vector<std::unique_ptr<Device>> remote_devices; in TEST() local
56 remote_devices.emplace_back( in TEST()
58 TF_ASSERT_OK(remote_device_mgr->AddDevices(std::move(remote_devices))); in TEST()
|
| D | context_distributed_manager.cc | 66 std::vector<std::unique_ptr<tensorflow::Device>> remote_devices; in AddRemoteDevicesToMgr() local
74 [i, &statuses, &counter, &remote_devices, &remote_devices_mu]( in AddRemoteDevicesToMgr()
81 remote_devices.emplace_back(d); in AddRemoteDevicesToMgr()
92 TF_RETURN_IF_ERROR(remote_device_mgr->AddDevices(std::move(remote_devices))); in AddRemoteDevicesToMgr()
110 const std::vector<tensorflow::Device*> remote_devices = in RemoveRemoteDevicesFromMgr() local
113 for (tensorflow::Device* d : remote_devices) { in RemoveRemoteDevicesFromMgr()
|