Searched refs:remote_devices (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | remote_device.cc | 73 std::vector<Device*> remote_devices; in NewRemoteDevices() local
75 [&worker_cache, &worker_name, &wi, &done, &remote_devices, &s, call] { in NewRemoteDevices()
77 done(s, &remote_devices); in NewRemoteDevices()
90 remote_devices.reserve(call->resp.device_attributes_size()); in NewRemoteDevices()
102 remote_devices.push_back(d); in NewRemoteDevices()
110 remote_devices.push_back(d); in NewRemoteDevices()
|
| 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()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 84 std::vector<std::unique_ptr<tensorflow::Device>> remote_devices; in GetAllRemoteDevices() local
91 [&status, &n, &remote_devices]( in GetAllRemoteDevices()
97 remote_devices.emplace_back(d); in GetAllRemoteDevices()
105 new tensorflow::DeviceMgr(std::move(remote_devices))); in GetAllRemoteDevices()
|