Searched refs:CUDA_AXIS_KERNEL_LOOP (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | generate_box_proposals_op.cu.cc | 58 CUDA_AXIS_KERNEL_LOOP(image_index, config.virtual_thread_count.y, Y) { in GeneratePreNMSUprightBoxesKernel()
59 CUDA_AXIS_KERNEL_LOOP(ibox, config.virtual_thread_count.x, X) { in GeneratePreNMSUprightBoxesKernel()
270 CUDA_AXIS_KERNEL_LOOP(img_idx, config.virtual_thread_count.y, Y) { in InitializeDataKernel()
271 CUDA_AXIS_KERNEL_LOOP(box_idx, config.virtual_thread_count.x, X) { in InitializeDataKernel()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | gpu_kernel_helper.h | 43 #define CUDA_AXIS_KERNEL_LOOP(i, n, axis) \ macro
|