Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/sparse/
Dsoftmax_op.cc58 const int total_nnz = logits_matrix->total_nnz(); in Compute() local
167 const int total_nnz = softmax_matrix->total_nnz(); in Compute() local
Dkernels.cc45 const int64 total_nnz = indices.dimension(0); in operator ()() local
Dconj_op.cc51 const int total_nnz = a.total_nnz(); in operator ()() local
Dcsr_sparse_matrix_to_sparse_tensor_op.cc84 const int64 total_nnz = csr_sparse_matrix->total_nnz(); in Compute() local
151 const int64 total_nnz = csr_sparse_matrix->total_nnz(); in Compute() local
Dsparse_tensor_to_csr_sparse_matrix_op.cc76 const int64 total_nnz = values.NumElements(); in Compute() local
198 int total_nnz = batch_ptr(batch_size); in ComputeAsync() local
Ddense_to_csr_sparse_matrix_op.cc97 const int64 total_nnz = indices.NumElements() / rank; in Compute() local
265 int total_nnz = batch_ptr(batch_size); in ComputeAsync() local
Dmul_op.cc130 const int total_nnz = a.total_nnz(); in Compute() local
Dsparse_cholesky_op.cc174 const int64 total_nnz = batch_ptr_vec(batch_size); in Compute() local
Dkernels_gpu.cu.cc58 const int total_nnz = indices.dimension(0); in operator ()() local
262 const int total_nnz) { in CSRSparseMatrixBatchMulVecKernel3D()
289 const int total_nnz = a.total_nnz(); in CSRSparseMatrixBatchMulVecImpl() local
Dcsr_sparse_matrix_to_dense_op.cc145 const int64 total_nnz = csr_sparse_matrix->total_nnz(); in Compute() local
Dtranspose_op.cc168 const int total_nnz = input_matrix.total_nnz(); in operator ()() local
Dsparse_mat_mul_op.cc212 const int64 total_nnz = batch_ptr_vec(batch_size); in Compute() local
475 const int total_nnz = c_batch_ptr(batch_size); in Compute() local
Dadd_op.cc152 const int total_nnz = c_batch_ptr(batch_size); in operator ()() local
Dsparse_matrix.h204 inline int total_nnz() const { in total_nnz() function
/external/tensorflow/tensorflow/core/kernels/
Dsparse_dense_binary_op_shared_test.cc239 const int total_nnz = B * M * nnz_inner; in MakeSparseTensor() local