Searched defs:csrSortedColIndA (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | rocm_sparse.cc | 185 const int* csrSortedColIndA, const Scalar* B, int ldb, in CsrmmImpl()
220 const int* csrSortedColIndA, const Scalar* x, in CsrmvImpl()
249 const int* csrSortedRowPtrA, const int* csrSortedColIndA, in CsrgemmNnz()
268 const int* csrSortedColIndA, const hipsparseMatDescr_t descrB, int nnzB, in CsrgemmImpl()
|
| D | cuda_sparse.cc | 424 const int* csrSortedColIndA, in CsrgeamNnz()
445 const int* csrSortedColIndA, const Scalar* B, int ldb, in CsrmmImpl()
484 const int* csrSortedRowPtrA, const int* csrSortedColIndA, const Scalar* x, in CsrmvImpl()
523 const int* csrSortedColIndA, const Scalar* beta, in CsrgeamImpl()
561 const int* csrSortedColIndA, const cusparseMatDescr_t descrB, int nnzB, in CsrgemmNnz()
579 const int* csrSortedRowPtrA, const int* csrSortedColIndA, in CsrgemmImpl()
|