Home
last modified time | relevance | path

Searched refs:transA (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Drocm_sparse.cc183 hipsparseOperation_t transA, hipsparseOperation_t transB, int m, int n, in CsrmmImpl() argument
188 TF_RETURN_IF_GPUSPARSE_ERROR(op(hipsparse_handle, transA, transB, m, n, k, in CsrmmImpl()
198 hipsparseOperation_t transA, hipsparseOperation_t transB, int m, int n, \
206 *gpusparse_handle_, transA, transB, m, n, k, nnz, \
216 hipsparseOperation_t transA, int m, int n, in CsrmvImpl() argument
224 op(hipsparse_handle, transA, m, n, nnz, alpha_host, descrA, csrSortedValA, in CsrmvImpl()
233 hipsparseOperation_t transA, int m, int n, int nnz, \
240 *gpusparse_handle_, transA, m, n, nnz, alpha_host, \
248 hipsparseOperation_t transA, hipsparseOperation_t transB, int m, int n, in CsrgemmNnz() argument
257 *gpusparse_handle_, transA, transB, m, n, k, descrA, nnzA, in CsrgemmNnz()
[all …]
Dcuda_sparse.cc394 cusparseOperation_t transA, cusparseOperation_t transB, int m, int n, int k, in CsrmmImpl() argument
407 cusparse_handle, transA, transB, m, n, k, nnz, AsCudaComplex(alpha_host), in CsrmmImpl()
416 cusparseOperation_t transA, cusparseOperation_t transB, int m, int n, \
424 *gpusparse_handle_, transA, transB, m, n, k, nnz, \
436 cusparseOperation_t transA, cusparseOperation_t transB, \
443 *gpusparse_handle_, transA, transB, alpha, matA, matB, beta, matC, \
453 cusparseOperation_t transA, cusparseOperation_t transB, \
458 TF_RETURN_IF_GPUSPARSE_ERROR(cusparseSpMM(*gpusparse_handle_, transA, \
473 cusparseOperation_t transA, int m, int n, int nnz, const Scalar* alpha_host, in CsrmvImpl() argument
478 op(cusparse_handle, transA, m, n, nnz, AsCudaComplex(alpha_host), descrA, in CsrmvImpl()
[all …]
Dcuda_sparse.h277 Status Csrmm(gpusparseOperation_t transA, gpusparseOperation_t transB, int m,
289 Status SpMMBufferSize(gpusparseOperation_t transA,
303 Status SpMM(gpusparseOperation_t transA, gpusparseOperation_t transB,
317 Status Csrmv(gpusparseOperation_t transA, int m, int n, int nnz,
324 Status Csrmv(gpusparseOperation_t transA, int m, int n, int nnz,
388 Status CsrgemmNnz(gpusparseOperation_t transA, gpusparseOperation_t transB,
414 Status Csrgemm(gpusparseOperation_t transA, gpusparseOperation_t transB,
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcusparse_10_0.inc639 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 …]
Dcusparse_9_0.inc639 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 …]
Dcusparse_10_1.inc653 cusparseSgemvi(cusparseHandle_t handle, cusparseOperation_t transA, int m,
663 return func_ptr(handle, transA, m, n, alpha, A, lda, nnz, xVal, xInd, beta, y,
668 cusparseSgemvi_bufferSize(cusparseHandle_t handle, cusparseOperation_t transA,
674 return func_ptr(handle, transA, m, n, nnz, pBufferSize);
678 cusparseDgemvi(cusparseHandle_t handle, cusparseOperation_t transA, int m,
688 return func_ptr(handle, transA, m, n, alpha, A, lda, nnz, xVal, xInd, beta, y,
693 cusparseDgemvi_bufferSize(cusparseHandle_t handle, cusparseOperation_t transA,
699 return func_ptr(handle, transA, m, n, nnz, pBufferSize);
703 cusparseHandle_t handle, cusparseOperation_t transA, int m, int n,
713 return func_ptr(handle, transA, m, n, alpha, A, lda, nnz, xVal, xInd, beta, y,
[all …]
Dcusparse_10_2.inc653 cusparseSgemvi(cusparseHandle_t handle, cusparseOperation_t transA, int m,
663 return func_ptr(handle, transA, m, n, alpha, A, lda, nnz, xVal, xInd, beta, y,
668 cusparseSgemvi_bufferSize(cusparseHandle_t handle, cusparseOperation_t transA,
674 return func_ptr(handle, transA, m, n, nnz, pBufferSize);
678 cusparseDgemvi(cusparseHandle_t handle, cusparseOperation_t transA, int m,
688 return func_ptr(handle, transA, m, n, alpha, A, lda, nnz, xVal, xInd, beta, y,
693 cusparseDgemvi_bufferSize(cusparseHandle_t handle, cusparseOperation_t transA,
699 return func_ptr(handle, transA, m, n, nnz, pBufferSize);
703 cusparseHandle_t handle, cusparseOperation_t transA, int m, int n,
713 return func_ptr(handle, transA, m, n, alpha, A, lda, nnz, xVal, xInd, beta, y,
[all …]
Dcusparse_11_0.inc533 cusparseSgemvi(cusparseHandle_t handle, cusparseOperation_t transA, int m,
543 return func_ptr(handle, transA, m, n, alpha, A, lda, nnz, xVal, xInd, beta, y,
548 cusparseSgemvi_bufferSize(cusparseHandle_t handle, cusparseOperation_t transA,
554 return func_ptr(handle, transA, m, n, nnz, pBufferSize);
558 cusparseDgemvi(cusparseHandle_t handle, cusparseOperation_t transA, int m,
568 return func_ptr(handle, transA, m, n, alpha, A, lda, nnz, xVal, xInd, beta, y,
573 cusparseDgemvi_bufferSize(cusparseHandle_t handle, cusparseOperation_t transA,
579 return func_ptr(handle, transA, m, n, nnz, pBufferSize);
583 cusparseHandle_t handle, cusparseOperation_t transA, int m, int n,
593 return func_ptr(handle, transA, m, n, alpha, A, lda, nnz, xVal, xInd, beta, y,
[all …]
/external/tensorflow/tensorflow/core/kernels/sparse/
Dmat_mul_op.cc824 const gpusparseOperation_t transA = CUSPARSE_OPERATION_NON_TRANSPOSE; in Compute() local
826 const gpusparseOperation_t transA = HIPSPARSE_OPERATION_NON_TRANSPOSE; in Compute() local
855 transA, transB, &alpha, matA, matB, &beta, matC, in Compute()
863 TF_RETURN_IF_ERROR(cuda_sparse.SpMM(transA, transB, &alpha, matA, matB, in Compute()
897 cuda_sparse.Csrmm(transA, transB, m, n, k, nnz, &alpha, descrA, in Compute()
/external/tensorflow/tensorflow/core/kernels/
Dsparse_matmul_op.cc1389 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()