Template Struct TypedSparseGemm< float >

• Defined in File prod_sparse_cu10.h

Struct Documentation

template<>
struct TypedSparseGemm<float>

Public Static Functions

static cusparseStatus_t cusparseSgemmiEx(cusparseHandle_t handle, int m, int n, int k, int nnz, const float *alpha, const float *A, int lda, const float *cscValB, const int *cscColPtrB, const int *cscRowIndB, const float *beta, float *C, int ldc)

static void CSRProd(marian::Tensor C, Ptr<Allocator> allocator, const marian::Tensor &S_values, const marian::Tensor &S_indices, const marian::Tensor &S_offsets, const marian::Tensor &D, bool transS, bool swapOperands, float beta)