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)#