Searched defs:new_col_indices (Results 1 – 2 of 2) sorted by relevance
412 const index_t* new_col_indices, in reduce_sparse_csr_dim0_cuda_kernel()483 auto new_col_indices = std::get<0>(at::_unique(col_indices, true, false)); in reduce_sparse_csr_dim0_cuda_template() local575 Tensor new_col_indices = at::empty({}, ioptions); in reduce_sparse_csr_dim1_cuda_template() local645 Tensor new_col_indices = at::zeros({nnz}, ioptions); in reduce_sparse_csr_dim01_cuda_template() local
1158 Tensor new_col_indices = at::empty({}, ioptions); in reduce_sparse_csr_dim1_cpu_template() local1260 Tensor new_col_indices = at::zeros({nnz}, ioptions); in reduce_sparse_csr_dim01_cpu_template() local