Struct ConvolutionDescriptor¶
Defined in File Descriptors.h
Page Contents
Inheritance Relationships¶
Base Type¶
public at::native::Descriptor< cudnnConvolutionStruct, &cudnnCreateConvolutionDescriptor, &cudnnDestroyConvolutionDescriptor >
(Template Class Descriptor)
Struct Documentation¶
-
struct ConvolutionDescriptor : public at::native::Descriptor<cudnnConvolutionStruct, &cudnnCreateConvolutionDescriptor, &cudnnDestroyConvolutionDescriptor>¶
Public Functions
-
inline void set(cudnnDataType_t dataType, int dim, int *pad, int *stride, int *upscale, int groups, bool allow_tf32)¶
-
inline void set(cudnnDataType_t dataType, int dim, int *pad, int *stride, int *upscale, int groups, bool allow_tf32)¶