Home
last modified time | relevance | path

Searched refs:descrA (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dcuda_sparse.cc368 int m, int n, const cusparseMatDescr_t descrA, int nnzA, in CsrgeamNnz() argument
377 *gpusparse_handle_, m, n, descrA, nnzA, csrSortedRowPtrA, in CsrgeamNnz()
382 *gpusparse_handle_, m, n, descrA, nnzA, csrSortedRowPtrA, in CsrgeamNnz()
395 int nnz, const Scalar* alpha_host, const cusparseMatDescr_t descrA, in CsrmmImpl() argument
408 descrA, AsCudaComplex(csrSortedValA), csrSortedRowPtrA, csrSortedColIndA, in CsrmmImpl()
418 const cusparseMatDescr_t descrA, const Scalar* csrSortedValA, \
425 alpha_host, descrA, csrSortedValA, csrSortedRowPtrA, \
474 const cusparseMatDescr_t descrA, const Scalar* csrSortedValA, in CsrmvImpl() argument
478 op(cusparse_handle, transA, m, n, nnz, AsCudaComplex(alpha_host), descrA, in CsrmvImpl()
489 const Scalar* alpha_host, const cusparseMatDescr_t descrA, \
[all …]
Drocm_sparse.cc184 int k, int nnz, const Scalar* alpha_host, const hipsparseMatDescr_t descrA, in CsrmmImpl() argument
189 nnz, alpha_host, descrA, csrSortedValA, in CsrmmImpl()
200 const hipsparseMatDescr_t descrA, const Scalar* csrSortedValA, \
207 alpha_host, descrA, csrSortedValA, csrSortedRowPtrA, \
218 const hipsparseMatDescr_t descrA, in CsrmvImpl() argument
224 op(hipsparse_handle, transA, m, n, nnz, alpha_host, descrA, csrSortedValA, in CsrmvImpl()
234 const Scalar* alpha_host, const hipsparseMatDescr_t descrA, \
241 descrA, csrSortedValA, csrSortedRowPtrA, \
249 int k, const hipsparseMatDescr_t descrA, int nnzA, in CsrgemmNnz() argument
257 *gpusparse_handle_, transA, transB, m, n, k, descrA, nnzA, in CsrgemmNnz()
[all …]
Dcuda_sparse.h279 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,
347 Status CsrgeamNnz(int m, int n, const gpusparseMatDescr_t descrA, int nnzA,
361 const gpusparseMatDescr_t descrA, int nnzA,
376 int m, int n, int k, const gpusparseMatDescr_t descrA, int nnzA,
389 int m, int k, int n, const gpusparseMatDescr_t descrA,
397 Status CsrgemmNnz(int m, int n, int k, const gpusparseMatDescr_t descrA,
415 int m, int k, int n, const gpusparseMatDescr_t descrA,
425 Status Csrgemm(int m, int n, int k, const gpusparseMatDescr_t descrA,
[all …]
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcusparse_10_0.inc72 cusparseCreateMatDescr(cusparseMatDescr_t *descrA) {
76 return func_ptr(descrA);
80 cusparseDestroyMatDescr(cusparseMatDescr_t descrA) {
84 return func_ptr(descrA);
96 cusparseStatus_t CUSPARSEAPI cusparseSetMatType(cusparseMatDescr_t descrA,
102 return func_ptr(descrA, type);
106 cusparseSetMatFillMode(cusparseMatDescr_t descrA, cusparseFillMode_t fillMode) {
111 return func_ptr(descrA, fillMode);
115 cusparseSetMatDiagType(cusparseMatDescr_t descrA, cusparseDiagType_t diagType) {
120 return func_ptr(descrA, diagType);
[all …]
Dcusparse_9_0.inc72 cusparseCreateMatDescr(cusparseMatDescr_t *descrA) {
76 return func_ptr(descrA);
80 cusparseDestroyMatDescr(cusparseMatDescr_t descrA) {
84 return func_ptr(descrA);
96 cusparseStatus_t CUSPARSEAPI cusparseSetMatType(cusparseMatDescr_t descrA,
102 return func_ptr(descrA, type);
106 cusparseSetMatFillMode(cusparseMatDescr_t descrA, cusparseFillMode_t fillMode) {
111 return func_ptr(descrA, fillMode);
115 cusparseSetMatDiagType(cusparseMatDescr_t descrA, cusparseDiagType_t diagType) {
120 return func_ptr(descrA, diagType);
[all …]
Dcusparse_10_1.inc86 cusparseCreateMatDescr(cusparseMatDescr_t *descrA) {
90 return func_ptr(descrA);
94 cusparseDestroyMatDescr(cusparseMatDescr_t descrA) {
98 return func_ptr(descrA);
110 cusparseStatus_t CUSPARSEAPI cusparseSetMatType(cusparseMatDescr_t descrA,
116 return func_ptr(descrA, type);
120 cusparseSetMatFillMode(cusparseMatDescr_t descrA, cusparseFillMode_t fillMode) {
125 return func_ptr(descrA, fillMode);
129 cusparseSetMatDiagType(cusparseMatDescr_t descrA, cusparseDiagType_t diagType) {
134 return func_ptr(descrA, diagType);
[all …]
Dcusparse_10_2.inc86 cusparseCreateMatDescr(cusparseMatDescr_t *descrA) {
90 return func_ptr(descrA);
94 cusparseDestroyMatDescr(cusparseMatDescr_t descrA) {
98 return func_ptr(descrA);
110 cusparseStatus_t CUSPARSEAPI cusparseSetMatType(cusparseMatDescr_t descrA,
116 return func_ptr(descrA, type);
120 cusparseSetMatFillMode(cusparseMatDescr_t descrA, cusparseFillMode_t fillMode) {
125 return func_ptr(descrA, fillMode);
129 cusparseSetMatDiagType(cusparseMatDescr_t descrA, cusparseDiagType_t diagType) {
134 return func_ptr(descrA, diagType);
[all …]
Dcusparse_11_0.inc88 cusparseCreateMatDescr(cusparseMatDescr_t *descrA) {
92 return func_ptr(descrA);
96 cusparseDestroyMatDescr(cusparseMatDescr_t descrA) {
100 return func_ptr(descrA);
112 cusparseStatus_t CUSPARSEAPI cusparseSetMatType(cusparseMatDescr_t descrA,
118 return func_ptr(descrA, type);
122 cusparseSetMatFillMode(cusparseMatDescr_t descrA, cusparseFillMode_t fillMode) {
127 return func_ptr(descrA, fillMode);
131 cusparseSetMatDiagType(cusparseMatDescr_t descrA, cusparseDiagType_t diagType) {
136 return func_ptr(descrA, diagType);
[all …]
/external/tensorflow/tensorflow/core/kernels/sparse/
Dmat_mul_op.cc877 gpusparseMatDescr_t descrA; in Compute() local
878 TF_RETURN_IF_GPUSPARSE_ERROR(cusparseCreateMatDescr(&descrA)); in Compute()
880 cusparseSetMatType(descrA, CUSPARSE_MATRIX_TYPE_GENERAL)); in Compute()
882 cusparseSetMatIndexBase(descrA, CUSPARSE_INDEX_BASE_ZERO)); in Compute()
888 gpusparseMatDescr_t descrA; in Compute() local
889 TF_RETURN_IF_GPUSPARSE_ERROR(wrap::hipsparseCreateMatDescr(&descrA)); in Compute()
891 wrap::hipsparseSetMatType(descrA, HIPSPARSE_MATRIX_TYPE_GENERAL)); in Compute()
893 wrap::hipsparseSetMatIndexBase(descrA, HIPSPARSE_INDEX_BASE_ZERO)); in Compute()
897 cuda_sparse.Csrmm(transA, transB, m, n, k, nnz, &alpha, descrA, in Compute()
935 gpusparseMatDescr_t descrA; in Compute() local
[all …]