Searched refs:GetGridLimits (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_driver.h | 436 static bool GetGridLimits(int* x, int* y, int* z, GpuDeviceHandle device);
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 825 if (!GpuDriver::GetGridLimits(&x, &y, &z, device_)) { in FillBlockDimLimit()
|
| D | rocm_driver.cc | 1204 /* static */ bool GpuDriver::GetGridLimits(int* x, int* y, int* z, in GetGridLimits() function in stream_executor::gpu::GpuDriver
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 923 if (!GpuDriver::GetGridLimits(&x, &y, &z, device_)) { in FillBlockDimLimit()
|
| D | cuda_driver.cc | 1444 /* static */ bool GpuDriver::GetGridLimits(int* x, int* y, int* z, in GetGridLimits() function in stream_executor::gpu::GpuDriver
|