Home
last modified time | relevance | path

Searched refs:CUDA_1D_KERNEL_LOOP (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Droll_op_gpu.cu.cc38 CUDA_1D_KERNEL_LOOP(out_idx, nthreads) { in RollKernel()
Drelu_op_gpu.cu.cc200 CUDA_1D_KERNEL_LOOP(index, vect_count) { in Relu_int8x4_kernel()
/external/tensorflow/tensorflow/core/util/
Dgpu_kernel_helper.h37 #define CUDA_1D_KERNEL_LOOP(i, n) \ macro
/external/tensorflow/tensorflow/core/kernels/image/
Dgenerate_box_proposals_op.cu.cc150 CUDA_1D_KERNEL_LOOP(i, nboxes.virtual_thread_count) { in WriteUprightBoxesOutput()