| /external/tensorflow/tensorflow/core/util/ |
| D | gpu_kernel_helper_test.cu.cc | 48 GPU_1D_KERNEL_LOOP(x, config.virtual_thread_count) { outbuf[x] = 0; } in SetOutbufZero()
54 GPU_1D_KERNEL_LOOP(x, config.virtual_thread_count) { in Count1D()
63 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()
71 int idx = x * 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()
91 x * config.virtual_thread_count.y * config.virtual_thread_count.z + in Count3D()
92 y * config.virtual_thread_count.z + z; in Count3D()
|
| D | gpu_launch_config.h | 115 int virtual_thread_count = -1; member
131 const int virtual_thread_count = work_element_count; in GetGpuLaunchConfig() local
134 virtual_thread_count); in GetGpuLaunchConfig()
140 config.virtual_thread_count = virtual_thread_count; in GetGpuLaunchConfig()
180 config.virtual_thread_count = work_element_count; in GetGpuLaunchConfig()
211 config.virtual_thread_count = work_element_count; in GetGpuLaunchConfigFixedBlockSize()
220 dim3 virtual_thread_count = dim3(0, 0, 0); member
244 config.virtual_thread_count = dim3(xdim, ydim, 1); in GetGpu2DLaunchConfig()
331 config.virtual_thread_count = dim3(xdim, ydim, zdim); in GetGpu3DLaunchConfig()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | diag_op_gpu.cu.cc | 59 int virtual_thread_count = int(size * size); in operator ()() local
64 GetGpuLaunchConfig(virtual_thread_count, device); in operator ()()
67 0, device.stream(), diag_config.virtual_thread_count, size, in, out)); in operator ()()
105 diag_config.virtual_thread_count, size, in, out)); in operator ()()
|
| D | inplace_ops_functor_gpu.cu.cc | 55 d.stream(), cfg.virtual_thread_count, nrows, ncols, loc, src, dst)); in DoParallelConcatUpdate()
125 d.stream(), cfg.virtual_thread_count, nrows, in DoInplaceOp()
131 cfg.virtual_thread_count, nrows, ncols, n, in DoInplaceOp()
137 cfg.virtual_thread_count, nrows, ncols, n, in DoInplaceOp()
159 d.stream(), cfg.virtual_thread_count, nrows, in DoInplaceOp()
|
| D | conv_2d_gpu.h | 499 0, d.stream(), config.virtual_thread_count,
505 0, d.stream(), config.virtual_thread_count,
536 0, d.stream(), config.virtual_thread_count,
550 0, d.stream(), config.virtual_thread_count,
587 config.thread_per_block, 0, d.stream(), config.virtual_thread_count,
593 config.thread_per_block, 0, d.stream(), config.virtual_thread_count,
1029 d.stream(), config.virtual_thread_count, input,
1091 0, d.stream(), config.virtual_thread_count,
|
| D | dilation_ops_gpu.cu.cc | 201 d.stream(), config.virtual_thread_count, input.data(), filter.data(), in operator ()()
242 config.thread_per_block, 0, d.stream(), config.virtual_thread_count, in operator ()()
284 config.thread_per_block, 0, d.stream(), config.virtual_thread_count, in operator ()()
|
| D | depthtospace_op_gpu.cu.cc | 166 config.virtual_thread_count, input.data(), block_size, batch_size, in operator ()()
231 config.virtual_thread_count, input.data(), block_size, input_width, in operator ()()
|
| D | spacetodepth_op_gpu.cu.cc | 163 config.virtual_thread_count, input.data(), block_size, batch_size, in operator ()()
228 config.virtual_thread_count, input.data(), block_size, output_width, in operator ()()
|
| D | tile_functor_gpu.h | 80 d.stream(), cfg.virtual_thread_count, p, in TileSimple()
|
| D | roll_op_gpu.cu.cc | 81 cfg.virtual_thread_count, num_dims, input, in operator ()()
|
| D | bias_op_gpu.cu.cc | 91 config.virtual_thread_count, input, bias, in compute()
96 config.virtual_thread_count, input, bias, in compute()
|
| D | avgpooling_op_gpu.cu.cc | 94 d.stream(), config.virtual_thread_count, top_diff, num, height, width, in RunAvePoolBackwardNHWC()
|
| D | multinomial_op_gpu.cu.cc | 120 config.thread_per_block, 0, d.stream(), config.virtual_thread_count, in operator ()()
|
| D | spacetobatch_functor_gpu.cu.cc | 146 config.virtual_thread_count, in operator ()()
|
| /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()
150 CUDA_1D_KERNEL_LOOP(i, nboxes.virtual_thread_count) { in WriteUprightBoxesOutput()
179 zconfig.virtual_thread_count, (*t).flat<T>().data()); in ResetTensor()
268 const int image_size = config.virtual_thread_count.x; in InitializeDataKernel()
269 const int num_images = config.virtual_thread_count.y; in InitializeDataKernel()
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()
|
| D | resize_bilinear_op_gpu.cu.cc | 438 config.virtual_thread_count = num_threads_per_pixel; in operator ()()
445 d.stream(), config.virtual_thread_count, images.data(), in operator ()()
493 config.thread_per_block, 0, d.stream(), config.virtual_thread_count, in operator ()()
501 d.stream(), config.virtual_thread_count, output_grad.data())); in operator ()()
508 config.thread_per_block, 0, d.stream(), config.virtual_thread_count, in operator ()()
515 config.thread_per_block, 0, d.stream(), config.virtual_thread_count, in operator ()()
|
| D | crop_and_resize_op_gpu.cu.cc | 378 0, d.stream(), config.virtual_thread_count, image.data(), in operator ()()
414 d.stream(), config.virtual_thread_count, grads_image.data())); in operator ()()
429 config.thread_per_block, 0, d.stream(), config.virtual_thread_count, in operator ()()
464 d.stream(), config.virtual_thread_count, grads_boxes.data())); in operator ()()
473 config.thread_per_block, 0, d.stream(), config.virtual_thread_count, in operator ()()
|
| D | non_max_suppression_op.cu.cc | 274 config.virtual_thread_count, in NmsGpu()
320 config.virtual_thread_count, 0, in NmsGpu()
464 config.virtual_thread_count, 0, in DoNMS()
485 config.virtual_thread_count, sorted_indices, in DoNMS()
527 config.virtual_thread_count, in DoNMS()
541 0, device.stream(), config.virtual_thread_count, in DoNMS()
|
| D | resize_nearest_neighbor_op_gpu.cu.cc | 185 d.stream(), config.virtual_thread_count, input.data(), in operator ()()
239 d.stream(), input_config.virtual_thread_count, input.data(), in_height, in operator ()()
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_set_diag_op_gpu.cu.cc | 122 config.virtual_thread_count, m, n, num_diags, max_diag_len, in Compute()
130 config.virtual_thread_count, m, n, num_diags, max_diag_len, in Compute()
|
| D | determinant_op_gpu.cu.cc | 135 config.virtual_thread_count, n, lu_factor.data(), pivots, nullptr, in operator ()()
157 config.virtual_thread_count, n, lu_factor.data(), pivots, sign.data(), in operator ()()
|
| D | matrix_diag_op_gpu.cu.cc | 94 device.stream(), config.virtual_thread_count, num_rows, num_cols, in Compute()
155 device.stream(), config.virtual_thread_count, num_rows, num_cols, in Compute()
|
| D | eye_functor_gpu.cu.cc | 57 config.virtual_thread_count, batch_size, m, n, in operator ()()
|
| D | matrix_band_part_op_gpu.cu.cc | 63 device.stream(), config.virtual_thread_count, in operator ()()
|
| D | svd_op_gpu.cu.cc | 68 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()
81 GPU_1D_KERNEL_LOOP(i, config.virtual_thread_count) { in ExtractSignOfVKernel()
|