Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dcuda_sparse.cc177 gpusparse_handle_ = &it->second.handle(); in Initialize()
219 return GtsvImpl(SPARSE_FN(gtsv, sparse_prefix), *gpusparse_handle_, m, n, \
232 *gpusparse_handle_, m, n, dl, d, du, B, ldb); \
257 *gpusparse_handle_, m, dl, d, du, x, \
280 return Gtsv2Impl(SPARSE_FN(gtsv2, sparse_prefix), *gpusparse_handle_, m, \
293 *gpusparse_handle_, m, n, dl, d, du, B, ldb, pBuffer); \
318 SPARSE_FN(gtsv2_bufferSizeExt, sparse_prefix), *gpusparse_handle_, m, \
332 *gpusparse_handle_, m, n, dl, d, du, B, ldb, bufferSizeInBytes); \
357 *gpusparse_handle_, m, dl, d, du, x, \
384 *gpusparse_handle_, m, dl, d, du, x, batchCount, batchStride, \
[all …]
Drocm_sparse.cc149 gpusparse_handle_ = &it->second.handle(); in Initialize()
164 TF_RETURN_IF_GPUSPARSE_ERROR(hipsparseXcoo2csr(*gpusparse_handle_, cooRowInd, in Coo2csr()
173 TF_RETURN_IF_GPUSPARSE_ERROR(hipsparseXcsr2coo(*gpusparse_handle_, csrRowPtr, in Csr2coo()
205 *gpusparse_handle_, transA, transB, m, n, k, nnz, \
239 *gpusparse_handle_, transA, m, n, nnz, alpha_host, \
256 *gpusparse_handle_, transA, transB, m, n, k, descrA, nnzA, in CsrgemmNnz()
292 *gpusparse_handle_, transA, transB, m, n, k, descrA, \
322 *gpusparse_handle_, m, n, nnz, csrVal, csrRowPtr, \
Dcuda_sparse.h389 gpusparseHandle_t* gpusparse_handle_; // not owned. variable