Searched defs:virtual_thread_count (Results 1 - 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| H A D | diag_op_gpu.cu.cc | 52 // CudaLaunchConfig uses an int for virtual_thread_count,
58 int virtual_thread_count = int(size * size); local
63 GetCudaLaunchConfig(virtual_thread_count, device);
65 device.stream()>>>(diag_config.virtual_thread_count, size,
105 0, device.stream()>>>(diag_config.virtual_thread_count,
|
| /external/tensorflow/tensorflow/core/util/ |
| H A D | cuda_launch_config.h | 51 CUDA_1D_KERNEL_LOOP(x, config.virtual_thread_count) {
57 CUDA_AXIS_KERNEL_LOOP(x, config.virtual_thread_count, x) {
58 CUDA_AXIS_KERNEL_LOOP(y, config.virtual_thread_count, y) {
65 CUDA_AXIS_KERNEL_LOOP(x, config.virtual_thread_count, x) {
66 CUDA_AXIS_KERNEL_LOOP(y, config.virtual_thread_count, y) {
67 CUDA_AXIS_KERNEL_LOOP(z, config.virtual_thread_count, z) {
114 int virtual_thread_count = -1; member in struct:tensorflow::CudaLaunchConfig
129 const int virtual_thread_count = work_element_count; local
132 virtual_thread_count);
138 config.virtual_thread_count
197 dim3 virtual_thread_count = dim3(0, 0, 0); member in struct:tensorflow::Cuda2DLaunchConfig
[all...] |
Completed in 336 milliseconds