Home
last modified time | relevance | path

Searched defs:nnzA (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/sparse/
Dadd_op.cc283 const int nnzA = a.col_ind.size(); in GetOutputStructure() local
311 const int nnzA = a.col_ind.size(); in Compute() local
Dsparse_mat_mul_op.cc579 const int nnzA = a.col_ind.size(); in GetOutputStructure() local
616 const int nnzA = a.col_ind.size(); in Compute() local
/external/eigen/unsupported/Eigen/src/KroneckerProduct/
DKroneckerTensorProduct.h168 VectorXi nnzA = VectorXi::Zero(Dest::IsRowMajor ? m_A.rows() : m_A.cols()); in evalTo() local
/external/tensorflow/tensorflow/core/kernels/
Dcuda_sparse.cc423 int nnzA, const int* csrSortedRowPtrA, in CsrgeamNnz()
522 int nnzA, const Scalar* csrSortedValA, const int* csrSortedRowPtrA, in CsrgeamImpl()
560 const cusparseMatDescr_t descrA, int nnzA, const int* csrSortedRowPtrA, in CsrgemmNnz()
578 const cusparseMatDescr_t descrA, int nnzA, const Scalar* csrSortedValA, in CsrgemmImpl()
Drocm_sparse.cc248 int k, const hipsparseMatDescr_t descrA, int nnzA, in CsrgemmNnz()
266 int k, const hipsparseMatDescr_t descrA, int nnzA, in CsrgemmImpl()