Searched refs:remote_devices (Results 1 – 7 of 7) 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 | 159 std::unique_ptr<DynamicDeviceMgr> remote_devices; in CreateSession() local
161 remote_devices = MakeUnique<DynamicDeviceMgr>(); in CreateSession()
163 remote_devices->AddDevices(std::move(cluster_devices))); in CreateSession()
171 std::move(remote_devices))); in CreateSession()
175 std::unique_ptr<DynamicDeviceMgr> remote_devices; in CreateSession() local
177 remote_devices = MakeUnique<DynamicDeviceMgr>(); in CreateSession()
179 remote_devices->AddDevices(std::move(cluster_devices))); in CreateSession()
189 std::move(remote_devices)); in CreateSession()
|
| D | master.cc | 376 std::unique_ptr<std::vector<std::unique_ptr<Device>>> remote_devices( in CreateSession() local
424 worker_cache, remote_devices.get()); in CreateSession()
427 for (auto&& d : *remote_devices) { in CreateSession()
442 worker_cache, remote_devices.get()); in CreateSession()
445 for (auto&& d : *remote_devices) { in CreateSession()
471 options, env_, std::move(remote_devices), std::move(worker_cache_ptr), in CreateSession()
602 std::vector<std::unique_ptr<Device>> remote_devices; in ListDevices() local
604 &remote_devices); in ListDevices()
609 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 | 60 std::vector<std::unique_ptr<Device>> remote_devices; in TEST() local
61 remote_devices.emplace_back( in TEST()
63 TF_ASSERT_OK(remote_device_mgr->AddDevices(std::move(remote_devices))); in TEST()
|
| D | context_distributed_manager.cc | 76 std::vector<std::unique_ptr<Device>> remote_devices; in AddRemoteDevicesToMgr() local
84 [i, &statuses, &counter, &remote_devices, &remote_devices_mu]( in AddRemoteDevicesToMgr()
90 remote_devices.emplace_back(d); in AddRemoteDevicesToMgr()
101 TF_RETURN_IF_ERROR(remote_device_mgr->AddDevices(std::move(remote_devices))); in AddRemoteDevicesToMgr()
118 const std::vector<Device*> remote_devices = in RemoveRemoteDevicesFromMgr() local
121 for (Device* d : remote_devices) { in RemoveRemoteDevicesFromMgr()
742 std::vector<std::unique_ptr<Device>> remote_devices; in EnableCollectiveOps() local
747 remote_devices.emplace_back(NewRemoteDevice(context_->TFEnv(), d)); in EnableCollectiveOps()
749 LOG_AND_RETURN_IF_ERROR(context_->AddDevices(std::move(remote_devices))); in EnableCollectiveOps()
|
| D | context.cc | 786 std::vector<std::unique_ptr<Device>> local_devices, remote_devices; in AddDevices() local
791 remote_devices.push_back(std::move(devices.front())); in AddDevices()
799 if (!remote_devices.empty()) { in AddDevices()
806 ->AddDevices(std::move(remote_devices))); in AddDevices()
|