mxnet
mshadow::expr::BLASEngine< gpu, half::half_t > Member List

This is the complete list of members for mshadow::expr::BLASEngine< gpu, half::half_t >, including all inherited members.

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)mshadow::expr::BLASEngine< gpu, half::half_t >inlinestatic
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)mshadow::expr::BLASEngine< gpu, half::half_t >inlinestatic
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)mshadow::expr::BLASEngine< gpu, half::half_t >inlinestatic
dot(Stream< gpu > *stream, int n, const half::half_t *X, int incX, const half::half_t *Y, int incY, half::half_t *ret)mshadow::expr::BLASEngine< gpu, half::half_t >inlinestatic
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)mshadow::expr::BLASEngine< gpu, half::half_t >inlinestatic
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)mshadow::expr::BLASEngine< gpu, half::half_t >inlinestatic
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)mshadow::expr::BLASEngine< gpu, half::half_t >inlinestatic
GetT(bool t)mshadow::expr::BLASEngine< gpu, half::half_t >inlinestatic
SetStream(Stream< gpu > *stream)mshadow::expr::BLASEngine< gpu, half::half_t >inlinestatic