Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
at::native::ConvolutionDescriptor Struct Reference
Inheritance diagram for at::native::ConvolutionDescriptor:
at::native::Descriptor< miopenConvolutionDescriptor,&miopenCreateConvolutionDescriptor,&miopenDestroyConvolutionDescriptor > at::native::Descriptor< cudnnConvolutionStruct,&cudnnCreateConvolutionDescriptor,&cudnnDestroyConvolutionDescriptor >

Public Member Functions

void set (cudnnDataType_t dataType, int dim, int *pad, int *stride, int *upscale, int groups)
 
void set (miopenDataType_t dataType, miopenConvolutionMode_t c_mode, int dim, int *pad, int *stride, int *upscale, int groups)
 
- Public Member Functions inherited from at::native::Descriptor< miopenConvolutionDescriptor,&miopenCreateConvolutionDescriptor,&miopenDestroyConvolutionDescriptor >
miopenConvolutionDescriptor * desc () const
 
miopenConvolutionDescriptor * desc ()
 
miopenConvolutionDescriptor * desc () const
 
miopenConvolutionDescriptor * desc ()
 
miopenConvolutionDescriptor * mut_desc ()
 
miopenConvolutionDescriptor * mut_desc ()
 
- Public Member Functions inherited from at::native::Descriptor< cudnnConvolutionStruct,&cudnnCreateConvolutionDescriptor,&cudnnDestroyConvolutionDescriptor >
cudnnConvolutionStruct * desc () const
 
cudnnConvolutionStruct * desc ()
 
cudnnConvolutionStruct * desc () const
 
cudnnConvolutionStruct * desc ()
 
cudnnConvolutionStruct * mut_desc ()
 
cudnnConvolutionStruct * mut_desc ()
 

Additional Inherited Members

- Protected Member Functions inherited from at::native::Descriptor< miopenConvolutionDescriptor,&miopenCreateConvolutionDescriptor,&miopenDestroyConvolutionDescriptor >
void init ()
 
void init ()
 
- Protected Member Functions inherited from at::native::Descriptor< cudnnConvolutionStruct,&cudnnCreateConvolutionDescriptor,&cudnnDestroyConvolutionDescriptor >
void init ()
 
void init ()
 

Detailed Description

Definition at line 147 of file Descriptors.h.


The documentation for this struct was generated from the following file: