Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc1055 GpuDriver::GetDeviceAttribute(CU_DEVICE_ATTRIBUTE_MAX_THREADS_PER_BLOCK, in PopulateDeviceDescription()
1060 thread_dim_limit.x = GpuDriver::GetDeviceAttribute( in PopulateDeviceDescription()
1063 thread_dim_limit.y = GpuDriver::GetDeviceAttribute( in PopulateDeviceDescription()
1066 thread_dim_limit.z = GpuDriver::GetDeviceAttribute( in PopulateDeviceDescription()
1072 GpuDriver::GetDeviceAttribute(CU_DEVICE_ATTRIBUTE_CLOCK_RATE, device_) in PopulateDeviceDescription()
1089 port::StatusOr<int> mem_clock_khz = GpuDriver::GetDeviceAttribute( in PopulateDeviceDescription()
1091 port::StatusOr<int> mem_bus_width_bits = GpuDriver::GetDeviceAttribute( in PopulateDeviceDescription()
1135 GpuDriver::GetDeviceAttribute( in PopulateDeviceDescription()
Dcuda_driver.cc1494 /* static */ port::StatusOr<int> GpuDriver::GetDeviceAttribute( in GetDeviceAttribute() function in stream_executor::gpu::GpuDriver
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_driver.h451 static port::StatusOr<int> GetDeviceAttribute(GpuDeviceAttribute attribute,
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver.cc1255 /* static */ port::StatusOr<int> GpuDriver::GetDeviceAttribute( in GetDeviceAttribute() function in stream_executor::gpu::GpuDriver