Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/sparse/
Dadd_op.cc284 const int nnzB = b.col_ind.size(); in GetOutputStructure() local
312 const int nnzB = b.col_ind.size(); in Compute() local
Dsparse_mat_mul_op.cc580 const int nnzB = b.col_ind.size(); in GetOutputStructure() local
617 const int nnzB = b.col_ind.size(); in Compute() local
/external/eigen/unsupported/Eigen/src/KroneckerProduct/
DKroneckerTensorProduct.h173 VectorXi nnzB = VectorXi::Zero(Dest::IsRowMajor ? m_B.rows() : m_B.cols()); in evalTo() local
/external/tensorflow/tensorflow/core/kernels/
Dcuda_sparse.cc425 const cusparseMatDescr_t descrB, int nnzB, in CsrgeamNnz()
524 const cusparseMatDescr_t descrB, int nnzB, const Scalar* csrSortedValB, in CsrgeamImpl()
561 const int* csrSortedColIndA, const cusparseMatDescr_t descrB, int nnzB, in CsrgemmNnz()
580 const cusparseMatDescr_t descrB, int nnzB, const Scalar* csrSortedValB, in CsrgemmImpl()
Drocm_sparse.cc250 const hipsparseMatDescr_t descrB, int nnzB, const int* csrSortedRowPtrB, in CsrgemmNnz()
268 const int* csrSortedColIndA, const hipsparseMatDescr_t descrB, int nnzB, in CsrgemmImpl()