Public Types | Public Member Functions | List of all members
ML::SimpleVec< T > Struct Template Reference

#include <dense.hpp>

Inheritance diagram for ML::SimpleVec< T >:
Inheritance graph
Collaboration diagram for ML::SimpleVec< T >:
Collaboration graph

Public Types

typedef SimpleDenseMat< T > Super
 
- Public Types inherited from ML::SimpleDenseMat< T >
typedef SimpleMat< T > Super
 

Public Member Functions

 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)
 
- Public Member Functions inherited from ML::SimpleDenseMat< T >
 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
 
- Public Member Functions inherited from ML::SimpleMat< T >
 SimpleMat (int m, int n)
 
void operator= (const SimpleMat< T > &other)=delete
 

Additional Inherited Members

- Static Public Member Functions inherited from ML::SimpleDenseMat< T >
static void gemm (const raft::handle_t &handle, const T alpha, const SimpleDenseMat< T > &A, const bool transA, const SimpleDenseMat< T > &B, const bool transB, const T beta, SimpleDenseMat< T > &C, cudaStream_t stream)
 
- Public Attributes inherited from ML::SimpleDenseMat< T >
int len
 
T * data
 
STORAGE_ORDER ord
 
- Public Attributes inherited from ML::SimpleMat< T >
int m
 
int n
 

Member Typedef Documentation

◆ Super

template<typename T >
typedef SimpleDenseMat<T> ML::SimpleVec< T >::Super

Constructor & Destructor Documentation

◆ SimpleVec() [1/2]

template<typename T >
ML::SimpleVec< T >::SimpleVec ( T *  data,
const int  n 
)
inline

◆ SimpleVec() [2/2]

template<typename T >
ML::SimpleVec< T >::SimpleVec ( )
inline

Member Function Documentation

◆ assign_gemv()

template<typename T >
void ML::SimpleVec< T >::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 
)
inline

◆ reset()

template<typename T >
void ML::SimpleVec< T >::reset ( T *  new_data,
int  n 
)
inline

The documentation for this struct was generated from the following file: