Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/profiler/internal/gpu/
Dcupti_collector.cc285 absl::optional<int> GetDeviceAttribute(CUdevice device, in GetDeviceAttribute() function in tensorflow::profiler::__anon2da8ba0c0111::PerDeviceCollector
493 GetDeviceAttribute(device, CU_DEVICE_ATTRIBUTE_CLOCK_RATE); in GetDeviceCapabilities()
502 GetDeviceAttribute(device, CU_DEVICE_ATTRIBUTE_MULTIPROCESSOR_COUNT); in GetDeviceCapabilities()
511 GetDeviceAttribute(device, CU_DEVICE_ATTRIBUTE_MEMORY_CLOCK_RATE); in GetDeviceCapabilities()
513 GetDeviceAttribute(device, CU_DEVICE_ATTRIBUTE_GLOBAL_MEMORY_BUS_WIDTH); in GetDeviceCapabilities()
533 auto compute_capability_major = GetDeviceAttribute( in GetDeviceCapabilities()
541 auto compute_capability_minor = GetDeviceAttribute( in GetDeviceCapabilities()
551 GetDeviceAttribute(device, CU_DEVICE_ATTRIBUTE_MAX_THREADS_PER_BLOCK); in GetDeviceCapabilities()
552 auto max_threads_per_sm = GetDeviceAttribute( in GetDeviceCapabilities()
555 GetDeviceAttribute(device, CU_DEVICE_ATTRIBUTE_REGISTERS_PER_BLOCK); in GetDeviceCapabilities()
[all …]
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc1001 GpuDriver::GetDeviceAttribute(CU_DEVICE_ATTRIBUTE_MAX_THREADS_PER_BLOCK, in CreateDeviceDescription()
1006 thread_dim_limit.x = GpuDriver::GetDeviceAttribute( in CreateDeviceDescription()
1009 thread_dim_limit.y = GpuDriver::GetDeviceAttribute( in CreateDeviceDescription()
1012 thread_dim_limit.z = GpuDriver::GetDeviceAttribute( in CreateDeviceDescription()
1018 GpuDriver::GetDeviceAttribute(CU_DEVICE_ATTRIBUTE_CLOCK_RATE, device) in CreateDeviceDescription()
1035 port::StatusOr<int> mem_clock_khz = GpuDriver::GetDeviceAttribute( in CreateDeviceDescription()
1037 port::StatusOr<int> mem_bus_width_bits = GpuDriver::GetDeviceAttribute( in CreateDeviceDescription()
1081 GpuDriver::GetDeviceAttribute( in CreateDeviceDescription()
Dcuda_driver.cc1501 /* static */ port::StatusOr<int> GpuDriver::GetDeviceAttribute( in GetDeviceAttribute() function in stream_executor::gpu::GpuDriver
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_virtual_mem_allocator.cc42 return GpuDriver::GetDeviceAttribute( in SupportsVirtualAddressManagement()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_driver.h523 static port::StatusOr<int> GetDeviceAttribute(GpuDeviceAttribute attribute,
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver.cc1207 /* static */ port::StatusOr<int> GpuDriver::GetDeviceAttribute( in GetDeviceAttribute() function in stream_executor::gpu::GpuDriver