Home
last modified time | relevance | path

Searched refs:GpuLaunchConfig (Results 1 – 25 of 56) sorted by relevance

123

/external/tensorflow/tensorflow/core/util/
Dgpu_launch_config.h111 struct GpuLaunchConfig { struct
121 CREATE_CUDA_TYPE_ALIAS(GpuLaunchConfig, CudaLaunchConfig); argument
127 inline GpuLaunchConfig GetGpuLaunchConfig(int work_element_count, in GetGpuLaunchConfig()
130 GpuLaunchConfig config; in GetGpuLaunchConfig()
156 GpuLaunchConfig GetGpuLaunchConfig(int work_element_count, in GetGpuLaunchConfig()
161 GpuLaunchConfig config; in GetGpuLaunchConfig()
192 GpuLaunchConfig GetGpuLaunchConfigFixedBlockSize( in GetGpuLaunchConfigFixedBlockSize()
196 GpuLaunchConfig config; in GetGpuLaunchConfigFixedBlockSize()
Dgpu_kernel_helper_test.cu.cc46 __global__ void SetOutbufZero(GpuLaunchConfig config, in SetOutbufZero()
52 __global__ void Count1D(GpuLaunchConfig config, int bufsize, in Count1D()
179 GpuLaunchConfig cfg; in TEST_F()
236 GpuLaunchConfig cfg1d; in TEST_F()
281 GpuLaunchConfig cfg1d; in TEST_F()
/external/tensorflow/tensorflow/core/kernels/
Dcwise_op_clip_gpu.cu.cc71 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 ()()
Ddiag_op_gpu.cu.cc63 GpuLaunchConfig diag_config = in operator ()()
101 GpuLaunchConfig diag_config = GetGpuLaunchConfig(size, device); in operator ()()
Dgather_functor_gpu.cu.h96 GpuLaunchConfig config = GetGpuLaunchConfig(
104 GpuLaunchConfig config = GetGpuLaunchConfig(
Dinplace_ops_functor_gpu.cu.cc47 GpuLaunchConfig cfg = GetGpuLaunchConfig(nelem, d); in DoParallelConcatUpdate()
112 GpuLaunchConfig cfg = GetGpuLaunchConfig(nelem, d); in DoInplaceOp()
147 GpuLaunchConfig cfg = GetGpuLaunchConfig(nelem, d); in DoInplaceOp()
Dspacetodepth_op_gpu.cu.cc160 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 ()()
Dsearchsorted_op_gpu.cu.cc73 GpuLaunchConfig config = GetGpuLaunchConfig(values.size(), device); in Compute()
96 GpuLaunchConfig config = GetGpuLaunchConfig(values.size(), device); in Compute()
Ddilation_ops_gpu.cu.cc197 GpuLaunchConfig config = GetGpuLaunchConfig(total_count, d); in operator ()()
228 GpuLaunchConfig config; in operator ()()
270 GpuLaunchConfig config; in operator ()()
Dscatter_functor_gpu.cu.h143 GpuLaunchConfig config = GetGpuLaunchConfig(updates_size, d);
164 GpuLaunchConfig config = GetGpuLaunchConfig(synthesized_updates_size, d);
Ddepthtospace_op_gpu.cu.cc163 GpuLaunchConfig config = GetGpuLaunchConfig(total_count, d); in operator ()()
197 GpuLaunchConfig config = GetGpuLaunchConfig(total_count, d); in operator ()()
Dbincount_op_gpu.cu.cc125 GpuLaunchConfig config = GetGpuLaunchConfig(nthreads, d); in Compute()
206 GpuLaunchConfig config = GetGpuLaunchConfig(num_rows * num_cols, d); in Compute()
Dpopulation_count_op_gpu.cu.cc76 GpuLaunchConfig config = GetGpuLaunchConfig(total_count, d); \
Dsegment_reduction_ops_gpu.cu.cc145 GpuLaunchConfig config = GetGpuLaunchConfig(output.size(), d); in operator ()()
193 GpuLaunchConfig config = GetGpuLaunchConfig(output.size(), d); in operator ()()
Drelu_op_gpu.cu.cc188 GpuLaunchConfig config = GetGpuLaunchConfigFixedBlockSize( in operator ()()
229 GpuLaunchConfig config = GetGpuLaunchConfigFixedBlockSize( in operator ()()
/external/tensorflow/tensorflow/core/kernels/sparse/
Dkernels_gpu.cu.cc155 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/linalg/
Dmatrix_set_diag_op_gpu.cu.cc117 GpuLaunchConfig config = in Compute()
126 GpuLaunchConfig config = GetGpuLaunchConfig(batch_size * m * n, device); in Compute()
Ddeterminant_op_gpu.cu.cc130 GpuLaunchConfig config = GetGpuLaunchConfig(num_matrices, device); in operator ()()
153 GpuLaunchConfig config = GetGpuLaunchConfig(num_matrices, device); in operator ()()
Dmatrix_diag_op_gpu.cu.cc90 GpuLaunchConfig config = in Compute()
151 GpuLaunchConfig config = in Compute()
Dlu_op_gpu.cu.cc65 GpuLaunchConfig config, const int64 num_rows, in ComputePermutationFromTranspositionsKernel()
227 GpuLaunchConfig cfgPivots = GetGpuLaunchConfig(batch_size, device); in ComputeAsync()
Deye_functor_gpu.cu.cc54 GpuLaunchConfig config = GetGpuLaunchConfig(batch_size * m * n, device); in operator ()()
Dmatrix_band_part_op_gpu.cu.cc60 GpuLaunchConfig config = GetGpuLaunchConfig(batch_size * m * n, device); in operator ()()
/external/tensorflow/tensorflow/core/kernels/image/
Dresize_nearest_neighbor_op_gpu.cu.cc175 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 ()()
Dadjust_saturation_op_gpu.cu.cc34 const GpuLaunchConfig config = in operator ()()
Dadjust_hue_op_gpu.cu.cc33 const GpuLaunchConfig config = in operator ()()

123