| /external/tensorflow/tensorflow/core/util/ |
| D | gpu_kernel_helper_test.cu.cc | 178 TEST_F(GpuLaunchConfigTest, GetGpuLaunchConfig) { in TEST_F() argument
183 cfg = GetGpuLaunchConfig(bufsize, d); \ in TEST_F()
188 cfg = GetGpuLaunchConfig(work_element_count, d); \ in TEST_F()
196 cfg = GetGpuLaunchConfig(bufsize, d, SetOutbufZero, 0, 0); \ in TEST_F()
201 cfg = GetGpuLaunchConfig(work_element_count, d, Count1D, 0, 0); \ in TEST_F()
240 cfg1d = GetGpuLaunchConfig(bufsize, d); \ in TEST_F()
253 cfg1d = GetGpuLaunchConfig(bufsize, d, SetOutbufZero, 0, 0); \ in TEST_F()
285 cfg1d = GetGpuLaunchConfig(bufsize, d, SetOutbufZero, 0, 0); \ in TEST_F()
|
| D | gpu_launch_config.h | 127 inline GpuLaunchConfig GetGpuLaunchConfig(int work_element_count, in GetGpuLaunchConfig() function
148 return GetGpuLaunchConfig(work_element_count, d); in GetCudaLaunchConfig()
156 GpuLaunchConfig GetGpuLaunchConfig(int work_element_count, in GetGpuLaunchConfig() function
185 CREATE_CUDA_HOST_FUNCTION_ALIAS(GetGpuLaunchConfig, GetCudaLaunchConfig);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | dilation_ops_gpu.cu.cc | 197 GpuLaunchConfig config = GetGpuLaunchConfig(total_count, d); in operator ()()
232 config = GetGpuLaunchConfig(total_count, d); in operator ()()
239 config = GetGpuLaunchConfig(total_count, d); in operator ()()
274 config = GetGpuLaunchConfig(total_count, d); in operator ()()
281 config = GetGpuLaunchConfig(total_count, d); in operator ()()
|
| D | cwise_op_clip_gpu.cu.cc | 71 GpuLaunchConfig config = GetGpuLaunchConfig(in0_flat.size(), d); in operator ()()
87 GpuLaunchConfig config = GetGpuLaunchConfig(in0_flat.size(), d); in operator ()()
103 GpuLaunchConfig config = GetGpuLaunchConfig(in0_flat.size(), d); in operator ()()
|
| D | segment_reduction_ops_gpu.cu.cc | 145 GpuLaunchConfig config = GetGpuLaunchConfig(output.size(), d); in operator ()()
171 config = GetGpuLaunchConfig(total_stripe_count, d); in operator ()()
193 GpuLaunchConfig config = GetGpuLaunchConfig(output.size(), d); in operator ()()
209 config = GetGpuLaunchConfig(data_size, d); in operator ()()
|
| D | diag_op_gpu.cu.cc | 64 GetGpuLaunchConfig(virtual_thread_count, device); in operator ()()
101 GpuLaunchConfig diag_config = GetGpuLaunchConfig(size, device); in operator ()()
|
| D | gather_functor_gpu.cu.h | 96 GpuLaunchConfig config = GetGpuLaunchConfig(
104 GpuLaunchConfig config = GetGpuLaunchConfig(
|
| D | depthtospace_op_gpu.cu.cc | 163 GpuLaunchConfig config = GetGpuLaunchConfig(total_count, d); in operator ()()
197 GpuLaunchConfig config = GetGpuLaunchConfig(total_count, d); in operator ()()
228 auto config = GetGpuLaunchConfig(total_count, d); in operator ()()
|
| D | inplace_ops_functor_gpu.cu.cc | 47 GpuLaunchConfig cfg = GetGpuLaunchConfig(nelem, d); in DoParallelConcatUpdate()
112 GpuLaunchConfig cfg = GetGpuLaunchConfig(nelem, d); in DoInplaceOp()
147 GpuLaunchConfig cfg = GetGpuLaunchConfig(nelem, d); in DoInplaceOp()
|
| D | spacetodepth_op_gpu.cu.cc | 160 GpuLaunchConfig config = GetGpuLaunchConfig(total_count, d); in operator ()()
194 GpuLaunchConfig config = GetGpuLaunchConfig(total_count, d); in operator ()()
225 GpuLaunchConfig config = GetGpuLaunchConfig(total_count, d); in operator ()()
|
| D | searchsorted_op_gpu.cu.cc | 73 GpuLaunchConfig config = GetGpuLaunchConfig(values.size(), device); in Compute()
96 GpuLaunchConfig config = GetGpuLaunchConfig(values.size(), device); in Compute()
|
| D | scatter_functor_gpu.cu.h | 143 GpuLaunchConfig config = GetGpuLaunchConfig(updates_size, d);
164 GpuLaunchConfig config = GetGpuLaunchConfig(synthesized_updates_size, d);
|
| D | dynamic_stitch_op_gpu.cu.cc | 58 auto config = GetGpuLaunchConfig(output_size, gpu_device); in DynamicStitchGPUImpl()
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | kernels_gpu.cu.cc | 155 GpuLaunchConfig config = GetGpuLaunchConfig(size, d); in operator ()()
232 GpuLaunchConfig config = GetGpuLaunchConfig(size, d); in operator ()()
246 GpuLaunchConfig config = GetGpuLaunchConfig(size, d); in operator ()()
314 GpuLaunchConfig config = GetGpuLaunchConfig(total_nnz, d); in CSRSparseMatrixBatchMulVecImpl()
433 GpuLaunchConfig config = GetGpuLaunchConfig(rows /*size*/, d); in CSRSparseMatrixSoftmaxGPUImpl()
452 GpuLaunchConfig config = GetGpuLaunchConfig(size, d); in CSRSparseMatrixSoftmaxGPUImpl()
628 GpuLaunchConfig config = GetGpuLaunchConfig(rows /*size*/, d); in CSRSparseMatrixSoftmaxGradGPUImpl()
656 GpuLaunchConfig config = GetGpuLaunchConfig(size, d); in CSRSparseMatrixSoftmaxGradGPUImpl()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | crop_and_resize_op_gpu.cu.cc | 375 GpuLaunchConfig config = GetGpuLaunchConfig(total_count, d); in operator ()()
411 config = GetGpuLaunchConfig(total_count, d); in operator ()()
426 config = GetGpuLaunchConfig(total_count, d); in operator ()()
461 config = GetGpuLaunchConfig(total_count, d); in operator ()()
470 config = GetGpuLaunchConfig(total_count, d); in operator ()()
|
| D | resize_nearest_neighbor_op_gpu.cu.cc | 175 GpuLaunchConfig config = GetGpuLaunchConfig(output_size, d); in operator ()()
219 GpuLaunchConfig output_config = GetGpuLaunchConfig(output_size, d); in operator ()()
228 GpuLaunchConfig input_config = GetGpuLaunchConfig(input_size, d); in operator ()()
|
| D | adjust_saturation_op_gpu.cu.cc | 35 GetGpuLaunchConfig(number_of_elements, *device); in operator ()()
|
| D | adjust_hue_op_gpu.cu.cc | 34 GetGpuLaunchConfig(number_of_elements, *device); in operator ()()
|
| D | resize_bilinear_op_gpu.cu.cc | 417 GpuLaunchConfig config = GetGpuLaunchConfig(total_count, d); in operator ()()
436 config = GetGpuLaunchConfig( in operator ()()
483 config = GetGpuLaunchConfig(total_count, d); in operator ()()
504 config = GetGpuLaunchConfig(total_count, d); in operator ()()
|
| D | non_max_suppression_op.cu.cc | 271 auto config = GetGpuLaunchConfig(d_nms_mask.NumElements(), device); in NmsGpu()
311 config = GetGpuLaunchConfig(num_boxes, device); in NmsGpu()
460 auto config = GetGpuLaunchConfig(num_boxes, device); in DoNMS()
524 config = GetGpuLaunchConfig(max_output_size, device); in DoNMS()
538 config = GetGpuLaunchConfig(num_outputs, device); in DoNMS()
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_set_diag_op_gpu.cu.cc | 118 GetGpuLaunchConfig(batch_size * num_diags * max_diag_len, device); in Compute()
126 GpuLaunchConfig config = GetGpuLaunchConfig(batch_size * m * n, device); in Compute()
|
| D | determinant_op_gpu.cu.cc | 130 GpuLaunchConfig config = GetGpuLaunchConfig(num_matrices, device); in operator ()()
153 GpuLaunchConfig config = GetGpuLaunchConfig(num_matrices, device); in operator ()()
|
| D | matrix_diag_op_gpu.cu.cc | 91 GetGpuLaunchConfig(batch_size * num_rows * num_cols, device); in Compute()
152 GetGpuLaunchConfig(batch_size * num_diags * max_diag_len, device); in Compute()
|
| D | eye_functor_gpu.cu.cc | 54 GpuLaunchConfig config = GetGpuLaunchConfig(batch_size * m * n, device); in operator ()()
|
| D | matrix_band_part_op_gpu.cu.cc | 60 GpuLaunchConfig config = GetGpuLaunchConfig(batch_size * m * n, device); in operator ()()
|