Home
last modified time | relevance | path

Searched refs:src_intersection_counts (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorShape.cpp2306 const auto src_intersection_counts = src_counts.mul(idx_counts > 0); in index_select_sparse_cpu() local
2307 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()