Searched defs:total_nnz (Results 1 – 15 of 15) 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
45 const int64 total_nnz = indices.dimension(0); in operator ()() local
51 const int total_nnz = a.total_nnz(); in operator ()() local
84 const int64 total_nnz = csr_sparse_matrix->total_nnz(); in Compute() local151 const int64 total_nnz = csr_sparse_matrix->total_nnz(); in Compute() local
76 const int64 total_nnz = values.NumElements(); in Compute() local198 int total_nnz = batch_ptr(batch_size); in ComputeAsync() local
97 const int64 total_nnz = indices.NumElements() / rank; in Compute() local265 int total_nnz = batch_ptr(batch_size); in ComputeAsync() local
130 const int total_nnz = a.total_nnz(); in Compute() local
174 const int64 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
145 const int64 total_nnz = csr_sparse_matrix->total_nnz(); in Compute() local
168 const int total_nnz = input_matrix.total_nnz(); in operator ()() local
212 const int64 total_nnz = batch_ptr_vec(batch_size); in Compute() local475 const int total_nnz = c_batch_ptr(batch_size); in Compute() 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