Searched refs:src_intersection_counts (Results 1 – 1 of 1) sorted by relevance
2306 const auto src_intersection_counts = src_counts.mul(idx_counts > 0); in index_select_sparse_cpu() local2307 const auto src_intersection_offsets = src_intersection_counts.cumsum(0); in index_select_sparse_cpu()2313 const auto* ptr_src_intersection_counts = src_intersection_counts.const_data_ptr<int64_t>(); in index_select_sparse_cpu()2349 const auto src_idx_offsets = src_intersection_offsets.sub_(src_intersection_counts); in index_select_sparse_cpu()