Searched refs:GetGpu2DLaunchConfig (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | gpu_launch_config.h | 226 inline Gpu2DLaunchConfig GetGpu2DLaunchConfig(int xdim, int ydim, in GetGpu2DLaunchConfig() function
256 return GetGpu2DLaunchConfig(xdim, ydim, d); in GetCuda2DLaunchConfig()
339 Gpu2DLaunchConfig GetGpu2DLaunchConfig(int xdim, int ydim, in GetGpu2DLaunchConfig() function
347 CREATE_CUDA_HOST_FUNCTION_ALIAS(GetGpu2DLaunchConfig, GetCuda2DLaunchConfig);
356 return GetGpu2DLaunchConfig(xdim, ydim, d, func, dynamic_shared_memory_size, in GetCuda2DLaunchConfig()
|
| D | gpu_kernel_helper_test.cu.cc | 234 TEST_F(GpuLaunchConfigTest, GetGpu2DLaunchConfig) { in TEST_F() argument
245 cfg = GetGpu2DLaunchConfig(dimx, dimy, d); \ in TEST_F()
258 cfg = GetGpu2DLaunchConfig(dimx, dimy, d, Count2D, 0, 0); \ in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | generate_box_proposals_op.cu.cc | 386 GetGpu2DLaunchConfig(conv_layer_nboxes, num_images, d); in Compute()
409 conf2d = GetGpu2DLaunchConfig(nboxes_to_generate, num_images, d); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | concat_lib_gpu_impl.cu.cc | 143 auto config = GetGpu2DLaunchConfig(output->dimension(1), output->dimension(0), in ConcatGPUImpl()
|
| D | split_lib_gpu.cu.cc | 223 auto config = GetGpu2DLaunchConfig(total_cols, total_rows, gpu_device); in Run()
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | svd_op_gpu.cu.cc | 243 Gpu2DLaunchConfig cfg2D = GetGpu2DLaunchConfig(batch_size, m, d); in RunSVD()
|