PyTorch
|
This is the complete list of members for at::native::DropoutDescriptor, including all inherited members.
initialize_rng(cudnnHandle_t handle, float dropout, long long int seed, const TensorOptions &options) | at::native::DropoutDescriptor | inline |
set(cudnnHandle_t handle, float dropout, at::Tensor state_) | at::native::DropoutDescriptor | inline |
set_no_dropout(cudnnHandle_t handle) | at::native::DropoutDescriptor | inline |
state | at::native::DropoutDescriptor |