Searched defs:descrC (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | rocm_sparse.cc | 328 const int* csrSortedColIndB, const hipsparseMatDescr_t descrC, in CsrgemmNnz()
347 const int* csrSortedColIndB, const hipsparseMatDescr_t descrC, in CsrgemmImpl()
463 const hipsparseMatDescr_t descrC, Scalar* csrSortedValC, in CsrgeamBufferSizeExtImpl()
499 const int* csrSortedColIndB, const hipsparseMatDescr_t descrC, in CsrgeamNnz()
518 const hipsparseMatDescr_t descrC, Scalar* csrSortedValC, in Csrgeam2Impl()
|
| D | cuda_sparse.cc | 371 const int* csrSortedColIndB, const cusparseMatDescr_t descrC, in CsrgeamNnz()
626 const cusparseMatDescr_t descrC, Scalar* csrSortedValC, in CsrgeamImpl()
666 const cusparseMatDescr_t descrC, Scalar* csrSortedValC, in Csrgeam2Impl()
727 const cusparseMatDescr_t descrC, Scalar* csrSortedValC, in CsrgeamBufferSizeExtImpl()
768 const cusparseMatDescr_t descrC, int* csrSortedRowPtrC, in CsrgemmNnz()
787 const cusparseMatDescr_t descrC, Scalar* csrSortedValC, in CsrgemmImpl()
856 const int* csrSortedColIndB, const cusparseMatDescr_t descrC, in CsrgemmNnz()
877 const int* csrSortedColIndB, const cusparseMatDescr_t descrC, in CsrgemmImpl()
|