struct nbla::CudnnConvResource

struct CudnnConvResource

cuDNN Convolution resource cache.

Public Functions

size_t max_workspace_size() const

Get maximum workspace size.

size_t fwd_workspace_size() const

Get forward workspace size.

size_t bwd_filter_workspace_size() const

Get backward-filter workspace size.

size_t bwd_data_workspace_size() const

Get backward-data workspace size.

Public Members

int device

Device ID.

cudnnTensorDescriptor_t x_desc

Input desc.

cudnnTensorDescriptor_t y_desc

Output desc.

cudnnTensorDescriptor_t b_desc

Bias desc.

cudnnTensorDescriptor_t b_desc_deconv

Bias desc for deconvolution.

cudnnFilterDescriptor_t w_desc

Weight desc.

CudnnConvolutionDescriptor conv_desc

Conv desc.

CudnnConvolutionDescriptor conv_dgrad_desc

Conv backward data desc.

CudnnConvolutionDescriptor conv_wgrad_desc

Conv backward filter desc.

cudnnConvolutionFwdAlgo_t fwd_algo

Best forward algorithm found.

cudnnConvolutionBwdFilterAlgo_t bwd_filter_algo

Best Backward filter algorithm found.

cudnnConvolutionBwdDataAlgo_t bwd_data_algo

Best backward data algorithm found.