Home
last modified time | relevance | path

Searched defs:new_nnz (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSparseCsrTensorMath.cu413 const int64_t new_nnz, in reduce_sparse_csr_dim0_cuda_kernel()
484 auto new_nnz = new_col_indices.numel(); in reduce_sparse_csr_dim0_cuda_template() local
601 index_t new_nnz = new_crow_indices[-1].item<index_t>(); in reduce_sparse_csr_dim1_cuda_template() local