Searched refs:gpu_count (Results 1 – 11 of 11) sorted by relevance
| /external/pytorch/binaries/ |
| D | inspect_gpu.cc | 37 int gpu_count; in main() local
38 CUDA_ENFORCE(cudaGetDeviceCount(&gpu_count)); in main()
39 for (int i = 0; i < gpu_count; ++i) { in main()
49 for (int i = 0; i < gpu_count; ++i) { in main()
50 for (int j = 0; j < gpu_count; ++j) { in main()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | config_test.py | 813 gpu_count = len(gpus)
821 gpu_options=config_pb2.GPUOptions(visible_device_list=str(gpu_count)))
830 visible_device_list=str(gpu_count - 1)))
834 str(gpu_count - 1))
|
| /external/tensorflow/tensorflow/core/profiler/backends/gpu/ |
| D | rocm_tracer.cc | 1359 int gpu_count; in NumGpus() local
1360 if (hipGetDeviceCount(&gpu_count) != hipSuccess) { in NumGpus()
1363 LOG(INFO) << "Profiler found " << gpu_count << " GPUs"; in NumGpus()
1364 return gpu_count; in NumGpus()
|
| D | cupti_tracer.cc | 1660 int gpu_count; in NumGpus() local
1661 if (cuDeviceGetCount(&gpu_count) != CUDA_SUCCESS) { in NumGpus()
1664 LOG(INFO) << "Profiler found " << gpu_count << " GPUs"; in NumGpus()
1665 return gpu_count; in NumGpus()
|
| /external/angle/src/tests/test_utils/third_party/ |
| D | vulkan_command_buffer_utils.cpp | 267 VkResult init_enumerate_device(struct sample_info &info, uint32_t gpu_count) in init_enumerate_device() argument
269 VkResult res = vkEnumeratePhysicalDevices(info.inst, &gpu_count, NULL); in init_enumerate_device()
270 ASSERT(gpu_count); in init_enumerate_device()
271 info.gpus.resize(gpu_count); in init_enumerate_device()
273 res = vkEnumeratePhysicalDevices(info.inst, &gpu_count, info.gpus.data()); in init_enumerate_device()
|
| D | vulkan_command_buffer_utils.h | 499 VkResult init_enumerate_device(struct sample_info &info, uint32_t gpu_count = 1);
|
| /external/vulkan-validation-layers/tests/ |
| D | vkrenderframework.cpp | 329 err = vkEnumeratePhysicalDevices(inst, &this->gpu_count, NULL); in InitFramework()
330 ASSERT_LE(this->gpu_count, ARRAY_SIZE(objs)) << "Too many gpus"; in InitFramework()
332 err = vkEnumeratePhysicalDevices(inst, &this->gpu_count, objs); in InitFramework()
334 ASSERT_GE(this->gpu_count, (uint32_t)1) << "No GPU available"; in InitFramework()
|
| D | vkrenderframework.h | 129 uint32_t gpu_count; variable
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context.py | 1525 gpu_count = self._config.device_count.get("GPU", None)
1538 if gpu_count is not None:
1539 visible_gpus = visible_gpus[:gpu_count]
|
| /external/google-cloud-java/java-batch/proto-google-cloud-batch-v1alpha/src/main/proto/google/cloud/batch/v1alpha/ |
| D | task.proto | 44 int64 gpu_count = 3; field
|
| /external/googleapis/google/cloud/batch/v1alpha/ |
| D | task.proto | 83 int64 gpu_count = 3; field
|