Trait coaster_nn::ConvolutionConfig

pub trait ConvolutionConfig<F> {
    // Provided method
    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.

Implementations on Foreign Types

impl<T> ConvolutionConfig<T> for ConvolutionConfig

where T: Float + DataTypeInfo,

fn workspace_size(&self) -> usize

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,