/external/tensorflow/tensorflow/core/util/ |
D | gpu_launch_config.h | 111 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()
|
D | gpu_kernel_helper_test.cu.cc | 46 __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/ |
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 | diag_op_gpu.cu.cc | 63 GpuLaunchConfig diag_config = 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 | 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 | dilation_ops_gpu.cu.cc | 197 GpuLaunchConfig config = GetGpuLaunchConfig(total_count, d); in operator ()() 228 GpuLaunchConfig config; in operator ()() 270 GpuLaunchConfig config; in operator ()()
|
D | scatter_functor_gpu.cu.h | 143 GpuLaunchConfig config = GetGpuLaunchConfig(updates_size, d); 164 GpuLaunchConfig config = GetGpuLaunchConfig(synthesized_updates_size, d);
|
D | depthtospace_op_gpu.cu.cc | 163 GpuLaunchConfig config = GetGpuLaunchConfig(total_count, d); in operator ()() 197 GpuLaunchConfig config = GetGpuLaunchConfig(total_count, d); in operator ()()
|
D | bincount_op_gpu.cu.cc | 125 GpuLaunchConfig config = GetGpuLaunchConfig(nthreads, d); in Compute() 206 GpuLaunchConfig config = GetGpuLaunchConfig(num_rows * num_cols, d); in Compute()
|
D | population_count_op_gpu.cu.cc | 76 GpuLaunchConfig config = GetGpuLaunchConfig(total_count, d); \
|
D | segment_reduction_ops_gpu.cu.cc | 145 GpuLaunchConfig config = GetGpuLaunchConfig(output.size(), d); in operator ()() 193 GpuLaunchConfig config = GetGpuLaunchConfig(output.size(), d); in operator ()()
|
D | relu_op_gpu.cu.cc | 188 GpuLaunchConfig config = GetGpuLaunchConfigFixedBlockSize( in operator ()() 229 GpuLaunchConfig config = GetGpuLaunchConfigFixedBlockSize( in operator ()()
|
/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/linalg/ |
D | matrix_set_diag_op_gpu.cu.cc | 117 GpuLaunchConfig config = 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 | 90 GpuLaunchConfig config = in Compute() 151 GpuLaunchConfig config = in Compute()
|
D | lu_op_gpu.cu.cc | 65 GpuLaunchConfig config, const int64 num_rows, in ComputePermutationFromTranspositionsKernel() 227 GpuLaunchConfig cfgPivots = GetGpuLaunchConfig(batch_size, device); in ComputeAsync()
|
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 ()()
|
/external/tensorflow/tensorflow/core/kernels/image/ |
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 | 34 const GpuLaunchConfig config = in operator ()()
|
D | adjust_hue_op_gpu.cu.cc | 33 const GpuLaunchConfig config = in operator ()()
|