Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Data Fields
at::native::RNNDescriptor Struct Reference
Inheritance diagram for at::native::RNNDescriptor:
at::native::Descriptor< cudnnRNNStruct,&cudnnCreateRNNDescriptor,&cudnnDestroyRNNDescriptor >

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: