Searched refs:Gpu2DLaunchConfig (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | gpu_launch_config.h | 219 struct Gpu2DLaunchConfig { struct
224 CREATE_CUDA_TYPE_ALIAS(Gpu2DLaunchConfig, Cuda2DLaunchConfig); argument
226 inline Gpu2DLaunchConfig GetGpu2DLaunchConfig(int xdim, int ydim, in GetGpu2DLaunchConfig()
228 Gpu2DLaunchConfig config; in GetGpu2DLaunchConfig()
263 using Gpu3DLaunchConfig = Gpu2DLaunchConfig;
339 Gpu2DLaunchConfig GetGpu2DLaunchConfig(int xdim, int ydim, in GetGpu2DLaunchConfig()
|
| D | gpu_kernel_helper_test.cu.cc | 61 __global__ void Count2D(Gpu2DLaunchConfig config, int bufsize, in Count2D()
222 bool operator==(const Gpu2DLaunchConfig& a, const Gpu2DLaunchConfig& b) { in operator ==()
235 Gpu2DLaunchConfig cfg; in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | generate_box_proposals_op.cu.cc | 47 const Gpu2DLaunchConfig config, const int* d_sorted_scores_keys, in GeneratePreNMSUprightBoxesKernel()
265 __global__ void InitializeDataKernel(const Gpu2DLaunchConfig config, in InitializeDataKernel()
385 Gpu2DLaunchConfig conf2d = in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | svd_op_gpu.cu.cc | 63 __global__ void ComputeValueOfVKernel(Gpu2DLaunchConfig config, int64 m, in ComputeValueOfVKernel()
243 Gpu2DLaunchConfig cfg2D = GetGpu2DLaunchConfig(batch_size, m, d); in RunSVD()
|