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
75 const int64 total_nnz = values.NumElements(); in Compute() local197 int total_nnz = batch_ptr(batch_size); in ComputeAsync() local
96 const int64 total_nnz = indices.NumElements() / rank; in Compute() local264 int total_nnz = batch_ptr(batch_size); in ComputeAsync() local
132 const int total_nnz = a.total_nnz(); in Compute() local
174 const int64 total_nnz = batch_ptr_vec(batch_size); in Compute() local
71 const int total_nnz = indices.dimension(0); in operator ()() local275 const int total_nnz) { in CSRSparseMatrixBatchMulVecKernel3D()302 const int total_nnz = a.total_nnz(); in CSRSparseMatrixBatchMulVecImpl() local
212 const int64 total_nnz = batch_ptr_vec(batch_size); in Compute() local444 const int total_nnz = c_batch_ptr(batch_size); in Compute() 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
132 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