at::native::DropoutDescriptor Struct Reference

#include <Descriptors.h>

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 Attributes
at::Tensor	state

Member Function Documentation

initialize_rng()

void at::native::DropoutDescriptor::initialize_rng ( cudnnHandle_t  handle, float  dropout, long long int  seed, const TensorOptions &  options  )

inline

set()

void at::native::DropoutDescriptor::set ( cudnnHandle_t  handle, float  dropout, at::Tensor  state_  )

inline

set_no_dropout()

void at::native::DropoutDescriptor::set_no_dropout ( cudnnHandle_t  handle )

inline

Member Data Documentation

state

at::Tensor at::native::DropoutDescriptor::state

---

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

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