Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dcuda_sparse.h33 using gpusparseOperation_t = cusparseOperation_t; variable
52 using gpusparseOperation_t = hipsparseOperation_t; variable
150 inline gpusparseOperation_t TransposeAndConjugateToGpuSparseOp(bool transpose, in TransposeAndConjugateToGpuSparseOp()
277 Status Csrmm(gpusparseOperation_t transA, gpusparseOperation_t transB, int m,
289 Status SpMMBufferSize(gpusparseOperation_t transA,
290 gpusparseOperation_t transB, const Scalar* alpha,
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,
[all …]
/external/tensorflow/tensorflow/core/kernels/sparse/
Dmat_mul_op.cc824 const gpusparseOperation_t transA = CUSPARSE_OPERATION_NON_TRANSPOSE; in Compute()
826 const gpusparseOperation_t transA = HIPSPARSE_OPERATION_NON_TRANSPOSE; in Compute()
834 const gpusparseOperation_t transB = CUSPARSE_OPERATION_TRANSPOSE; in Compute()
875 const gpusparseOperation_t transB = CUSPARSE_OPERATION_TRANSPOSE; in Compute()
886 const gpusparseOperation_t transB = HIPSPARSE_OPERATION_TRANSPOSE; in Compute()
970 const gpusparseOperation_t transA_;
Dsparse_mat_mul_op.cc735 gpusparseOperation_t transA_; in CSRSparseSparseMatrixMatMul()
736 gpusparseOperation_t transB_; in CSRSparseSparseMatrixMatMul()