Home
last modified time | relevance | path

Searched refs:GPU_AXIS_KERNEL_LOOP (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dgpu_kernel_helper_test.cu.cc63 GPU_AXIS_KERNEL_LOOP(x, config.virtual_thread_count.x, X) { in Count2D()
67 GPU_AXIS_KERNEL_LOOP(y, config.virtual_thread_count.y, Y) { in Count2D()
78 GPU_AXIS_KERNEL_LOOP(x, config.virtual_thread_count.x, X) { in Count3D()
82 GPU_AXIS_KERNEL_LOOP(y, config.virtual_thread_count.y, Y) { in Count3D()
86 GPU_AXIS_KERNEL_LOOP(z, config.virtual_thread_count.z, Z) { in Count3D()
Dgpu_kernel_helper.h41 #define GPU_AXIS_KERNEL_LOOP(i, n, axis) \ macro
/external/tensorflow/tensorflow/core/kernels/linalg/
Dsvd_op_gpu.cu.cc68 GPU_AXIS_KERNEL_LOOP(batch, config.virtual_thread_count.x, X) { in ComputeValueOfVKernel()
69 GPU_AXIS_KERNEL_LOOP(i, config.virtual_thread_count.y, Y) { in ComputeValueOfVKernel()