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

#include <dot_engine-inl.h>

Collaboration diagram for mshadow::expr::BLASEngine< gpu, half::half_t >:
Collaboration graph

Static Public Member Functions

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

Member Function Documentation

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

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