Struct RNNDescriptor¶
Defined in File Descriptors.h
Page Contents
Inheritance Relationships¶
Base Type¶
public at::native::Descriptor< cudnnRNNStruct, &cudnnCreateRNNDescriptor, &cudnnDestroyRNNDescriptor >(Template Class Descriptor)
Struct Documentation¶
-
struct RNNDescriptor : public at::native::Descriptor<cudnnRNNStruct, &cudnnCreateRNNDescriptor, &cudnnDestroyRNNDescriptor>¶
Public Functions
-
inline 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_¶
-
inline 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)¶
