Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Data Fields
at::native::DropoutDescriptor Struct Reference
Inheritance diagram for at::native::DropoutDescriptor:
at::native::Descriptor< cudnnDropoutStruct,&cudnnCreateDropoutDescriptor,&cudnnDestroyDropoutDescriptor >

Public Member Functions

void initialize_rng (cudnnHandle_t handle, float dropout, long long int seed, const TensorOptions &options)
 
void set (cudnnHandle_t handle, float dropout, at::Tensor state_)
 
void set_no_dropout (cudnnHandle_t handle)
 
- Public Member Functions inherited from at::native::Descriptor< cudnnDropoutStruct,&cudnnCreateDropoutDescriptor,&cudnnDestroyDropoutDescriptor >
cudnnDropoutStruct * desc () const
 
cudnnDropoutStruct * desc ()
 
cudnnDropoutStruct * desc () const
 
cudnnDropoutStruct * desc ()
 
cudnnDropoutStruct * mut_desc ()
 
cudnnDropoutStruct * mut_desc ()
 

Data Fields

at::Tensor state
 

Additional Inherited Members

- Protected Member Functions inherited from at::native::Descriptor< cudnnDropoutStruct,&cudnnCreateDropoutDescriptor,&cudnnDestroyDropoutDescriptor >
void init ()
 
void init ()
 

Detailed Description

Definition at line 176 of file Descriptors.h.


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