Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorShape.cpp1987 const auto n_threads_index = std::max<int64_t>( in index_select_sparse_cpu() local
1998 = (nnz / n_threads_nnz <= index_len / n_threads_index) in index_select_sparse_cpu()
2005 && (index_len * std::log2(nnz) / n_threads_index in index_select_sparse_cpu()
2263 const auto n_threads_index = std::max<int64_t>( in index_select_sparse_cpu() local
2275 …onst auto index_max_work_per_thread = index_max_copy_work_per_thread * index_len / n_threads_index; in index_select_sparse_cpu()