Searched refs:GetGpu3DLaunchConfig (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | gpu_launch_config.h | 267 Gpu3DLaunchConfig GetGpu3DLaunchConfig(int xdim, int ydim, int zdim, in GetGpu3DLaunchConfig() function
336 CREATE_CUDA_HOST_FUNCTION_ALIAS(GetGpu3DLaunchConfig, GetCuda3DLaunchConfig);
344 return GetGpu3DLaunchConfig(xdim, ydim, 1, d, func, in GetGpu2DLaunchConfig()
|
| D | gpu_kernel_helper_test.cu.cc | 279 TEST_F(GpuLaunchConfigTest, GetGpu3DLaunchConfig) { in TEST_F() argument
290 cfg = GetGpu3DLaunchConfig(dimx, dimy, dimz, d, Count3D, 0, 0); \ in TEST_F()
|