Searched refs:CUDA_AXIS_KERNEL_LOOP (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/contrib/reduce_slice_ops/kernels/ |
| D | reduce_slice_ops_gpu.cu.cc | 42 CUDA_AXIS_KERNEL_LOOP(x, config.virtual_thread_count.x, X) { \
43 CUDA_AXIS_KERNEL_LOOP(y, config.virtual_thread_count.y, Y) { \
44 CUDA_AXIS_KERNEL_LOOP(z, config.virtual_thread_count.z, Z) { \
|
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_kernel_helper_test.cu.cc | 57 CUDA_AXIS_KERNEL_LOOP(x, config.virtual_thread_count.x, X) { in Count2D()
61 CUDA_AXIS_KERNEL_LOOP(y, config.virtual_thread_count.y, Y) { in Count2D()
71 CUDA_AXIS_KERNEL_LOOP(x, config.virtual_thread_count.x, X) { in Count3D()
75 CUDA_AXIS_KERNEL_LOOP(y, config.virtual_thread_count.y, Y) { in Count3D()
79 CUDA_AXIS_KERNEL_LOOP(z, config.virtual_thread_count.z, Z) { in Count3D()
|
| D | cuda_kernel_helper.h | 30 #define CUDA_AXIS_KERNEL_LOOP(i, n, axis) \ macro
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | svd_op_gpu.cu.cc | 66 CUDA_AXIS_KERNEL_LOOP(batch, config.virtual_thread_count.x, X) { in ComputeValueOfVKernel()
67 CUDA_AXIS_KERNEL_LOOP(i, config.virtual_thread_count.y, Y) { in ComputeValueOfVKernel()
|