Searched defs:SRC_DIM (Results 1 – 1 of 1) sorted by relevance
1360 #define SMALL_INDEX(TENSOR_TYPE, INDICES_TYPE, TYPE, DST_DIM, SRC_DIM, IDX_DIM) \ in index_select_out_cuda_impl() argument1369 DST_DIM, SRC_DIM, IDX_DIM, IDX_IS_MAJOR) \ in index_select_out_cuda_impl() argument