Searched refs:src_offset_counts_per_thread (Results 1 – 1 of 1) sorted by relevance
2283 Tensor src, src_counts_per_thread, src_offset_counts_per_thread; in index_select_sparse_cpu() local2286 src, src_counts_per_thread, src_offset_counts_per_thread in index_select_sparse_cpu()2300 const auto src_counts = src_offset_counts_per_thread.select(0, -1); in index_select_sparse_cpu()2329 = src_offset_counts_per_thread.select(0, tid).const_data_ptr<int64_t>(); in index_select_sparse_cpu()