Searched refs:host_system_devices (Results 1 – 2 of 2) sorted by relevance
98 std::vector<Device*>* host_system_devices) { in GetHostSystemDevices() argument114 device_set.FindMatchingDevices(host_spec, host_system_devices); in GetHostSystemDevices()116 TF_RET_CHECK(!host_system_devices->empty()) in GetHostSystemDevices()121 TF_RET_CHECK((*host_system_devices)[0]->parsed_name().has_job); in GetHostSystemDevices()122 const string& job_name = (*host_system_devices)[0]->parsed_name().job; in GetHostSystemDevices()123 int replica = (*host_system_devices)[0]->parsed_name().replica; in GetHostSystemDevices()124 for (const auto host_device : *host_system_devices) { in GetHostSystemDevices()139 std::sort(host_system_devices->begin(), host_system_devices->end(), in GetHostSystemDevices()155 std::vector<Device*> host_system_devices; in GetTPUDevices() local157 system_spec, device_set, &host_system_devices)); in GetTPUDevices()[all …]
51 const DeviceSet& device_set, std::vector<Device*>* host_system_devices);