Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dcuda_launch_config.h196 struct Cuda2DLaunchConfig { struct
202 inline Cuda2DLaunchConfig GetCuda2DLaunchConfig(int xdim, int ydim, in GetCuda2DLaunchConfig() argument
204 Cuda2DLaunchConfig config; in GetCuda2DLaunchConfig()
233 using Cuda3DLaunchConfig = Cuda2DLaunchConfig;
283 inline Cuda2DLaunchConfig GetCuda2DLaunchConfig( in GetCuda2DLaunchConfig()
Dcuda_kernel_helper_test.cu.cc56 __global__ void Count2D(Cuda2DLaunchConfig config, int bufsize, int* outbuf) { in Count2D()
191 bool operator==(const Cuda2DLaunchConfig& a, const Cuda2DLaunchConfig& b) { in operator ==()
204 Cuda2DLaunchConfig cfg; in TEST_F()
/external/tensorflow/tensorflow/core/kernels/
Dsvd_op_gpu.cu.cc62 __global__ void ComputeValueOfVKernel(Cuda2DLaunchConfig config, int64 m, in ComputeValueOfVKernel()
198 Cuda2DLaunchConfig cfg2D = GetCuda2DLaunchConfig(batch_size, m, d); in RunSVD()