Searched refs:gpusparseMatDescr_t (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cuda_sparse.h | 33 using gpusparseMatDescr_t = cusparseMatDescr_t; variable
44 using gpusparseMatDescr_t = hipsparseMatDescr_t; variable
296 const gpusparseMatDescr_t descrA, const Scalar* csrSortedValA,
308 const Scalar* alpha_host, const gpusparseMatDescr_t descrA,
318 Status CsrgeamNnz(int m, int n, const gpusparseMatDescr_t descrA, int nnzA,
320 const gpusparseMatDescr_t descrB, int nnzB,
322 const gpusparseMatDescr_t descrC, int* csrSortedRowPtrC,
332 const gpusparseMatDescr_t descrA, int nnzA,
335 const gpusparseMatDescr_t descrB, int nnzB,
337 const int* csrSortedColIndB, const gpusparseMatDescr_t descrC,
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mat_mul_op.cc | 758 gpusparseMatDescr_t descrA; in Compute()
767 gpusparseMatDescr_t descrA; in Compute()
835 gpusparseMatDescr_t descrA; in Compute()
|