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.
-
size_t max_workspace_size() const