Collaboration diagram for mshadow::expr::BLASEngine< gpu, double >:

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, double alpha, const double A, int lda, const double B, int ldb, double beta, double C, int ldc)

static void	batched_gemm (Stream< gpu > stream, bool transa, bool transb, int m, int n, int k, double alpha, const double A, int lda, const double B, int ldb, double beta, double C, int ldc, int batch_count, double **workspace)

static void	gemv (Stream< gpu > stream, bool trans, int m, int n, double alpha, const double A, int lda, const double X, int incX, double beta, double Y, int incY)

static void	batched_gemv (Stream< gpu > stream, bool trans, int m, int n, double alpha, const double A, int lda, const double X, int incX, double beta, double Y, int incY, int batch_count)

static void	ger (Stream< gpu > stream, int m, int n, double alpha, const double X, int incX, const double Y, int incY, double A, int lda)

static void	batched_ger (Stream< gpu > stream, int m, int n, double alpha, const double X, int incX, const double Y, int incY, double A, int lda, int batch_count)

static void	dot (Stream< gpu > stream, int n, const double X, int incX, const double Y, int incY, double ret)

Member Function Documentation

static void mshadow::expr::BLASEngine< gpu, double >::batched_gemm	(	Stream< gpu > *	stream,
		bool	transa,
		bool	transb,
		int	m,
		int	n,
		int	k,
		double	alpha,
		const double *	A,
		int	lda,
		const double *	B,
		int	ldb,
		double	beta,
		double *	C,
		int	ldc,
		int	batch_count,
		double **	workspace
	)

inlinestatic

static void mshadow::expr::BLASEngine< gpu, double >::batched_gemv	(	Stream< gpu > *	stream,
		bool	trans,
		int	m,
		int	n,
		double	alpha,
		const double *	A,
		int	lda,
		const double *	X,
		int	incX,
		double	beta,
		double *	Y,
		int	incY,
		int	batch_count
	)

inlinestatic

static void mshadow::expr::BLASEngine< gpu, double >::batched_ger	(	Stream< gpu > *	stream,
		int	m,
		int	n,
		double	alpha,
		const double *	X,
		int	incX,
		const double *	Y,
		int	incY,
		double *	A,
		int	lda,
		int	batch_count
	)

inlinestatic

static void mshadow::expr::BLASEngine< gpu, double >::dot	(	Stream< gpu > *	stream,
		int	n,
		const double *	X,
		int	incX,
		const double *	Y,
		int	incY,
		double *	ret
	)

inlinestatic

static void mshadow::expr::BLASEngine< gpu, double >::gemm	(	Stream< gpu > *	stream,
		bool	transa,
		bool	transb,
		int	m,
		int	n,
		int	k,
		double	alpha,
		const double *	A,
		int	lda,
		const double *	B,
		int	ldb,
		double	beta,
		double *	C,
		int	ldc
	)

inlinestatic

static void mshadow::expr::BLASEngine< gpu, double >::gemv	(	Stream< gpu > *	stream,
		bool	trans,
		int	m,
		int	n,
		double	alpha,
		const double *	A,
		int	lda,
		const double *	X,
		int	incX,
		double	beta,
		double *	Y,
		int	incY
	)

inlinestatic

static void mshadow::expr::BLASEngine< gpu, double >::ger	(	Stream< gpu > *	stream,
		int	m,
		int	n,
		double	alpha,
		const double *	X,
		int	incX,
		const double *	Y,
		int	incY,
		double *	A,
		int	lda
	)

inlinestatic

static cublasOperation_t mshadow::expr::BLASEngine< gpu, double >::GetT ( bool t )

inlinestatic

static void mshadow::expr::BLASEngine< gpu, double >::SetStream ( Stream< gpu > * stream )

inlinestatic

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

Static Public Member Functions