Searched defs:nnzA (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | add_op.cc | 301 const int nnzA = a.col_ind.size(); in GetWorkspaceSize() local
329 const int nnzA = a.col_ind.size(); in GetOutputStructure() local
357 const int nnzA = a.col_ind.size(); in Compute() local
|
| D | sparse_mat_mul_op.cc | 615 const int nnzA = a.col_ind.size(); in CSRSparseSparseMatrixMatMul() local
647 const int nnzA = a.col_ind.size(); in CSRSparseSparseMatrixMatMul() local
691 const int nnzA = a.col_ind.size(); in CSRSparseSparseMatrixMatMul() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_sparse.cc | 368 int m, int n, const cusparseMatDescr_t descrA, int nnzA, in CsrgeamNnz()
622 int nnzA, const Scalar* csrSortedValA, const int* csrSortedRowPtrA, in CsrgeamImpl()
662 int nnzA, const Scalar* csrSortedValA, const int* csrSortedRowPtrA, in Csrgeam2Impl()
723 int nnzA, const Scalar* csrSortedValA, const int* csrSortedRowPtrA, in CsrgeamBufferSizeExtImpl()
765 const cusparseMatDescr_t descrA, int nnzA, const int* csrSortedRowPtrA, in CsrgemmNnz()
783 const cusparseMatDescr_t descrA, int nnzA, const Scalar* csrSortedValA, in CsrgemmImpl()
853 int m, int n, int k, const cusparseMatDescr_t descrA, int nnzA, in CsrgemmNnz()
873 int m, int n, int k, const cusparseMatDescr_t descrA, int nnzA, in CsrgemmImpl()
|
| D | rocm_sparse.cc | 249 int k, const hipsparseMatDescr_t descrA, int nnzA, in CsrgemmNnz()
267 int k, const hipsparseMatDescr_t descrA, int nnzA, in CsrgemmImpl()
|
| /external/eigen/unsupported/Eigen/src/KroneckerProduct/ |
| D | KroneckerTensorProduct.h | 168 VectorXi nnzA = VectorXi::Zero(Dest::IsRowMajor ? m_A.rows() : m_A.cols()); in evalTo() local
|