Searched refs:valid_platform_gpu_ids (Results 1 – 1 of 1) sorted by relevance
817 const std::vector<PlatformGpuId>& valid_platform_gpu_ids) { in VerifyVirtualDeviceSettings() argument839 if (valid_platform_gpu_ids.size() != virtual_devices.size()) { in VerifyVirtualDeviceSettings()843 " #valid GPUs: ", valid_platform_gpu_ids.size(), in VerifyVirtualDeviceSettings()998 std::vector<PlatformGpuId> valid_platform_gpu_ids; in CreateDevices() local1007 GetValidDeviceIds(visible_gpu_order, &valid_platform_gpu_ids)); in CreateDevices()1009 if (num_gpus_to_use > valid_platform_gpu_ids.size()) { in CreateDevices()1010 num_gpus_to_use = valid_platform_gpu_ids.size(); in CreateDevices()1012 if (!valid_platform_gpu_ids.empty()) { in CreateDevices()1031 for (PlatformGpuId platform_gpu_id : valid_platform_gpu_ids) { in CreateDevices()1109 valid_platform_gpu_ids)); in CreateDevices()[all …]