Home
last modified time | relevance | path

Searched refs:GetCuda3DLaunchConfig (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dcuda_launch_config.h236 inline Cuda3DLaunchConfig GetCuda3DLaunchConfig( in GetCuda3DLaunchConfig() function
286 return GetCuda3DLaunchConfig(xdim, ydim, 1, d, func, in GetCuda2DLaunchConfig()
Dcuda_kernel_helper_test.cu.cc247 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/
Dreduce_slice_ops_gpu.cu.cc76 Cuda3DLaunchConfig config = GetCuda3DLaunchConfig( \