Searched refs:num_gpus_to_use (Results 1 – 1 of 1) sorted by relevance
815 const size_t num_gpus_to_use, const GPUOptions& gpu_options, in VerifyVirtualDeviceSettings() argument825 if (num_gpus_to_use < virtual_devices.size()) { in VerifyVirtualDeviceSettings()829 num_gpus_to_use, " #virtual_devices: ", virtual_devices.size()); in VerifyVirtualDeviceSettings()991 size_t num_gpus_to_use = INT_MAX; in CreateDevices() local994 num_gpus_to_use = iter->second; in CreateDevices()1003 if (num_gpus_to_use > 0) { 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()1107 TF_RETURN_IF_ERROR(VerifyVirtualDeviceSettings(num_gpus_to_use, gpu_options, in CreateDevices()1111 num_gpus_to_use = virtual_devices.size(); in CreateDevices()[all …]