CUDNN Frontend API
8.2.0
|
#include <cudnn_frontend_ConvDesc.h>
Public Member Functions | |
std::string | describe () const override |
Return a string describing the backend Descriptor. More... | |
ConvDesc_v8 (ConvDesc_v8 &&from) | |
~ConvDesc_v8 ()=default | |
![]() | |
cudnnBackendDescriptor_t | get_raw_desc () const |
cudnnStatus_t | get_status () const |
Current status of the descriptor. More... | |
void | set_status (cudnnStatus_t const status_) const |
Set status of the descriptor. More... | |
void | set_error (const char *message) const |
Set Diagonistic error message. More... | |
const char * | get_error () const |
Diagonistic error message if any. More... | |
ManagedOpaqueDescriptor | get_desc () const |
Returns a copy of underlying managed descriptor. More... | |
cudnnStatus_t | initialize_managed_backend_pointer (cudnnBackendDescriptorType_t type) |
Initializes the underlying managed descriptor. More... | |
Private Member Functions | |
ConvDesc_v8 ()=default | |
ConvDesc_v8 (ConvDesc_v8 const &)=delete | |
ConvDesc_v8 & | operator= (ConvDesc_v8 const &)=delete |
Private Attributes | |
cudnnDataType_t | data_type = CUDNN_DATA_FLOAT |
cudnnConvolutionMode_t | mode = CUDNN_CONVOLUTION |
Convolution operation data type. More... | |
int64_t | nDims = -1 |
Convolution vs cross correlation. More... | |
int64_t | padLower [CUDNN_DIM_MAX+1] = {0} |
number of dimensions More... | |
int64_t | padUpper [CUDNN_DIM_MAX+1] = {0} |
n, g, c, d, h, w More... | |
int64_t | dilation [CUDNN_DIM_MAX+1] = {0} |
n, g, c, d, h, w More... | |
int64_t | stride [CUDNN_DIM_MAX+1] = {-1} |
n, g, c, d, h, w More... | |
Friends | |
class | ConvDescBuilder_v8 |
Additional Inherited Members | |
![]() | |
BackendDescriptor (ManagedOpaqueDescriptor pointer_, cudnnStatus_t status_, std::string err_msg_) | |
BackendDescriptor ()=default | |
![]() | |
ManagedOpaqueDescriptor | pointer |
cudnnStatus_t | status = CUDNN_STATUS_SUCCESS |
Shared pointer of the OpaqueBackendPointer. More... | |
std::string | err_msg |
Error message if any being set. More... | |
Convolution Descriptor Class This class tells the properties of the Convolution operation Properties:
Use ConvDescBuilder_v8 to build this class. Describe returns a string describing the convolution operation
Definition at line 54 of file cudnn_frontend_ConvDesc.h.
|
inline |
Definition at line 88 of file cudnn_frontend_ConvDesc.h.
References ConvDesc_v8(), dilation, operator=(), padLower, padUpper, stride, and ~ConvDesc_v8().
|
default |
Referenced by ConvDesc_v8().
|
privatedefault |
Referenced by ConvDesc_v8().
|
privatedelete |
|
inlineoverridevirtual |
Return a string describing the backend Descriptor.
Implements cudnn_frontend::BackendDescriptor.
Definition at line 58 of file cudnn_frontend_ConvDesc.h.
References data_type, dilation, mode, nDims, padLower, padUpper, stride, and nlohmann::to_string().
|
privatedelete |
Referenced by cudnn_frontend::ConvDescBuilder_v8::build(), and ConvDesc_v8().
|
friend |
Definition at line 56 of file cudnn_frontend_ConvDesc.h.
Referenced by cudnn_frontend::ConvDescBuilder_v8::build().
|
private |
Definition at line 107 of file cudnn_frontend_ConvDesc.h.
Referenced by describe().
|
private |
n, g, c, d, h, w
Definition at line 112 of file cudnn_frontend_ConvDesc.h.
Referenced by ConvDesc_v8(), describe(), and cudnn_frontend::ConvDescBuilder_v8::setDilation().
|
private |
Convolution operation data type.
Definition at line 108 of file cudnn_frontend_ConvDesc.h.
Referenced by describe().
|
private |
Convolution vs cross correlation.
Definition at line 109 of file cudnn_frontend_ConvDesc.h.
Referenced by describe().
|
private |
number of dimensions
Definition at line 110 of file cudnn_frontend_ConvDesc.h.
Referenced by ConvDesc_v8(), and describe().
|
private |
n, g, c, d, h, w
Definition at line 111 of file cudnn_frontend_ConvDesc.h.
Referenced by ConvDesc_v8(), and describe().
|
private |
n, g, c, d, h, w
Definition at line 113 of file cudnn_frontend_ConvDesc.h.
Referenced by ConvDesc_v8(), and describe().