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)#