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.

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,