Home
last modified time | relevance | path

Searched refs:gpusparse_handle_ (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dcuda_sparse.cc178 gpusparse_handle_ = &it->second.handle(); in Initialize()
225 return Gtsv2Impl(SPARSE_FN(gtsv2, sparse_prefix), *gpusparse_handle_, m, \
238 *gpusparse_handle_, m, n, dl, d, du, B, ldb, pBuffer); \
263 SPARSE_FN(gtsv2_bufferSizeExt, sparse_prefix), *gpusparse_handle_, m, \
277 *gpusparse_handle_, m, n, dl, d, du, B, ldb, bufferSizeInBytes); \
302 *gpusparse_handle_, m, dl, d, du, x, \
329 *gpusparse_handle_, m, dl, d, du, x, batchCount, batchStride, \
345 TF_RETURN_IF_GPUSPARSE_ERROR(cusparseXcoo2csr(*gpusparse_handle_, cooRowInd, in Coo2csr()
361 TF_RETURN_IF_GPUSPARSE_ERROR(cusparseXcsr2coo(*gpusparse_handle_, csrRowPtr, in Csr2coo()
377 *gpusparse_handle_, m, n, descrA, nnzA, csrSortedRowPtrA, in CsrgeamNnz()
[all …]
Drocm_sparse.cc150 gpusparse_handle_ = &it->second.handle(); in Initialize()
166 wrap::hipsparseXcoo2csr(*gpusparse_handle_, cooRowInd, nnz, m, csrRowPtr, in Coo2csr()
175 wrap::hipsparseXcsr2coo(*gpusparse_handle_, csrRowPtr, nnz, m, cooRowInd, in Csr2coo()
206 *gpusparse_handle_, transA, transB, m, n, k, nnz, \
240 *gpusparse_handle_, transA, m, n, nnz, alpha_host, \
257 *gpusparse_handle_, transA, transB, m, n, k, descrA, nnzA, in CsrgemmNnz()
293 *gpusparse_handle_, transA, transB, m, n, k, descrA, \
323 *gpusparse_handle_, m, n, nnz, csrVal, csrRowPtr, \
Dcuda_sparse.h454 gpusparseHandle_t* gpusparse_handle_; // not owned. variable