HomeSort by relevance Sort by last modified time
    Searched refs:blockDim (Results 1 - 22 of 22) sorted by null

  /external/eigen/unsupported/Eigen/CXX11/src/Tensor/
TensorReductionCuda.h 115 const Index thread_id = blockIdx.x * blockDim.x + threadIdx.x;
116 const Index num_threads = blockDim.x * gridDim.x;
192 eigen_assert(blockDim.x == 1);
205 const Index thread_id = blockIdx.x * blockDim.x + threadIdx.x;
206 const Index num_threads = blockDim.x * gridDim.x;
378 eigen_assert(blockDim.y == 1);
379 eigen_assert(blockDim.z == 1);
386 const Index input_col_blocks = divup<Index>(num_coeffs_to_reduce, blockDim.x * NumPerThread);
389 const Index num_threads = blockDim.x * gridDim.x;
390 const Index thread_id = blockIdx.x * blockDim.x + threadIdx.x
    [all...]
TensorRandom.h 24 blockIdx.x * blockDim.x + threadIdx.x +
25 gridDim.x * blockDim.x * (blockIdx.y * blockDim.y + threadIdx.y);
TensorConvolution.h 584 const int first_plane = blockIdx.y * blockDim.y;
585 const int plane_stride = blockDim.y * gridDim.y;
592 for (int i = threadIdx.x; i < num_x_input; i += blockDim.x) {
603 for (int i = threadIdx.x; i < num_x_output; i += blockDim.x) {
638 const int first_plane = blockIdx.z * blockDim.z;
639 const int plane_stride = blockDim.z * gridDim.z;
648 for (int j = threadIdx.y; j < num_y_input; j += blockDim.y) {
651 for (int i = threadIdx.x; i < num_x_input; i += blockDim.x) {
663 for (int j = threadIdx.y; j < num_y_output; j += blockDim.y) {
665 for (int i = threadIdx.x; i < num_x_output; i += blockDim.x)
    [all...]
TensorExecutor.h 239 const Index first_index = blockIdx.x * blockDim.x + threadIdx.x;
240 const Index step_size = blockDim.x * gridDim.x;
TensorScan.h 253 Index val = threadIdx.x + blockIdx.x * blockDim.x;
  /prebuilts/clang/host/darwin-x86/clang-2812033/lib64/clang/3.8/include/
cuda_builtin_vars.h 98 __CUDA_BUILTIN_VAR __cuda_builtin_blockDim_t blockDim;
  /prebuilts/clang/host/linux-x86/clang-2812033/lib64/clang/3.8/include/
cuda_builtin_vars.h 98 __CUDA_BUILTIN_VAR __cuda_builtin_blockDim_t blockDim;
  /external/eigen/test/
cuda_common.h 11 dim3 threadIdx, blockDim, blockIdx;
26 int i = threadIdx.x + blockIdx.x*blockDim.x;
  /external/clang/lib/Headers/
cuda_builtin_vars.h 92 // blockDim should be convertible to dim3 (in fact in nvcc, it *is* a
114 __CUDA_BUILTIN_VAR __cuda_builtin_blockDim_t blockDim;
  /prebuilts/clang/host/darwin-x86/clang-3016494/lib64/clang/3.8/include/
cuda_builtin_vars.h 92 // blockDim should be convertible to dim3 (in fact in nvcc, it *is* a
114 __CUDA_BUILTIN_VAR __cuda_builtin_blockDim_t blockDim;
  /prebuilts/clang/host/darwin-x86/clang-3217047/lib64/clang/3.8/include/
cuda_builtin_vars.h 92 // blockDim should be convertible to dim3 (in fact in nvcc, it *is* a
114 __CUDA_BUILTIN_VAR __cuda_builtin_blockDim_t blockDim;
  /prebuilts/clang/host/darwin-x86/clang-3289846/lib64/clang/3.8/include/
cuda_builtin_vars.h 92 // blockDim should be convertible to dim3 (in fact in nvcc, it *is* a
114 __CUDA_BUILTIN_VAR __cuda_builtin_blockDim_t blockDim;
  /prebuilts/clang/host/darwin-x86/clang-3362437/lib64/clang/3.8/include/
cuda_builtin_vars.h 92 // blockDim should be convertible to dim3 (in fact in nvcc, it *is* a
114 __CUDA_BUILTIN_VAR __cuda_builtin_blockDim_t blockDim;
  /prebuilts/clang/host/darwin-x86/clang-3688880/lib64/clang/4.0/include/
__clang_cuda_builtin_vars.h 92 // blockDim should be convertible to dim3 (in fact in nvcc, it *is* a
114 __CUDA_BUILTIN_VAR __cuda_builtin_blockDim_t blockDim;
  /prebuilts/clang/host/darwin-x86/clang-3859424/lib64/clang/4.0/include/
__clang_cuda_builtin_vars.h 92 // blockDim should be convertible to dim3 (in fact in nvcc, it *is* a
114 __CUDA_BUILTIN_VAR __cuda_builtin_blockDim_t blockDim;
  /prebuilts/clang/host/linux-x86/clang-3016494/lib64/clang/3.8/include/
cuda_builtin_vars.h 92 // blockDim should be convertible to dim3 (in fact in nvcc, it *is* a
114 __CUDA_BUILTIN_VAR __cuda_builtin_blockDim_t blockDim;
  /prebuilts/clang/host/linux-x86/clang-3217047/lib64/clang/3.8/include/
cuda_builtin_vars.h 92 // blockDim should be convertible to dim3 (in fact in nvcc, it *is* a
114 __CUDA_BUILTIN_VAR __cuda_builtin_blockDim_t blockDim;
  /prebuilts/clang/host/linux-x86/clang-3289846/lib64/clang/3.8/include/
cuda_builtin_vars.h 92 // blockDim should be convertible to dim3 (in fact in nvcc, it *is* a
114 __CUDA_BUILTIN_VAR __cuda_builtin_blockDim_t blockDim;
  /prebuilts/clang/host/linux-x86/clang-3362437/lib64/clang/3.8/include/
cuda_builtin_vars.h 92 // blockDim should be convertible to dim3 (in fact in nvcc, it *is* a
114 __CUDA_BUILTIN_VAR __cuda_builtin_blockDim_t blockDim;
  /prebuilts/clang/host/linux-x86/clang-3688880/lib64/clang/4.0/include/
__clang_cuda_builtin_vars.h 92 // blockDim should be convertible to dim3 (in fact in nvcc, it *is* a
114 __CUDA_BUILTIN_VAR __cuda_builtin_blockDim_t blockDim;
  /prebuilts/clang/host/linux-x86/clang-3859424/lib64/clang/4.0/include/
__clang_cuda_builtin_vars.h 92 // blockDim should be convertible to dim3 (in fact in nvcc, it *is* a
114 __CUDA_BUILTIN_VAR __cuda_builtin_blockDim_t blockDim;
  /prebuilts/sdk/renderscript/clang-include/
cuda_builtin_vars.h 92 // blockDim should be convertible to dim3 (in fact in nvcc, it *is* a
114 __CUDA_BUILTIN_VAR __cuda_builtin_blockDim_t blockDim;

Completed in 254 milliseconds