Trait coaster_nn::ConvolutionConfig

pub trait ConvolutionConfig<F> {
    fn workspace_size(&self) -> usize { ... }
}

Provides Convolution Config functionality.

Needs to be implemented for Operation specific configurations.

Provided Methods

fn workspace_size(&self) -> usize

Returns the largest workspace size in bytes needed for any of the convolution operations.

Implementors

• impl<'a, T> ConvolutionConfig<T> for coaster_nn::frameworks::native::helper::ConvolutionConfig where
    T: Add<T, Output = T> + Mul<T, Output = T> + Default + Copy, 
• impl<T> ConvolutionConfig<T> for rcudnn::utils::ConvolutionConfig where
    T: Float + DataTypeInfo,