Home
last modified time | relevance | path

Searched refs:GetCudaLaunchConfig (Results 1 – 25 of 48) sorted by relevance

12

/external/tensorflow/tensorflow/core/util/
Dcuda_kernel_helper_test.cu.cc151 TEST_F(CudaLaunchConfigTest, GetCudaLaunchConfig) { in TEST_F() argument
156 cfg = GetCudaLaunchConfig(bufsize, d); \ in TEST_F()
161 cfg = GetCudaLaunchConfig(work_element_count, d); \ in TEST_F()
167 cfg = GetCudaLaunchConfig(bufsize, d, SetOutbufZero, 0, 0); \ in TEST_F()
172 cfg = GetCudaLaunchConfig(work_element_count, d, Count1D, 0, 0); \ in TEST_F()
209 cfg1d = GetCudaLaunchConfig(bufsize, d); \ in TEST_F()
221 cfg1d = GetCudaLaunchConfig(bufsize, d, SetOutbufZero, 0, 0); \ in TEST_F()
253 cfg1d = GetCudaLaunchConfig(bufsize, d, SetOutbufZero, 0, 0); \ in TEST_F()
Dcuda_launch_config.h125 inline CudaLaunchConfig GetCudaLaunchConfig(int work_element_count, in GetCudaLaunchConfig() function
148 inline CudaLaunchConfig GetCudaLaunchConfig(int work_element_count, in GetCudaLaunchConfig() function
/external/tensorflow/tensorflow/core/kernels/
Dsegment_reduction_ops_gpu.cu.cc140 CudaLaunchConfig config = GetCudaLaunchConfig(output.size(), d); in operator ()()
165 config = GetCudaLaunchConfig(total_stripe_count, d); in operator ()()
186 CudaLaunchConfig config = GetCudaLaunchConfig(output.size(), d); in operator ()()
200 config = GetCudaLaunchConfig(data_size, d); in operator ()()
Dcwise_op_clip_gpu.cu.cc63 CudaLaunchConfig config = GetCudaLaunchConfig(in0_flat.size(), d); in operator ()()
79 CudaLaunchConfig config = GetCudaLaunchConfig(in0_flat.size(), d); in operator ()()
95 CudaLaunchConfig config = GetCudaLaunchConfig(in0_flat.size(), d); in operator ()()
Ddilation_ops_gpu.cu.cc197 CudaLaunchConfig config = GetCudaLaunchConfig(total_count, d); in operator ()()
232 config = GetCudaLaunchConfig(total_count, d); in operator ()()
239 config = GetCudaLaunchConfig(total_count, d); in operator ()()
274 config = GetCudaLaunchConfig(total_count, d); in operator ()()
281 config = GetCudaLaunchConfig(total_count, d); in operator ()()
Dfused_batch_norm_op.cu.cc40 CudaLaunchConfig config = GetCudaLaunchConfig(channels, d); in operator ()()
62 CudaLaunchConfig config = GetCudaLaunchConfig(channels, d); in operator ()()
Dmatrix_set_diag_op_gpu.cu.cc75 GetCudaLaunchConfig(batch_size * minsize, device); in Compute()
82 CudaLaunchConfig config = GetCudaLaunchConfig(batch_size * m * n, device); in Compute()
Ddiag_op_gpu.cu.cc63 GetCudaLaunchConfig(virtual_thread_count, device); in operator ()()
103 CudaLaunchConfig diag_config = GetCudaLaunchConfig(size, device); in operator ()()
Dcrop_and_resize_op_gpu.cu.cc373 CudaLaunchConfig config = GetCudaLaunchConfig(total_count, d); in operator ()()
408 config = GetCudaLaunchConfig(total_count, d); in operator ()()
423 config = GetCudaLaunchConfig(total_count, d); in operator ()()
458 config = GetCudaLaunchConfig(total_count, d); in operator ()()
467 config = GetCudaLaunchConfig(total_count, d); in operator ()()
Ddepthtospace_op_gpu.cu.cc164 CudaLaunchConfig config = GetCudaLaunchConfig(total_count, d); in operator ()()
198 CudaLaunchConfig config = GetCudaLaunchConfig(total_count, d); in operator ()()
229 auto config = GetCudaLaunchConfig(total_count, d); in operator ()()
Dspacetodepth_op_gpu.cu.cc160 CudaLaunchConfig config = GetCudaLaunchConfig(total_count, d); in operator ()()
194 CudaLaunchConfig config = GetCudaLaunchConfig(total_count, d); in operator ()()
225 CudaLaunchConfig config = GetCudaLaunchConfig(total_count, d); in operator ()()
Dinplace_ops_functor_gpu.cu.cc46 CudaLaunchConfig cfg = GetCudaLaunchConfig(nelem, d); in DoParallelConcatUpdate()
109 CudaLaunchConfig cfg = GetCudaLaunchConfig(nelem, d); in DoInplaceOp()
144 CudaLaunchConfig cfg = GetCudaLaunchConfig(nelem, d); in DoInplaceOp()
Dsearchsorted_op_gpu.cu.cc69 GetCudaLaunchConfig(values.size(), context->eigen_gpu_device()); in Compute()
89 GetCudaLaunchConfig(values.size(), context->eigen_gpu_device()); in Compute()
Dresize_nearest_neighbor_op_gpu.cu.cc176 CudaLaunchConfig config = GetCudaLaunchConfig(output_size, d); in operator ()()
222 CudaLaunchConfig output_config = GetCudaLaunchConfig(output_size, d); in operator ()()
230 CudaLaunchConfig input_config = GetCudaLaunchConfig(input_size, d); in operator ()()
Ddeterminant_op_gpu.cu.cc131 CudaLaunchConfig config = GetCudaLaunchConfig(num_matrices, device); in operator ()()
154 CudaLaunchConfig config = GetCudaLaunchConfig(num_matrices, device); in operator ()()
Dresize_bilinear_op_gpu.cu.cc282 CudaLaunchConfig config = GetCudaLaunchConfig(total_count, d); in operator ()()
321 config = GetCudaLaunchConfig(total_count, d); in operator ()()
328 config = GetCudaLaunchConfig(total_count, d); in operator ()()
Dscatter_functor_gpu.cu.h129 CudaLaunchConfig config = GetCudaLaunchConfig(updates_size, d);
150 CudaLaunchConfig config = GetCudaLaunchConfig(synthesized_updates_size, d);
Dadjust_saturation_op_gpu.cu.cc35 GetCudaLaunchConfig(number_of_elements, *device); in operator ()()
Dadjust_hue_op_gpu.cu.cc34 GetCudaLaunchConfig(number_of_elements, *device); in operator ()()
Deye_functor_gpu.cu.cc54 CudaLaunchConfig config = GetCudaLaunchConfig(batch_size * m * n, device); in operator ()()
Dmatrix_band_part_op_gpu.cu.cc60 CudaLaunchConfig config = GetCudaLaunchConfig(batch_size * m * n, device); in operator ()()
Ddynamic_stitch_op_gpu.cu.cc58 auto config = GetCudaLaunchConfig(output_size, gpu_device); in DynamicStitchGPUImpl()
Dsplit_lib_gpu.cu.cc202 CudaLaunchConfig config = GetCudaLaunchConfig( in Run()
219 GetCudaLaunchConfig(total_rows * total_cols, gpu_device); in Run()
Dpopulation_count_op_gpu.cu.cc73 CudaLaunchConfig config = GetCudaLaunchConfig(total_count, d); \
/external/tensorflow/tensorflow/contrib/resampler/kernels/
Dresampler_ops_gpu.cu.cc121 ::tensorflow::GetCudaLaunchConfig(output_data_size, d); in operator ()()
256 ::tensorflow::GetCudaLaunchConfig(grad_warp_size, d); in operator ()()
261 config = ::tensorflow::GetCudaLaunchConfig(grad_data_size, d); in operator ()()
268 config = ::tensorflow::GetCudaLaunchConfig(resampler_output_size, d); in operator ()()

12