Searched defs:num_gpus_to_use (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_device.cc710 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