Searched refs:num_gpus_to_use (Results 1 – 1 of 1) sorted by relevance
932 const size_t num_gpus_to_use, const GPUOptions& gpu_options, in VerifyVirtualDeviceSettings() argument943 if (num_gpus_to_use < virtual_devices.size()) { in VerifyVirtualDeviceSettings()947 num_gpus_to_use, " #virtual_devices: ", virtual_devices.size()); in VerifyVirtualDeviceSettings()1281 size_t num_gpus_to_use = INT_MAX; in CreateDevices() local1284 num_gpus_to_use = iter->second; in CreateDevices()1293 if (num_gpus_to_use > 0) { in CreateDevices()1319 if (num_gpus_to_use > valid_platform_device_ids.size()) { in CreateDevices()1320 num_gpus_to_use = valid_platform_device_ids.size(); in CreateDevices()1446 num_gpus_to_use, gpu_options, visible_gpu_order, in CreateDevices()1449 num_gpus_to_use = virtual_devices.size(); in CreateDevices()[all …]