Home
last modified time | relevance | path

Searched refs:nnzB (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dcuda_sparse.cc370 const cusparseMatDescr_t descrB, int nnzB, const int* csrSortedRowPtrB, in CsrgeamNnz() argument
378 csrSortedColIndA, descrB, nnzB, csrSortedRowPtrB, csrSortedColIndB, in CsrgeamNnz()
383 csrSortedColIndA, descrB, nnzB, csrSortedRowPtrB, csrSortedColIndB, in CsrgeamNnz()
624 const cusparseMatDescr_t descrB, int nnzB, const Scalar* csrSortedValB, in CsrgeamImpl() argument
631 AsCudaComplex(beta), descrB, nnzB, AsCudaComplex(csrSortedValB), in CsrgeamImpl()
643 const cusparseMatDescr_t descrB, int nnzB, const Scalar* csrSortedValB, \
651 beta, descrB, nnzB, csrSortedValB, csrSortedRowPtrB, \
664 const cusparseMatDescr_t descrB, int nnzB, const Scalar* csrSortedValB, in Csrgeam2Impl() argument
671 AsCudaComplex(beta), descrB, nnzB, AsCudaComplex(csrSortedValB), in Csrgeam2Impl()
683 const cusparseMatDescr_t descrB, int nnzB, const Scalar* csrSortedValB, \
[all …]
Dcuda_sparse.h337 const gpusparseMatDescr_t descrB, int nnzB, const Scalar* csrSortedValB,
349 const gpusparseMatDescr_t descrB, int nnzB,
364 const gpusparseMatDescr_t descrB, int nnzB,
378 const gpusparseMatDescr_t descrB, int nnzB, const int* csrSortedRowPtrB,
392 const gpusparseMatDescr_t descrB, int nnzB,
400 const gpusparseMatDescr_t descrB, int nnzB,
418 const gpusparseMatDescr_t descrB, int nnzB,
428 const gpusparseMatDescr_t descrB, int nnzB,
Drocm_sparse.cc251 const hipsparseMatDescr_t descrB, int nnzB, const int* csrSortedRowPtrB, in CsrgemmNnz() argument
258 csrSortedRowPtrA, csrSortedColIndA, descrB, nnzB, csrSortedRowPtrB, in CsrgemmNnz()
269 const int* csrSortedColIndA, const hipsparseMatDescr_t descrB, int nnzB, in CsrgemmImpl() argument
275 csrSortedRowPtrA, csrSortedColIndA, descrB, nnzB, csrSortedValB, in CsrgemmImpl()
287 const int* csrSortedColIndA, const hipsparseMatDescr_t descrB, int nnzB, \
295 csrSortedColIndA, descrB, nnzB, csrSortedValB, \
/external/tensorflow/tensorflow/core/kernels/sparse/
Dadd_op.cc302 const int nnzB = b.col_ind.size(); in GetWorkspaceSize() local
311 a.col_ind.data(), &beta_, descrB_.descr(), nnzB, b.values.data(), in GetWorkspaceSize()
330 const int nnzB = b.col_ind.size(); in GetOutputStructure() local
338 descrB_.descr(), nnzB, b.row_ptr.data(), b.col_ind.data(), in GetOutputStructure()
358 const int nnzB = b.col_ind.size(); in Compute() local
366 a.col_ind.data(), &beta_, descrB_.descr(), nnzB, b.values.data(), in Compute()
Dsparse_mat_mul_op.cc616 const int nnzB = b.col_ind.size(); in CSRSparseSparseMatrixMatMul() local
623 descrB_.descr(), nnzB, b.row_ptr.data(), b.col_ind.data(), info_, in CSRSparseSparseMatrixMatMul()
648 const int nnzB = b.col_ind.size(); in CSRSparseSparseMatrixMatMul() local
658 a.col_ind.data(), descrB_.descr(), nnzB, b.row_ptr.data(), in CSRSparseSparseMatrixMatMul()
663 descrB_.descr(), nnzB, b.row_ptr.data(), b.col_ind.data(), in CSRSparseSparseMatrixMatMul()
692 const int nnzB = b.col_ind.size(); in CSRSparseSparseMatrixMatMul() local
701 a.row_ptr.data(), a.col_ind.data(), descrB_.descr(), nnzB, in CSRSparseSparseMatrixMatMul()
707 a.col_ind.data(), descrB_.descr(), nnzB, b.values.data(), in CSRSparseSparseMatrixMatMul()
/external/eigen/unsupported/Eigen/src/KroneckerProduct/
DKroneckerTensorProduct.h173 VectorXi nnzB = VectorXi::Zero(Dest::IsRowMajor ? m_B.rows() : m_B.cols()); in evalTo() local
176 nnzB(Dest::IsRowMajor ? itB.row() : itB.col())++; in evalTo()
178 Matrix<int,Dynamic,Dynamic,ColMajor> nnzAB = nnzB * nnzA.transpose(); in evalTo()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcusparse_10_0.inc4718 const cusparseMatDescr_t descrB, const int nnzB,
4730 csrSortedRowPtrA, csrSortedColIndA, descrB, nnzB,
4740 const cusparseMatDescr_t descrB, const int nnzB, const float *csrSortedValB,
4753 csrSortedRowPtrA, csrSortedColIndA, descrB, nnzB,
4763 const cusparseMatDescr_t descrB, int nnzB, const double *csrSortedValB,
4775 csrSortedRowPtrA, csrSortedColIndA, descrB, nnzB,
4785 const cusparseMatDescr_t descrB, int nnzB, const cuComplex *csrSortedValB,
4798 csrSortedRowPtrA, csrSortedColIndA, descrB, nnzB,
4808 const int *csrSortedColIndA, const cusparseMatDescr_t descrB, int nnzB,
4822 csrSortedRowPtrA, csrSortedColIndA, descrB, nnzB,
[all …]
Dcusparse_10_1.inc4689 const cusparseMatDescr_t descrB, const int nnzB,
4701 csrSortedRowPtrA, csrSortedColIndA, descrB, nnzB,
4711 const cusparseMatDescr_t descrB, const int nnzB, const float *csrSortedValB,
4724 csrSortedRowPtrA, csrSortedColIndA, descrB, nnzB,
4734 const cusparseMatDescr_t descrB, int nnzB, const double *csrSortedValB,
4746 csrSortedRowPtrA, csrSortedColIndA, descrB, nnzB,
4756 const cusparseMatDescr_t descrB, int nnzB, const cuComplex *csrSortedValB,
4769 csrSortedRowPtrA, csrSortedColIndA, descrB, nnzB,
4779 const int *csrSortedColIndA, const cusparseMatDescr_t descrB, int nnzB,
4793 csrSortedRowPtrA, csrSortedColIndA, descrB, nnzB,
[all …]
Dcusparse_10_2.inc4689 const cusparseMatDescr_t descrB, const int nnzB,
4701 csrSortedRowPtrA, csrSortedColIndA, descrB, nnzB,
4711 const cusparseMatDescr_t descrB, const int nnzB, const float *csrSortedValB,
4724 csrSortedRowPtrA, csrSortedColIndA, descrB, nnzB,
4734 const cusparseMatDescr_t descrB, int nnzB, const double *csrSortedValB,
4746 csrSortedRowPtrA, csrSortedColIndA, descrB, nnzB,
4756 const cusparseMatDescr_t descrB, int nnzB, const cuComplex *csrSortedValB,
4769 csrSortedRowPtrA, csrSortedColIndA, descrB, nnzB,
4779 const int *csrSortedColIndA, const cusparseMatDescr_t descrB, int nnzB,
4793 csrSortedRowPtrA, csrSortedColIndA, descrB, nnzB,
[all …]
Dcusparse_11_0.inc3604 const int *csrSortedColIndA, const cusparseMatDescr_t descrB, int nnzB,
3617 csrSortedColIndA, descrB, nnzB, csrSortedRowPtrB,
3626 const int *csrSortedColIndA, const cusparseMatDescr_t descrB, int nnzB,
3639 csrSortedColIndA, descrB, nnzB, csrSortedRowPtrB,
3648 const int *csrSortedColIndA, const cusparseMatDescr_t descrB, int nnzB,
3662 csrSortedColIndA, descrB, nnzB, csrSortedRowPtrB,
3671 const int *csrSortedColIndA, const cusparseMatDescr_t descrB, int nnzB,
3685 csrSortedColIndA, descrB, nnzB, csrSortedRowPtrB,
3694 const int *csrSortedColIndA, const cusparseMatDescr_t descrB, int nnzB,
3708 csrSortedColIndA, descrB, nnzB, csrSortedRowPtrB,
[all …]
Dcusparse_9_0.inc4255 const cusparseMatDescr_t descrB, const int nnzB,
4267 csrSortedRowPtrA, csrSortedColIndA, descrB, nnzB,
4277 const cusparseMatDescr_t descrB, const int nnzB, const float *csrSortedValB,
4290 csrSortedRowPtrA, csrSortedColIndA, descrB, nnzB,
4300 const cusparseMatDescr_t descrB, int nnzB, const double *csrSortedValB,
4312 csrSortedRowPtrA, csrSortedColIndA, descrB, nnzB,
4322 const cusparseMatDescr_t descrB, int nnzB, const cuComplex *csrSortedValB,
4335 csrSortedRowPtrA, csrSortedColIndA, descrB, nnzB,
4345 const int *csrSortedColIndA, const cusparseMatDescr_t descrB, int nnzB,
4359 csrSortedRowPtrA, csrSortedColIndA, descrB, nnzB,
[all …]