Searched defs:INDICES_TYPE (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | Indexing.cu | 893 #define SMALL_INDEX(TENSOR_TYPE, INDICES_TYPE, TYPE, SELF_DIM, SOURCE_DIM, IDX_DIM) \ in index_add_cuda_impl() argument
901 #define LARGE_INDEX(TENSOR_TYPE, INDICES_TYPE, TYPE, \ 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
1074 #define LARGE_INDEX(TENSOR_TYPE, INDICES_TYPE, TYPE, \ 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
1368 #define LARGE_INDEX(TENSOR_TYPE, INDICES_TYPE, TYPE, \ in index_select_out_cuda_impl() argument
|