mxnet
Static Public Member Functions | List of all members
mshadow::expr::BLASEngine< cpu, float > Struct Template Reference

#include <dot_engine-inl.h>

Collaboration diagram for mshadow::expr::BLASEngine< cpu, float >:
Collaboration graph

Static Public Member Functions

static CBLAS_TRANSPOSE GetT (bool t)
 
static void SetStream (Stream< cpu > *stream)
 
static void gemm (Stream< cpu > *stream, bool transa, bool transb, int m, int n, int k, float alpha, const float *A, int lda, const float *B, int ldb, float beta, float *C, int ldc)
 
static void batched_gemm (Stream< cpu > *stream, bool transa, bool transb, int m, int n, int k, float alpha, const float *A, int lda, const float *B, int ldb, float beta, float *C, int ldc, int batch_count, float **workspace)
 
static void gemv (Stream< cpu > *stream, bool trans, int m, int n, float alpha, const float *A, int lda, const float *X, int incX, float beta, float *Y, int incY)
 
static void batched_gemv (Stream< cpu > *stream, bool trans, int m, int n, float alpha, const float *A, int lda, const float *X, int incX, float beta, float *Y, int incY, int batch_count)
 
static void ger (Stream< cpu > *stream, int m, int n, float alpha, const float *X, int incX, const float *Y, int incY, float *A, int lda)
 
static void batched_ger (Stream< cpu > *stream, int m, int n, float alpha, const float *X, int incX, const float *Y, int incY, float *A, int lda, int batch_count)
 
static void dot (Stream< cpu > *stream, int n, const float *X, int incX, const float *Y, int incY, float *ret)
 

Member Function Documentation

static void mshadow::expr::BLASEngine< cpu, float >::batched_gemm ( Stream< cpu > *  stream,
bool  transa,
bool  transb,
int  m,
int  n,
int  k,
float  alpha,
const float *  A,
int  lda,
const float *  B,
int  ldb,
float  beta,
float *  C,
int  ldc,
int  batch_count,
float **  workspace 
)
inlinestatic
static void mshadow::expr::BLASEngine< cpu, float >::batched_gemv ( Stream< cpu > *  stream,
bool  trans,
int  m,
int  n,
float  alpha,
const float *  A,
int  lda,
const float *  X,
int  incX,
float  beta,
float *  Y,
int  incY,
int  batch_count 
)
inlinestatic
static void mshadow::expr::BLASEngine< cpu, float >::batched_ger ( Stream< cpu > *  stream,
int  m,
int  n,
float  alpha,
const float *  X,
int  incX,
const float *  Y,
int  incY,
float *  A,
int  lda,
int  batch_count 
)
inlinestatic
static void mshadow::expr::BLASEngine< cpu, float >::dot ( Stream< cpu > *  stream,
int  n,
const float *  X,
int  incX,
const float *  Y,
int  incY,
float *  ret 
)
inlinestatic
static void mshadow::expr::BLASEngine< cpu, float >::gemm ( Stream< cpu > *  stream,
bool  transa,
bool  transb,
int  m,
int  n,
int  k,
float  alpha,
const float *  A,
int  lda,
const float *  B,
int  ldb,
float  beta,
float *  C,
int  ldc 
)
inlinestatic
static void mshadow::expr::BLASEngine< cpu, float >::gemv ( Stream< cpu > *  stream,
bool  trans,
int  m,
int  n,
float  alpha,
const float *  A,
int  lda,
const float *  X,
int  incX,
float  beta,
float *  Y,
int  incY 
)
inlinestatic
static void mshadow::expr::BLASEngine< cpu, float >::ger ( Stream< cpu > *  stream,
int  m,
int  n,
float  alpha,
const float *  X,
int  incX,
const float *  Y,
int  incY,
float *  A,
int  lda 
)
inlinestatic
static CBLAS_TRANSPOSE mshadow::expr::BLASEngine< cpu, float >::GetT ( bool  t)
inlinestatic
static void mshadow::expr::BLASEngine< cpu, float >::SetStream ( Stream< cpu > *  stream)
inlinestatic

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