struct nbla::CudnnConvDesc
-
struct CudnnConvDesc
cuDNN Convolution Descriptor used as a key to find previously used (cached) config.
Public Functions
-
bool operator==(const CudnnConvDesc &right) const
Operator == compares all elements.
Public Members
-
int ndim
Dimension of spatial samples.
-
int device
Device ID.
-
cudnnDataType_t dtype
Data type.
-
cudnnConvolutionMode_t mode
CUDNN_CONVOLUTION or CUDNN_CROSS_CORRELATION;.
-
int n
Batch size.
-
int c
Channels of input.
-
int o
Channels of output.
-
int group
Number of groups.
-
bool channel_last
Channels at last dimension (NHWC).
-
vector<int> sample
Sample size of each dimension.
-
vector<int> kernel
Kernel size of each dimension.
-
vector<int> pad
Padding size of each dimension.
-
vector<int> stride
Stride size of each dimension.
-
vector<int> dilation
Dilation size of each dimension.
-
class Hash
Custom hash function for CudnnConvDesc.
-
bool operator==(const CudnnConvDesc &right) const