Home
last modified time | relevance | path

Searched defs:descrA (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Drocm_sparse.cc192 int k, int nnz, const Scalar* alpha_host, const hipsparseMatDescr_t descrA, in CsrmmImpl()
264 const hipsparseMatDescr_t descrA, in CsrmvImpl()
296 int k, const hipsparseMatDescr_t descrA, int nnzA, in CsrgemmNnz()
314 int k, const hipsparseMatDescr_t descrA, int nnzA, in CsrgemmImpl()
356 const hipsparseMatDescr_t descrA, in Csru2csrImpl()
429 int m, int n, const Scalar* alpha, const hipsparseMatDescr_t descrA, in CsrgeamBufferSizeExtImpl()
467 int m, int n, const hipsparseMatDescr_t descrA, int nnzA, in CsrgeamNnz()
484 int m, int n, const Scalar* alpha, const hipsparseMatDescr_t descrA, in Csrgeam2Impl()
Dcuda_sparse.cc368 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 …]
/external/tensorflow/tensorflow/core/kernels/sparse/
Dmat_mul_op.cc921 gpusparseMatDescr_t descrA; in Compute() local
932 gpusparseMatDescr_t descrA; in Compute() local
979 gpusparseMatDescr_t descrA; in Compute() local
986 gpusparseMatDescr_t descrA; in Compute() local