Home
last modified time | relevance | path

Searched refs:valid_platform_gpu_ids (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc817 const std::vector<PlatformGpuId>& valid_platform_gpu_ids) { in VerifyVirtualDeviceSettings() argument
839 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() local
1007 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 …]