Searched defs:descrB (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cuda_sparse.cc | 425 const cusparseMatDescr_t descrB, int nnzB, in CsrgeamNnz()
524 const cusparseMatDescr_t descrB, int nnzB, const Scalar* csrSortedValB, in CsrgeamImpl()
561 const int* csrSortedColIndA, const cusparseMatDescr_t descrB, int nnzB, in CsrgemmNnz()
580 const cusparseMatDescr_t descrB, int nnzB, const Scalar* csrSortedValB, in CsrgemmImpl()
|
| D | rocm_sparse.cc | 250 const hipsparseMatDescr_t descrB, int nnzB, const int* csrSortedRowPtrB, in CsrgemmNnz()
268 const int* csrSortedColIndA, const hipsparseMatDescr_t descrB, int nnzB, in CsrgemmImpl()
|