Shortcuts

Struct RNNDescriptor

Inheritance Relationships

Base Type

  • public at::native::Descriptor< cudnnRNNStruct,&cudnnCreateRNNDescriptor,&cudnnDestroyRNNDescriptor > (Template Class Descriptor)

Struct Documentation

struct at::native::RNNDescriptor : public at::native::Descriptor<cudnnRNNStruct, &cudnnCreateRNNDescriptor, &cudnnDestroyRNNDescriptor>

Public Functions

void set(cudnnHandle_t handle, int hidden_size, int proj_size, int num_layers, DropoutDescriptor &&dropout_desc, cudnnRNNInputMode_t input_mode, cudnnDirectionMode_t bidirectional, cudnnRNNMode_t mode, cudnnDataType_t datatype, cudnnDataType_t input_type, cudnnRNNAlgo_t algo, bool allow_tf32)

Public Members

DropoutDescriptor dropout_desc_

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources