Searched refs:gpu_count (Results 1 – 8 of 8) sorted by relevance
| /external/llvm-project/openmp/libomptarget/plugins/amdgpu/impl/ |
| D | system.cpp | 524 int gpu_count = g_atl_machine.processorCount<ATLGPUProcessor>(); in init_hsa() local
525 KernelInfoTable.resize(gpu_count); in init_hsa()
526 SymbolInfoTable.resize(gpu_count); in init_hsa()
541 int gpu_count = g_atl_machine.processorCount<ATLGPUProcessor>(); in init_tasks() local
542 for (int gpu = 0; gpu < gpu_count; gpu++) { in init_tasks()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | config_test.py | 761 gpu_count = len(gpus)
769 gpu_options=config_pb2.GPUOptions(visible_device_list=str(gpu_count)))
777 gpu_options=config_pb2.GPUOptions(visible_device_list=str(gpu_count-1)))
781 str(gpu_count-1))
|
| /external/angle/src/tests/test_utils/third_party/ |
| D | vulkan_command_buffer_utils.cpp | 265 VkResult init_enumerate_device(struct sample_info &info, uint32_t gpu_count) in init_enumerate_device() argument
267 VkResult res = vkEnumeratePhysicalDevices(info.inst, &gpu_count, NULL); in init_enumerate_device()
268 ASSERT(gpu_count); in init_enumerate_device()
269 info.gpus.resize(gpu_count); in init_enumerate_device()
271 res = vkEnumeratePhysicalDevices(info.inst, &gpu_count, info.gpus.data()); in init_enumerate_device()
|
| D | vulkan_command_buffer_utils.h | 498 VkResult init_enumerate_device(struct sample_info &info, uint32_t gpu_count = 1);
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context.py | 1374 gpu_count = self._config.device_count.get("GPU", None)
1387 if gpu_count is not None:
1388 visible_gpus = visible_gpus[:gpu_count]
|
| /external/tensorflow/tensorflow/core/profiler/internal/gpu/ |
| D | cupti_tracer.cc | 1607 int gpu_count; in NumGpus() local
1608 if (cuDeviceGetCount(&gpu_count) != CUDA_SUCCESS) { in NumGpus()
1611 LOG(INFO) << "Profiler found " << gpu_count << " GPUs"; in NumGpus()
1612 return gpu_count; in NumGpus()
|
| /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
|