Searched defs:nnzB (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | add_op.cc | 302 const int nnzB = b.col_ind.size(); in GetWorkspaceSize() local
330 const int nnzB = b.col_ind.size(); in GetOutputStructure() local
358 const int nnzB = b.col_ind.size(); in Compute() local
|
| D | sparse_mat_mul_op.cc | 616 const int nnzB = b.col_ind.size(); in CSRSparseSparseMatrixMatMul() local
648 const int nnzB = b.col_ind.size(); in CSRSparseSparseMatrixMatMul() local
692 const int nnzB = b.col_ind.size(); in CSRSparseSparseMatrixMatMul() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_sparse.cc | 370 const cusparseMatDescr_t descrB, int nnzB, const int* csrSortedRowPtrB, in CsrgeamNnz()
624 const cusparseMatDescr_t descrB, int nnzB, const Scalar* csrSortedValB, in CsrgeamImpl()
664 const cusparseMatDescr_t descrB, int nnzB, const Scalar* csrSortedValB, in Csrgeam2Impl()
725 const cusparseMatDescr_t descrB, int nnzB, const Scalar* csrSortedValB, in CsrgeamBufferSizeExtImpl()
766 const int* csrSortedColIndA, const cusparseMatDescr_t descrB, int nnzB, in CsrgemmNnz()
785 const cusparseMatDescr_t descrB, int nnzB, const Scalar* csrSortedValB, in CsrgemmImpl()
855 const cusparseMatDescr_t descrB, int nnzB, const int* csrSortedRowPtrB, in CsrgemmNnz()
875 const int* csrSortedColIndA, const cusparseMatDescr_t descrB, int nnzB, in CsrgemmImpl()
|
| D | rocm_sparse.cc | 251 const hipsparseMatDescr_t descrB, int nnzB, const int* csrSortedRowPtrB, in CsrgemmNnz()
269 const int* csrSortedColIndA, const hipsparseMatDescr_t descrB, int nnzB, in CsrgemmImpl()
|
| /external/eigen/unsupported/Eigen/src/KroneckerProduct/ |
| D | KroneckerTensorProduct.h | 173 VectorXi nnzB = VectorXi::Zero(Dest::IsRowMajor ? m_B.rows() : m_B.cols()); in evalTo() local
|