Searched defs:total_nnz (Results 1 – 17 of 17) sorted by relevance
58 const int total_nnz = logits_matrix->total_nnz(); in Compute() local167 const int total_nnz = softmax_matrix->total_nnz(); in Compute() local
46 const int64_t total_nnz = indices.dimension(0); in operator ()() local
51 const int total_nnz = a.total_nnz(); in operator ()() local
84 const int64_t total_nnz = csr_sparse_matrix->total_nnz(); in Compute() local152 const int64_t total_nnz = csr_sparse_matrix->total_nnz(); in Compute() local
83 const int64_t total_nnz = values.NumElements(); in Compute() local214 int total_nnz = batch_ptr(batch_size); in ComputeAsync() local
97 const int64_t total_nnz = indices.NumElements() / rank; in Compute() local266 int total_nnz = batch_ptr(batch_size); in ComputeAsync() local
130 const int total_nnz = a.total_nnz(); in Compute() local
176 const int64_t total_nnz = batch_ptr_vec(batch_size); in Compute() local
58 const int total_nnz = indices.dimension(0); in operator ()() local262 const int total_nnz) { in CSRSparseMatrixBatchMulVecKernel3D()289 const int total_nnz = a.total_nnz(); in CSRSparseMatrixBatchMulVecImpl() local
146 const int64_t total_nnz = csr_sparse_matrix->total_nnz(); in Compute() local
212 const int64_t total_nnz = batch_ptr_vec(batch_size); in Compute() local475 const int total_nnz = c_batch_ptr(batch_size); in Compute() local
165 const int total_nnz = input_matrix.total_nnz(); in operator ()() local
152 const int total_nnz = c_batch_ptr(batch_size); in operator ()() local
204 inline int total_nnz() const { in total_nnz() function
239 const int total_nnz = B * M * nnz_inner; in MakeSparseTensor() local
107 int64_t total_nnz = col_indices.numel(); in convert_indices_from_csr_to_coo_cuda() local
1557 int64_t total_nnz = col_indices.numel(); in convert_indices_from_csr_to_coo_cpu() local