Searched refs:GetCuda2DLaunchConfig (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_kernel_helper_test.cu.cc | 203 TEST_F(CudaLaunchConfigTest, GetCuda2DLaunchConfig) { in TEST_F() argument
214 cfg = GetCuda2DLaunchConfig(dimx, dimy, d); \ in TEST_F()
226 cfg = GetCuda2DLaunchConfig(dimx, dimy, d, Count2D, 0, 0); \ in TEST_F()
|
| D | cuda_launch_config.h | 202 inline Cuda2DLaunchConfig GetCuda2DLaunchConfig(int xdim, int ydim, in GetCuda2DLaunchConfig() function
283 inline Cuda2DLaunchConfig GetCuda2DLaunchConfig( in GetCuda2DLaunchConfig() function
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | concat_lib_gpu_impl.cu.cc | 143 auto config = GetCuda2DLaunchConfig(output->dimension(1), in ConcatGPUImpl()
|
| D | split_lib_gpu.cu.cc | 225 auto config = GetCuda2DLaunchConfig(total_cols, total_rows, gpu_device); in Run()
|
| D | svd_op_gpu.cu.cc | 198 Cuda2DLaunchConfig cfg2D = GetCuda2DLaunchConfig(batch_size, m, d); in RunSVD()
|