opennn::matmul::cudnn::Problem Struct Reference

#include <opennn/core/cuda/matmul_cudnn.h>

Public Attributes
int	m = 0
int	n = 0
int	k = 0
cublasOperation_t	transA = CUBLAS_OP_N
cublasOperation_t	transB = CUBLAS_OP_N
cublasLtEpilogue_t	epilogue = CUBLASLT_EPILOGUE_DEFAULT
cudaDataType_t	dtype_a = CUDA_R_32F
cudaDataType_t	dtype_b = CUDA_R_32F
cudaDataType_t	out_dtype = CUDA_R_32F
int	lda = 0
int	ldb = 0
int	ldd = 0
bool	beta_is_zero = true

Member Data Documentation

beta_is_zero

bool opennn::matmul::cudnn::Problem::beta_is_zero = true

dtype_a

cudaDataType_t opennn::matmul::cudnn::Problem::dtype_a = CUDA_R_32F

dtype_b

cudaDataType_t opennn::matmul::cudnn::Problem::dtype_b = CUDA_R_32F

epilogue

cublasLtEpilogue_t opennn::matmul::cudnn::Problem::epilogue = CUBLASLT_EPILOGUE_DEFAULT

k

int opennn::matmul::cudnn::Problem::k = 0

lda

int opennn::matmul::cudnn::Problem::lda = 0

ldb

int opennn::matmul::cudnn::Problem::ldb = 0

ldd

int opennn::matmul::cudnn::Problem::ldd = 0

m

int opennn::matmul::cudnn::Problem::m = 0

n

int opennn::matmul::cudnn::Problem::n = 0

out_dtype

cudaDataType_t opennn::matmul::cudnn::Problem::out_dtype = CUDA_R_32F

transA

cublasOperation_t opennn::matmul::cudnn::Problem::transA = CUBLAS_OP_N

transB

cublasOperation_t opennn::matmul::cudnn::Problem::transB = CUBLAS_OP_N