Searched defs:SOURCE_DIM (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Indexing.cu | 893 #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
|