Searched refs:gpusparseMatDescr_t (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_sparse.h | 34 using gpusparseMatDescr_t = cusparseMatDescr_t; variable
53 using gpusparseMatDescr_t = hipsparseMatDescr_t; variable
279 const gpusparseMatDescr_t descrA, const Scalar* csrSortedValA,
318 const Scalar* alpha_host, const gpusparseMatDescr_t descrA,
334 int m, int n, const Scalar* alpha, const gpusparseMatDescr_t descrA,
337 const gpusparseMatDescr_t descrB, int nnzB, const Scalar* csrSortedValB,
339 const gpusparseMatDescr_t descrC, Scalar* csrSortedValC,
347 Status CsrgeamNnz(int m, int n, const gpusparseMatDescr_t descrA, int nnzA,
349 const gpusparseMatDescr_t descrB, int nnzB,
351 const gpusparseMatDescr_t descrC, int* csrSortedRowPtrC,
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mat_mul_op.cc | 877 gpusparseMatDescr_t descrA; in Compute()
888 gpusparseMatDescr_t descrA; in Compute()
935 gpusparseMatDescr_t descrA; in Compute()
942 gpusparseMatDescr_t descrA; in Compute()
|