Searched defs:descrA (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_sparse.cc | 368 int m, int n, const cusparseMatDescr_t descrA, int nnzA, in CsrgeamNnz()
395 int nnz, const Scalar* alpha_host, const cusparseMatDescr_t descrA, in CsrmmImpl()
474 const cusparseMatDescr_t descrA, const Scalar* csrSortedValA, in CsrmvImpl()
520 cusparseMatDescr_t descrA; in CsrmvExImpl() local
621 int m, int n, const Scalar* alpha, const cusparseMatDescr_t descrA, in CsrgeamImpl()
661 int m, int n, const Scalar* alpha, const cusparseMatDescr_t descrA, in Csrgeam2Impl()
722 int m, int n, const Scalar* alpha, const cusparseMatDescr_t descrA, 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()
[all …]
|
| D | rocm_sparse.cc | 184 int k, int nnz, const Scalar* alpha_host, const hipsparseMatDescr_t descrA, in CsrmmImpl()
218 const hipsparseMatDescr_t descrA, in CsrmvImpl()
249 int k, const hipsparseMatDescr_t descrA, int nnzA, in CsrgemmNnz()
267 int k, const hipsparseMatDescr_t descrA, int nnzA, in CsrgemmImpl()
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mat_mul_op.cc | 877 gpusparseMatDescr_t descrA; in Compute() local
888 gpusparseMatDescr_t descrA; in Compute() local
935 gpusparseMatDescr_t descrA; in Compute() local
942 gpusparseMatDescr_t descrA; in Compute() local
|