Searched refs:visible_gpu_order (Results 1 – 2 of 2) sorted by relevance
769 std::vector<PlatformGpuId>* visible_gpu_order) { in ParseVisibleDeviceList() argument770 visible_gpu_order->clear(); in ParseVisibleDeviceList()776 visible_gpu_order->resize(gpu_manager->VisibleDeviceCount()); in ParseVisibleDeviceList()779 std::generate(visible_gpu_order->begin(), visible_gpu_order->end(), in ParseVisibleDeviceList()799 visible_gpu_order->push_back(PlatformGpuId(platform_gpu_id)); in ParseVisibleDeviceList()804 std::set<PlatformGpuId> visible_device_set(visible_gpu_order->begin(), in ParseVisibleDeviceList()805 visible_gpu_order->end()); in ParseVisibleDeviceList()806 if (visible_device_set.size() != visible_gpu_order->size()) { in ParseVisibleDeviceList()816 const std::vector<PlatformGpuId>& visible_gpu_order, in VerifyVirtualDeviceSettings() argument832 visible_gpu_order.size() != virtual_devices.size()) { in VerifyVirtualDeviceSettings()[all …]
281 const std::vector<PlatformGpuId>& visible_gpu_order,316 Status GetValidDeviceIds(const std::vector<PlatformGpuId>& visible_gpu_order,