| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_launch_config.h | 110 struct CudaLaunchConfig { struct
125 inline CudaLaunchConfig GetCudaLaunchConfig(int work_element_count, in GetCudaLaunchConfig() argument
128 CudaLaunchConfig config; in GetCudaLaunchConfig()
148 inline CudaLaunchConfig GetCudaLaunchConfig(int work_element_count, in GetCudaLaunchConfig()
154 CudaLaunchConfig config; in GetCudaLaunchConfig()
177 inline CudaLaunchConfig GetCudaLaunchConfigFixedBlockSize( in GetCudaLaunchConfigFixedBlockSize()
181 CudaLaunchConfig config; in GetCudaLaunchConfigFixedBlockSize()
|
| D | cuda_kernel_helper_test.cu.cc | 43 __global__ void SetOutbufZero(CudaLaunchConfig config, int* outbuf) { in SetOutbufZero()
48 __global__ void Count1D(CudaLaunchConfig config, int bufsize, int* outbuf) { in Count1D()
152 CudaLaunchConfig cfg; in TEST_F()
205 CudaLaunchConfig cfg1d; in TEST_F()
249 CudaLaunchConfig cfg1d; in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cwise_op_clip_gpu.cu.cc | 63 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 ()()
|
| D | fused_batch_norm_op.cu.cc | 40 CudaLaunchConfig config = GetCudaLaunchConfig(channels, d); in operator ()()
62 CudaLaunchConfig config = GetCudaLaunchConfig(channels, d); in operator ()()
|
| D | matrix_set_diag_op_gpu.cu.cc | 74 CudaLaunchConfig config = in Compute()
82 CudaLaunchConfig config = GetCudaLaunchConfig(batch_size * m * n, device); in Compute()
|
| D | diag_op_gpu.cu.cc | 62 CudaLaunchConfig diag_config = in operator ()()
103 CudaLaunchConfig diag_config = GetCudaLaunchConfig(size, device); in operator ()()
|
| D | spacetodepth_op_gpu.cu.cc | 160 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 ()()
|
| D | inplace_ops_functor_gpu.cu.cc | 46 CudaLaunchConfig cfg = GetCudaLaunchConfig(nelem, d); in DoParallelConcatUpdate()
109 CudaLaunchConfig cfg = GetCudaLaunchConfig(nelem, d); in DoInplaceOp()
144 CudaLaunchConfig cfg = GetCudaLaunchConfig(nelem, d); in DoInplaceOp()
|
| D | searchsorted_op_gpu.cu.cc | 68 CudaLaunchConfig config = in Compute()
88 CudaLaunchConfig config = in Compute()
|
| D | resize_nearest_neighbor_op_gpu.cu.cc | 176 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 ()()
|
| D | relu_op_gpu.cu.cc | 107 CudaLaunchConfig config = GetCudaLaunchConfigFixedBlockSize( in operator ()()
136 CudaLaunchConfig config = GetCudaLaunchConfigFixedBlockSize( in operator ()()
|
| D | determinant_op_gpu.cu.cc | 131 CudaLaunchConfig config = GetCudaLaunchConfig(num_matrices, device); in operator ()()
154 CudaLaunchConfig config = GetCudaLaunchConfig(num_matrices, device); in operator ()()
|
| D | dilation_ops_gpu.cu.cc | 197 CudaLaunchConfig config = GetCudaLaunchConfig(total_count, d); in operator ()()
228 CudaLaunchConfig config; in operator ()()
270 CudaLaunchConfig config; in operator ()()
|
| D | scatter_functor_gpu.cu.h | 129 CudaLaunchConfig config = GetCudaLaunchConfig(updates_size, d);
150 CudaLaunchConfig config = GetCudaLaunchConfig(synthesized_updates_size, d);
|
| D | lu_op_gpu.cu.cc | 64 CudaLaunchConfig config, const int64 num_rows, const int* all_pivots, in ComputePermutationFromTranspositionsKernel()
225 CudaLaunchConfig cfgPivots = GetCudaLaunchConfig(batch_size, device); in ComputeAsync()
|
| D | adjust_saturation_op_gpu.cu.cc | 34 const CudaLaunchConfig config = in operator ()()
|
| D | adjust_hue_op_gpu.cu.cc | 33 const CudaLaunchConfig config = in operator ()()
|
| D | depthtospace_op_gpu.cu.cc | 164 CudaLaunchConfig config = GetCudaLaunchConfig(total_count, d); in operator ()()
198 CudaLaunchConfig config = GetCudaLaunchConfig(total_count, d); in operator ()()
|
| D | segment_reduction_ops_gpu.cu.cc | 140 CudaLaunchConfig config = GetCudaLaunchConfig(output.size(), d); in operator ()()
186 CudaLaunchConfig config = GetCudaLaunchConfig(output.size(), d); in operator ()()
|
| D | eye_functor_gpu.cu.cc | 54 CudaLaunchConfig config = GetCudaLaunchConfig(batch_size * m * n, device); in operator ()()
|
| D | matrix_band_part_op_gpu.cu.cc | 60 CudaLaunchConfig config = GetCudaLaunchConfig(batch_size * m * n, device); in operator ()()
|
| D | crop_and_resize_op_gpu.cu.cc | 373 CudaLaunchConfig config = GetCudaLaunchConfig(total_count, d); in operator ()()
403 CudaLaunchConfig config; in operator ()()
453 CudaLaunchConfig config; in operator ()()
|
| D | split_lib_gpu.cu.cc | 202 CudaLaunchConfig config = GetCudaLaunchConfig( in Run()
218 CudaLaunchConfig config = in Run()
|
| D | population_count_op_gpu.cu.cc | 73 CudaLaunchConfig config = GetCudaLaunchConfig(total_count, d); \
|
| /external/tensorflow/tensorflow/contrib/resampler/kernels/ |
| D | resampler_ops_gpu.cu.cc | 120 ::tensorflow::CudaLaunchConfig config = in operator ()()
255 ::tensorflow::CudaLaunchConfig config = in operator ()()
|