PyTorch
Loading...
Searching...
No Matches
Public Member Functions | List of all members
at::native::ConvolutionDescriptor Struct Reference

#include <Descriptors.h>

Inheritance diagram for at::native::ConvolutionDescriptor:
at::native::Descriptor< cudnnConvolutionStruct, &cudnnCreateConvolutionDescriptor, &cudnnDestroyConvolutionDescriptor >

Public Member Functions

void set (cudnnDataType_t dataType, int dim, int *pad, int *stride, int *upscale, int groups, bool allow_tf32)
 

Member Function Documentation

◆ set()

void at::native::ConvolutionDescriptor::set ( cudnnDataType_t  dataType,
int  dim,
int *  pad,
int *  stride,
int *  upscale,
int  groups,
bool  allow_tf32 
)
inline

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