Searched defs:counts_per_thread (Results 1 – 1 of 1) sorted by relevance
2210 const auto counts_per_thread = [&get_counts, size]( in index_select_sparse_cpu() local2223 auto counts_per_thread = at::zeros({n_threads, size}, idx.options()); in index_select_sparse_cpu() local2360 auto counts_per_thread = idx_counts_per_thread.mul_(src_counts).sum(-1); in index_select_sparse_cpu() local