Searched refs:src_n_threads_shape (Results 1 – 1 of 1) sorted by relevance
2041 const std::vector<int64_t> src_n_threads_shape = { in index_select_sparse_cpu() local2048 auto src_int_idx = at::empty(src_n_threads_shape, src.options()); in index_select_sparse_cpu()