| /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, \
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()
265 hipsparseOperation_t transA, hipsparseOperation_t transB, int m, int n, in CsrgemmImpl() argument
273 op(hipsparse_handle, transA, transB, m, n, k, descrA, nnzA, csrSortedValA, in CsrgemmImpl()
283 hipsparseOperation_t transA, hipsparseOperation_t transB, int m, int n, \
292 *gpusparse_handle_, transA, transB, m, n, k, descrA, \
|
| 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, \
559 cusparseOperation_t transA, cusparseOperation_t transB, int m, int k, int n, in CsrgemmNnz() argument
568 *gpusparse_handle_, transA, transB, m, k, n, descrA, nnzA, in CsrgemmNnz()
577 cusparseOperation_t transA, cusparseOperation_t transB, int m, int k, int n, in CsrgemmImpl() argument
585 op(cusparse_handle, transA, transB, m, k, n, descrA, nnzA, in CsrgemmImpl()
596 cusparseOperation_t transA, cusparseOperation_t transB, int m, int k, \
605 *gpusparse_handle_, transA, transB, m, k, n, descrA, \
|
| D | cuda_sparse.h | 294 Status Csrmm(gpusparseOperation_t transA, gpusparseOperation_t transB, int m,
346 Status CsrgemmNnz(gpusparseOperation_t transA, gpusparseOperation_t transB,
361 Status Csrgemm(gpusparseOperation_t transA, gpusparseOperation_t transB,
|
| D | sparse_matmul_op.cc | 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()
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()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cusparse_10_0.inc | 1986 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 …]
|
| D | cusparse_10_1.inc | 2028 cusparseOperation_t transB, int m, int n, int k, int nnz,
2039 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2046 cusparseOperation_t transB, int m, int n, int k, int nnz,
2058 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2065 cusparseOperation_t transB, int m, int n, int k, int nnz,
2077 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2084 cusparseOperation_t transB, int m, int n, int k, int nnz,
2097 return func_ptr(handle, transA, transB, m, n, k, nnz, alpha, descrA,
2104 cusparseOperation_t transA, cusparseOperation_t transB, int mb, int n,
2116 return func_ptr(handle, dirA, transA, transB, mb, n, kb, nnzb, alpha, descrA,
[all …]
|
| D | cusparse_9_0.inc | 1986 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/sparse/ |
| D | mat_mul_op.cc | 756 const gpusparseOperation_t transB = CUSPARSE_OPERATION_TRANSPOSE; in Compute() local
765 const gpusparseOperation_t transB = HIPSPARSE_OPERATION_TRANSPOSE; in Compute() local
800 cuda_sparse.Csrmm(transA, transB, m, n, k, nnz, &alpha, descrA, in Compute()
|