Searched refs:visible_gpu_order (Results 1 – 2 of 2) sorted by relevance
781 std::vector<PlatformGpuId>* visible_gpu_order) { in ParseVisibleDeviceList() argument782 visible_gpu_order->clear(); in ParseVisibleDeviceList()788 visible_gpu_order->resize(gpu_manager->VisibleDeviceCount()); in ParseVisibleDeviceList()791 std::generate(visible_gpu_order->begin(), visible_gpu_order->end(), in ParseVisibleDeviceList()811 visible_gpu_order->push_back(PlatformGpuId(platform_gpu_id)); in ParseVisibleDeviceList()816 std::set<PlatformGpuId> visible_device_set(visible_gpu_order->begin(), in ParseVisibleDeviceList()817 visible_gpu_order->end()); in ParseVisibleDeviceList()818 if (visible_device_set.size() != visible_gpu_order->size()) { in ParseVisibleDeviceList()828 const std::vector<PlatformGpuId>& visible_gpu_order, in VerifyVirtualDeviceSettings() argument845 visible_gpu_order.size() != virtual_devices.size()) { in VerifyVirtualDeviceSettings()[all …]
335 const std::vector<PlatformGpuId>& visible_gpu_order,366 Status EnablePeerAccess(const std::vector<PlatformGpuId>& visible_gpu_order);373 Status GetValidDeviceIds(const std::vector<PlatformGpuId>& visible_gpu_order,