Searched defs:num_gpus_to_use (Results 1 - 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
H A D | gpu_device.cc | 710 const size_t num_gpus_to_use, const GPUOptions& gpu_options, 720 if (num_gpus_to_use < virtual_devices.size()) { 723 " num_gpus_to_use: ", 724 num_gpus_to_use, " #virtual_devices: ", virtual_devices.size()); 859 size_t num_gpus_to_use = INT_MAX; local 862 num_gpus_to_use = iter->second; 871 if (num_gpus_to_use > valid_cuda_gpu_ids.size()) { 872 num_gpus_to_use = valid_cuda_gpu_ids.size(); 937 num_gpus_to_use, gpu_options, visible_gpu_order, valid_cuda_gpu_ids)); 938 // We've verified that num_gpus_to_use > 709 VerifyVirtualDeviceSettings( const size_t num_gpus_to_use, const GPUOptions& gpu_options, const std::vector<CudaGpuId>& visible_gpu_order, const std::vector<CudaGpuId>& valid_cuda_gpu_ids) argument [all...] |
Completed in 52 milliseconds