Searched refs:valid_platform_gpu_ids (Results 1 – 1 of 1) sorted by relevance
829 const std::vector<PlatformGpuId>& valid_platform_gpu_ids, in VerifyVirtualDeviceSettings() argument852 if (valid_platform_gpu_ids.size() != virtual_devices.size()) { in VerifyVirtualDeviceSettings()856 " #valid GPUs: ", valid_platform_gpu_ids.size(), in VerifyVirtualDeviceSettings()885 const int gpu_id = valid_platform_gpu_ids[i].value(); in VerifyVirtualDeviceSettings()1163 std::vector<PlatformGpuId> valid_platform_gpu_ids; in CreateDevices() local1191 GetValidDeviceIds(visible_gpu_order, &valid_platform_gpu_ids)); in CreateDevices()1193 if (num_gpus_to_use > valid_platform_gpu_ids.size()) { in CreateDevices()1194 num_gpus_to_use = valid_platform_gpu_ids.size(); in CreateDevices()1197 if (!valid_platform_gpu_ids.empty()) { in CreateDevices()1216 for (PlatformGpuId platform_gpu_id : valid_platform_gpu_ids) { in CreateDevices()[all …]