Struct rcudnn::utils::DropoutConfig
source · pub struct DropoutConfig { /* private fields */ }
Expand description
Provides a convenient interface to access cuDNN’s Dropout Descriptor.
You wouldn’t use this struct yourself, but rather obtain it through Cudnn.init_dropout()
.
Implementations§
source§impl DropoutConfig
impl DropoutConfig
sourcepub fn new(
dropout_desc: DropoutDescriptor,
reserve: CudaDeviceMemory
) -> DropoutConfig
pub fn new( dropout_desc: DropoutDescriptor, reserve: CudaDeviceMemory ) -> DropoutConfig
Returns a new DropoutConfig.
sourcepub fn dropout_desc(&self) -> &DropoutDescriptor
pub fn dropout_desc(&self) -> &DropoutDescriptor
Returns dropout_desc
.
sourcepub fn take_mem(self) -> CudaDeviceMemory
pub fn take_mem(self) -> CudaDeviceMemory
Take the Reserve Memory of the DropoutDescriptor
sourcepub fn reserved_space(&self) -> &CudaDeviceMemory
pub fn reserved_space(&self) -> &CudaDeviceMemory
Returns the reserved space.
Trait Implementations§
Auto Trait Implementations§
impl RefUnwindSafe for DropoutConfig
impl !Send for DropoutConfig
impl !Sync for DropoutConfig
impl Unpin for DropoutConfig
impl UnwindSafe for DropoutConfig
Blanket Implementations§
source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more