Struct rcudnn::RnnDescriptor [−][src]
pub struct RnnDescriptor { /* fields omitted */ }
Expand description
Describes a Recurrent Descriptor.
Implementations
pub fn new(
handle: &Cudnn,
hidden_size: i32,
num_layers: i32,
dropout_config: DropoutConfig,
input_mode: cudnnRNNInputMode_t,
direction: cudnnDirectionMode_t,
mode: cudnnRNNMode_t,
algorithm: cudnnRNNAlgo_t,
data_type: DataType,
padding_mode: cudnnRNNPaddingMode_t
) -> Result<RnnDescriptor, Error>
pub fn new(
handle: &Cudnn,
hidden_size: i32,
num_layers: i32,
dropout_config: DropoutConfig,
input_mode: cudnnRNNInputMode_t,
direction: cudnnDirectionMode_t,
mode: cudnnRNNMode_t,
algorithm: cudnnRNNAlgo_t,
data_type: DataType,
padding_mode: cudnnRNNPaddingMode_t
) -> Result<RnnDescriptor, Error>
Initializes a new CUDA cuDNN RnnDescriptor.
Initializes a new CUDA cuDNN RnnDescriptor from its C type.
Returns the CUDA cuDNN RnnDescriptor as its C type.