Searched defs:src_idx_offsets (Results 1 – 1 of 1) sorted by relevance
2302 Tensor src_idx, src_idx_offsets; in index_select_sparse_cpu() local2349 const auto src_idx_offsets = src_intersection_offsets.sub_(src_intersection_counts); in index_select_sparse_cpu() local