CUDNN API
8
|
Functions | |
auto | cudnn_frontend::ConvDescBuilder_v8::setDataType (cudnnDataType_t data_type_) -> ConvDescBuilder_v8 & |
Set Datatype for the Convolution Operation. More... | |
auto | cudnn_frontend::ConvDescBuilder_v8::setPrePadding (int64_t ndims, int64_t const *padding) -> ConvDescBuilder_v8 & |
Set Padding Lower of the convDesc. More... | |
auto | cudnn_frontend::ConvDescBuilder_v8::setPostPadding (int64_t ndims, int64_t const *padding) -> ConvDescBuilder_v8 & |
Set Padding Upper of the convDesc. More... | |
auto | cudnn_frontend::ConvDescBuilder_v8::setDilation (int64_t ndims, int64_t const *dilation) -> ConvDescBuilder_v8 & |
Set Dilation of the convDesc. More... | |
auto | cudnn_frontend::ConvDescBuilder_v8::setStrides (int64_t ndims, int64_t const *strides) -> ConvDescBuilder_v8 & |
Set Strides of the convDesc. More... | |
auto | cudnn_frontend::ConvDescBuilder_v8::setNDims (int64_t nDims_) -> ConvDescBuilder_v8 & |
Set Num Spatial Dimensions of the convolution Operation. More... | |
auto | cudnn_frontend::ConvDescBuilder_v8::setMathMode (cudnnConvolutionMode_t mode_) -> ConvDescBuilder_v8 & |
Set Convolution Mode of the convolution Operation. More... | |
Set individual property of ConvDesc_v8 class
|
inline |
Set Datatype for the Convolution Operation.
Definition at line 127 of file cudnn_frontend_ConvDesc.h.
|
inline |
Set Dilation of the convDesc.
Definition at line 145 of file cudnn_frontend_ConvDesc.h.
References cudnn_frontend::ConvDesc_v8::dilation.
|
inline |
Set Convolution Mode of the convolution Operation.
Definition at line 163 of file cudnn_frontend_ConvDesc.h.
|
inline |
Set Num Spatial Dimensions of the convolution Operation.
Definition at line 157 of file cudnn_frontend_ConvDesc.h.
|
inline |
Set Padding Upper of the convDesc.
Definition at line 139 of file cudnn_frontend_ConvDesc.h.
|
inline |
Set Padding Lower of the convDesc.
Definition at line 133 of file cudnn_frontend_ConvDesc.h.
|
inline |
Set Strides of the convDesc.
Definition at line 151 of file cudnn_frontend_ConvDesc.h.