|
| SimpleVecOwning ()=delete |
|
| SimpleVecOwning (int n, cudaStream_t stream) |
|
void | operator= (const SimpleVec< T > &other)=delete |
|
| SimpleVec (T *data, const int n) |
|
void | assign_gemv (const raft::handle_t &handle, const T alpha, const SimpleDenseMat< T > &A, bool transA, const SimpleVec< T > &x, const T beta, cudaStream_t stream) |
|
| SimpleVec () |
|
void | reset (T *new_data, int n) |
|
| SimpleDenseMat (STORAGE_ORDER order=COL_MAJOR) |
|
| SimpleDenseMat (T *data, int m, int n, STORAGE_ORDER order=COL_MAJOR) |
|
void | reset (T *data_, int m_, int n_) |
|
void | gemmb (const raft::handle_t &handle, const T alpha, const SimpleDenseMat< T > &A, const bool transA, const bool transB, const T beta, SimpleDenseMat< T > &C, cudaStream_t stream) const override |
|
void | assign_gemm (const raft::handle_t &handle, const T alpha, const SimpleDenseMat< T > &A, const bool transA, const SimpleMat< T > &B, const bool transB, const T beta, cudaStream_t stream) |
|
void | ax (const T a, const SimpleDenseMat< T > &x, cudaStream_t stream) |
|
void | axpy (const T a, const SimpleDenseMat< T > &x, const SimpleDenseMat< T > &y, cudaStream_t stream) |
|
template<typename Lambda > |
void | assign_unary (const SimpleDenseMat< T > &other, Lambda f, cudaStream_t stream) |
|
template<typename Lambda > |
void | assign_binary (const SimpleDenseMat< T > &other1, const SimpleDenseMat< T > &other2, Lambda &f, cudaStream_t stream) |
|
template<typename Lambda > |
void | assign_ternary (const SimpleDenseMat< T > &other1, const SimpleDenseMat< T > &other2, const SimpleDenseMat< T > &other3, Lambda &f, cudaStream_t stream) |
|
void | fill (const T val, cudaStream_t stream) |
|
void | copy_async (const SimpleDenseMat< T > &other, cudaStream_t stream) |
|
void | print (std::ostream &oss) const override |
|
void | operator= (const SimpleDenseMat< T > &other)=delete |
|
| SimpleMat (int m, int n) |
|
void | operator= (const SimpleMat< T > &other)=delete |
|