| /external/tensorflow/tensorflow/core/kernels/ |
| D | rocm_sparse.cc | 182 hipsparseOperation_t transA, hipsparseOperation_t transB, int m, int n, in CsrmmImpl() argument
187 TF_RETURN_IF_GPUSPARSE_ERROR(op(hipsparse_handle, transA, transB, m, n, k, in CsrmmImpl()
197 hipsparseOperation_t transA, hipsparseOperation_t transB, int m, int n, \
205 *gpusparse_handle_, transA, transB, m, n, k, nnz, \
215 hipsparseOperation_t transA, int m, int n, in CsrmvImpl() argument
223 op(hipsparse_handle, transA, m, n, nnz, alpha_host, descrA, csrSortedValA, in CsrmvImpl()
232 hipsparseOperation_t transA, int m, int n, int nnz, \
239 *gpusparse_handle_, transA, m, n, nnz, alpha_host, \
247 hipsparseOperation_t transA, hipsparseOperation_t transB, int m, int n, in CsrgemmNnz() argument
256 *gpusparse_handle_, transA, transB, m, n, k, descrA, nnzA, in CsrgemmNnz()
[all …]
|
| D | cuda_sparse.cc | 442 cusparseOperation_t transA, cusparseOperation_t transB, int m, int n, int k, in CsrmmImpl() argument
455 cusparse_handle, transA, transB, m, n, k, nnz, AsCudaComplex(alpha_host), in CsrmmImpl()
464 cusparseOperation_t transA, cusparseOperation_t transB, int m, int n, \
472 *gpusparse_handle_, transA, transB, m, n, k, nnz, \
482 cusparseOperation_t transA, int m, int n, int nnz, const Scalar* alpha_host, in CsrmvImpl() argument
487 op(cusparse_handle, transA, m, n, nnz, AsCudaComplex(alpha_host), descrA, in CsrmvImpl()
497 cusparseOperation_t transA, int m, int n, int nnz, \
503 if (transA == CUSPARSE_OPERATION_NON_TRANSPOSE) { \
505 *gpusparse_handle_, transA, m, n, nnz, alpha_host, \
510 *gpusparse_handle_, transA, m, n, nnz, alpha_host, \
[all …]
|
| D | cuda_sparse.h | 294 Status Csrmm(gpusparseOperation_t transA, gpusparseOperation_t transB, int m,
307 Status Csrmv(gpusparseOperation_t transA, int m, int n, int nnz,
346 Status CsrgemmNnz(gpusparseOperation_t transA, gpusparseOperation_t transB,
361 Status Csrgemm(gpusparseOperation_t transA, gpusparseOperation_t transB,
|
| D | sparse_matmul_op.cc | 1389 empty_type_wrapper<float>, const libxsmm_spmdm_handle* handle, char transA, in wrapper_libxsmm_spmdm_createSparseSlice_generic_thread() argument
1393 handle, transA, A, libxsmm_output_csr_a, block_id, tid, nthreads); in wrapper_libxsmm_spmdm_createSparseSlice_generic_thread()
1397 char transA, const bfloat16* A, in wrapper_libxsmm_spmdm_createSparseSlice_generic_thread() argument
1401 handle, transA, reinterpret_cast<const libxsmm_bfloat16*>(A), in wrapper_libxsmm_spmdm_createSparseSlice_generic_thread()
1407 char transA, char transB, const bfloat16* alpha, in wrapper_libxsmm_spmdm_compute_generic_thread() argument
1411 handle, transA, transB, reinterpret_cast<const libxsmm_bfloat16*>(alpha), in wrapper_libxsmm_spmdm_compute_generic_thread()
1417 empty_type_wrapper<float>, const libxsmm_spmdm_handle* handle, char transA, in wrapper_libxsmm_spmdm_compute_generic_thread() argument
1421 return libxsmm_spmdm_compute_fp32_thread(handle, transA, transB, alpha, in wrapper_libxsmm_spmdm_compute_generic_thread()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cusparse_10_0.inc | 639 cusparseSgemvi(cusparseHandle_t handle, cusparseOperation_t transA, int m,
650 return func_ptr(handle, transA, m, n, alpha, A, lda, nnz, xVal, xInd, beta, y,
655 cusparseSgemvi_bufferSize(cusparseHandle_t handle, cusparseOperation_t transA,
661 return func_ptr(handle, transA, m, n, nnz, pBufferSize);
665 cusparseDgemvi(cusparseHandle_t handle, cusparseOperation_t transA, int m,
676 return func_ptr(handle, transA, m, n, alpha, A, lda, nnz, xVal, xInd, beta, y,
681 cusparseDgemvi_bufferSize(cusparseHandle_t handle, cusparseOperation_t transA,
687 return func_ptr(handle, transA, m, n, nnz, pBufferSize);
691 cusparseHandle_t handle, cusparseOperation_t transA, int m, int n,
702 return func_ptr(handle, transA, m, n, alpha, A, lda, nnz, xVal, xInd, beta, y,
[all …]
|
| D | cusparse_9_0.inc | 639 cusparseSgemvi(cusparseHandle_t handle, cusparseOperation_t transA, int m,
650 return func_ptr(handle, transA, m, n, alpha, A, lda, nnz, xVal, xInd, beta, y,
655 cusparseSgemvi_bufferSize(cusparseHandle_t handle, cusparseOperation_t transA,
661 return func_ptr(handle, transA, m, n, nnz, pBufferSize);
665 cusparseDgemvi(cusparseHandle_t handle, cusparseOperation_t transA, int m,
676 return func_ptr(handle, transA, m, n, alpha, A, lda, nnz, xVal, xInd, beta, y,
681 cusparseDgemvi_bufferSize(cusparseHandle_t handle, cusparseOperation_t transA,
687 return func_ptr(handle, transA, m, n, nnz, pBufferSize);
691 cusparseHandle_t handle, cusparseOperation_t transA, int m, int n,
702 return func_ptr(handle, transA, m, n, alpha, A, lda, nnz, xVal, xInd, beta, y,
[all …]
|
| D | cusparse_10_1.inc | 685 cusparseSgemvi(cusparseHandle_t handle, cusparseOperation_t transA, int m,
695 return func_ptr(handle, transA, m, n, alpha, A, lda, nnz, xVal, xInd, beta, y,
700 cusparseSgemvi_bufferSize(cusparseHandle_t handle, cusparseOperation_t transA,
706 return func_ptr(handle, transA, m, n, nnz, pBufferSize);
710 cusparseDgemvi(cusparseHandle_t handle, cusparseOperation_t transA, int m,
720 return func_ptr(handle, transA, m, n, alpha, A, lda, nnz, xVal, xInd, beta, y,
725 cusparseDgemvi_bufferSize(cusparseHandle_t handle, cusparseOperation_t transA,
731 return func_ptr(handle, transA, m, n, nnz, pBufferSize);
735 cusparseHandle_t handle, cusparseOperation_t transA, int m, int n,
745 return func_ptr(handle, transA, m, n, alpha, A, lda, nnz, xVal, xInd, beta, y,
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mat_mul_op.cc | 747 const gpusparseOperation_t transA = CUSPARSE_OPERATION_NON_TRANSPOSE; in Compute() local
749 const gpusparseOperation_t transA = HIPSPARSE_OPERATION_NON_TRANSPOSE; in Compute() local
800 cuda_sparse.Csrmm(transA, transB, m, n, k, nnz, &alpha, descrA, in Compute()
|