Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/
H A Dgpu_driver.h493 static port::StatusOr<int64_t> GetMaxThreadsPerMultiprocessor(
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_gpu_executor.cc966 GpuDriver::GetMaxThreadsPerMultiprocessor(device).ValueOrDie()); in CreateDeviceDescription()
H A Drocm_driver.cc1164 /* static */ port::StatusOr<int64_t> GpuDriver::GetMaxThreadsPerMultiprocessor( in GetMaxThreadsPerMultiprocessor() function in stream_executor::gpu::GpuDriver
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_gpu_executor.cc1114 GpuDriver::GetMaxThreadsPerMultiprocessor(device).ValueOrDie()); in CreateDeviceDescription()
H A Dcuda_driver.cc1440 /* static */ port::StatusOr<int64_t> GpuDriver::GetMaxThreadsPerMultiprocessor( in GetMaxThreadsPerMultiprocessor() function in stream_executor::gpu::GpuDriver