Searched defs:tid_counts (Results 1 – 1 of 1) sorted by relevance
2228 auto tid_counts = counts_per_thread.select(0, tid); in index_select_sparse_cpu() local2330 auto tid_counts = at::zeros({size}, src.options()); in index_select_sparse_cpu() local