Searched refs:GetCuda3DLaunchConfig (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_launch_config.h | 236 inline Cuda3DLaunchConfig GetCuda3DLaunchConfig( in GetCuda3DLaunchConfig() function
286 return GetCuda3DLaunchConfig(xdim, ydim, 1, d, func, in GetCuda2DLaunchConfig()
|
| D | cuda_kernel_helper_test.cu.cc | 247 TEST_F(CudaLaunchConfigTest, GetCuda3DLaunchConfig) { in TEST_F() argument
258 cfg = GetCuda3DLaunchConfig(dimx, dimy, dimz, d, Count3D, 0, 0); \ in TEST_F()
|
| /external/tensorflow/tensorflow/contrib/reduce_slice_ops/kernels/ |
| D | reduce_slice_ops_gpu.cu.cc | 76 Cuda3DLaunchConfig config = GetCuda3DLaunchConfig( \
|