Home
last modified time | relevance | path

Searched defs:IDX_DIM (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/cuda/
DIndexing.cu893 #define SMALL_INDEX(TENSOR_TYPE, INDICES_TYPE, TYPE, SELF_DIM, SOURCE_DIM, IDX_DIM) \ in index_add_cuda_impl() argument
902 SELF_DIM, SOURCE_DIM, IDX_DIM, IDX_IS_MAJOR) \ in index_add_cuda_impl() argument
1066 #define SMALL_INDEX(TENSOR_TYPE, INDICES_TYPE, TYPE, SELF_DIM, SOURCE_DIM, IDX_DIM) … in index_reduce_func_cuda_impl() argument
1075 SELF_DIM, SOURCE_DIM, IDX_DIM, IDX_IS_MAJOR) \ in index_reduce_func_cuda_impl() argument
1360 #define SMALL_INDEX(TENSOR_TYPE, INDICES_TYPE, TYPE, DST_DIM, SRC_DIM, IDX_DIM) \ in index_select_out_cuda_impl() argument
1369 DST_DIM, SRC_DIM, IDX_DIM, IDX_IS_MAJOR) \ in index_select_out_cuda_impl() argument