Method create_dropout_descriptor

crates/cudnn/src/context.rs:514–539

Creates and initializes a generic dropout descriptor. # Arguments `dropout` - probability with which the value from input is set to zero during the dropout layer. `states` - user-allocated GPU memory that will hold random number generator states. `seed` - seed used to initialize random number generator states. Do note** that the exact amount of memory can be obtained with `get_dropout_states_

(
        &self,
        dropout: f32,
        states: T,
        seed: u64,
    )

    /// # }
    /// ```
    pub fn create_dropout_descriptor<T: GpuBuffer<u8>>(
        &self,
        dropout: f32,
        states: T,
        seed: u64,
    ) -> Result<DropoutDescriptor<T>, CudnnError> {
        let mut raw = MaybeUninit::uninit();

        unsafe {
            sys::cudnnCreateDropoutDescriptor(raw.as_mut_ptr()).into_result()?;

            let mut raw = raw.assume_init();

            sys::cudnnSetDropoutDescriptor(
                raw,
                self.raw,
                dropout,
                states.as_device_ptr().as_ptr() as *mut std::ffi::c_void,
                states.len(),
                seed,
            )
            .into_result()?;

            Ok(DropoutDescriptor::new(raw, states))
        }
    }

    /// This function serves as a heuristic for obtaining the best suited algorithm for
    /// `convolution_forward()` for the given layer specifications.