at::native::RNNDescriptor Struct Reference

Inheritance diagram for at::native::RNNDescriptor:

Public Member Functions
void	set (cudnnHandle_t handle, int hidden_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)
Public Member Functions inherited from at::native::Descriptor< cudnnRNNStruct,&cudnnCreateRNNDescriptor,&cudnnDestroyRNNDescriptor >
cudnnRNNStruct *	desc () const
cudnnRNNStruct *	desc ()
cudnnRNNStruct *	desc () const
cudnnRNNStruct *	desc ()
cudnnRNNStruct *	mut_desc ()
cudnnRNNStruct *	mut_desc ()

Data Fields
DropoutDescriptor	dropout_desc_

Additional Inherited Members
Protected Member Functions inherited from at::native::Descriptor< cudnnRNNStruct,&cudnnCreateRNNDescriptor,&cudnnDestroyRNNDescriptor >
void	init ()
void	init ()

Detailed Description

Definition at line 218 of file Descriptors.h.

---

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

• aten/src/ATen/cudnn/Descriptors.h