HomeSort by relevance Sort by last modified time
    Searched refs:GetCuda2DLaunchConfig (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/core/util/
cuda_kernel_helper_test.cu.cc 199 TEST_F(CudaLaunchConfigTest, GetCuda2DLaunchConfig) {
209 cfg = GetCuda2DLaunchConfig(dimx, dimy, d); \
219 cfg = GetCuda2DLaunchConfig(dimx, dimy, d, Count2D, 0, 0); \
cuda_launch_config.h 30 // Usage of GetCudaLaunchConfig, GetCuda2DLaunchConfig, and
33 // There are two versions of GetCudaLaunchConfig and GetCuda2DLaunchConfig, one
44 // launch. GetCuda2DLaunchConfig and GetCuda3DLaunchConfig does the same thing
79 Cuda2DLaunchConfig cfg2 = GetCuda2DLaunchConfig(10240, 10240, d);
90 Cuda2DLaunchConfig cfg5 = GetCuda2DLaunchConfig(10240, 10240, d,
202 inline Cuda2DLaunchConfig GetCuda2DLaunchConfig(int xdim, int ydim,
283 inline Cuda2DLaunchConfig GetCuda2DLaunchConfig(
  /external/tensorflow/tensorflow/core/kernels/
concat_lib_gpu_impl.cu.cc 142 auto config = GetCuda2DLaunchConfig(output->dimension(1),
split_lib_gpu.cu.cc 220 auto config = GetCuda2DLaunchConfig(total_cols, total_rows, gpu_device);
svd_op_gpu.cu.cc 159 Cuda2DLaunchConfig cfg2D = GetCuda2DLaunchConfig(batch_size, m, d);

Completed in 767 milliseconds