Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/
H A Dgpu_driver.h507 static port::StatusOr<int64_t> GetMaxSharedMemoryPerBlock(
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_gpu_executor.cc962 GpuDriver::GetMaxSharedMemoryPerBlock(device).ValueOrDie()); in CreateDeviceDescription()
H A Drocm_driver.cc1158 /* static */ port::StatusOr<int64_t> GpuDriver::GetMaxSharedMemoryPerBlock( in GetMaxSharedMemoryPerBlock() function in stream_executor::gpu::GpuDriver
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_gpu_executor.cc1110 GpuDriver::GetMaxSharedMemoryPerBlock(device).ValueOrDie()); in CreateDeviceDescription()
H A Dcuda_driver.cc1434 /* static */ port::StatusOr<int64_t> GpuDriver::GetMaxSharedMemoryPerBlock( in GetMaxSharedMemoryPerBlock() function in stream_executor::gpu::GpuDriver