Home
last modified time | relevance | path

Searched refs:transB (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, \
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()
266 hipsparseOperation_t transA, hipsparseOperation_t transB, int m, int n, in CsrgemmImpl() argument
274 op(hipsparse_handle, transA, transB, m, n, k, descrA, nnzA, csrSortedValA, in CsrgemmImpl()
284 hipsparseOperation_t transA, hipsparseOperation_t transB, int m, int n, \
293 *gpusparse_handle_, transA, transB, m, n, k, descrA, \
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, \
459 transB, alpha, matA, matB, beta, \
764 cusparseOperation_t transA, cusparseOperation_t transB, int m, int k, int n, in CsrgemmNnz() argument
773 *gpusparse_handle_, transA, transB, m, k, n, descrA, nnzA, in CsrgemmNnz()
[all …]
Dcuda_sparse.h277 Status Csrmm(gpusparseOperation_t transA, gpusparseOperation_t transB, int m,
290 gpusparseOperation_t transB, const Scalar* alpha,
303 Status SpMM(gpusparseOperation_t transA, gpusparseOperation_t transB,
388 Status CsrgemmNnz(gpusparseOperation_t transA, gpusparseOperation_t transB,
414 Status Csrgemm(gpusparseOperation_t transA, gpusparseOperation_t transB,
/external/tensorflow/tensorflow/core/kernels/sparse/
Dmat_mul_op.cc834 const gpusparseOperation_t transB = CUSPARSE_OPERATION_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()
875 const gpusparseOperation_t transB = CUSPARSE_OPERATION_TRANSPOSE; in Compute() local
886 const gpusparseOperation_t transB = HIPSPARSE_OPERATION_TRANSPOSE; in Compute() local
897 cuda_sparse.Csrmm(transA, transB, m, n, k, nnz, &alpha, descrA, in Compute()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcusparse_10_0.inc1986 cusparseOperation_t transB, int m, int n, int k, int nnz,
1997 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2004 cusparseOperation_t transB, int m, int n, int k, int nnz,
2016 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2023 cusparseOperation_t transB, int m, int n, int k, int nnz,
2035 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2042 cusparseOperation_t transB, int m, int n, int k, int nnz,
2055 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2062 cusparseOperation_t transA, cusparseOperation_t transB, int mb, int n,
2074 return func_ptr(handle, dirA, transA, transB, mb, n, kb, nnzb, alpha, descrA,
[all …]
Dcusparse_10_1.inc1996 cusparseOperation_t transB, int m, int n, int k, int nnz,
2007 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2014 cusparseOperation_t transB, int m, int n, int k, int nnz,
2026 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2033 cusparseOperation_t transB, int m, int n, int k, int nnz,
2045 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2052 cusparseOperation_t transB, int m, int n, int k, int nnz,
2065 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2072 cusparseOperation_t transA, cusparseOperation_t transB, int mb, int n,
2084 return func_ptr(handle, dirA, transA, transB, mb, n, kb, nnzb, alpha, descrA,
[all …]
Dcusparse_10_2.inc1996 cusparseOperation_t transB, int m, int n, int k, int nnz,
2007 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2014 cusparseOperation_t transB, int m, int n, int k, int nnz,
2026 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2033 cusparseOperation_t transB, int m, int n, int k, int nnz,
2045 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2052 cusparseOperation_t transB, int m, int n, int k, int nnz,
2065 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2072 cusparseOperation_t transA, cusparseOperation_t transB, int mb, int n,
2084 return func_ptr(handle, dirA, transA, transB, mb, n, kb, nnzb, alpha, descrA,
[all …]
Dcusparse_11_0.inc1369 cusparseOperation_t transA, cusparseOperation_t transB, int mb, int n,
1381 return func_ptr(handle, dirA, transA, transB, mb, n, kb, nnzb, alpha, descrA,
1388 cusparseOperation_t transA, cusparseOperation_t transB, int mb, int n,
1400 return func_ptr(handle, dirA, transA, transB, mb, n, kb, nnzb, alpha, descrA,
1407 cusparseOperation_t transA, cusparseOperation_t transB, int mb, int n,
1420 return func_ptr(handle, dirA, transA, transB, mb, n, kb, nnzb, alpha, descrA,
1427 cusparseOperation_t transA, cusparseOperation_t transB, int mb, int n,
1441 return func_ptr(handle, dirA, transA, transB, mb, n, kb, nnzb, alpha, descrA,
1529 cusparseOperation_t transB, int m, int nrhs, int nnz, const float *alpha,
1541 return func_ptr(handle, algo, transA, transB, m, nrhs, nnz, alpha, descrA,
[all …]
Dcusparse_9_0.inc1986 cusparseOperation_t transB, int m, int n, int k, int nnz,
1997 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2004 cusparseOperation_t transB, int m, int n, int k, int nnz,
2016 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2023 cusparseOperation_t transB, int m, int n, int k, int nnz,
2035 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2042 cusparseOperation_t transB, int m, int n, int k, int nnz,
2055 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2062 cusparseOperation_t transA, cusparseOperation_t transB, int mb, int n,
2074 return func_ptr(handle, dirA, transA, transB, mb, n, kb, nnzb, alpha, descrA,
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dsparse_matmul_op.cc1407 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()
1418 char transB, const float* alpha, libxsmm_CSR_sparseslice* A_sparse, 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()