at::native::DropoutDescriptor Struct Reference

Inheritance diagram for at::native::DropoutDescriptor:

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:

• aten/src/ATen/cudnn/Descriptors.h