Searched refs:src_intersection_offsets (Results 1 – 1 of 1) sorted by relevance
2307 const auto src_intersection_offsets = src_intersection_counts.cumsum(0); in index_select_sparse_cpu() local2308 const auto src_idx_len = src_intersection_offsets.const_data_ptr<int64_t>()[size - 1]; in index_select_sparse_cpu()2314 … const auto* ptr_src_intersection_offsets = src_intersection_offsets.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()