Trait coaster_nn::RnnConfig

source ·
pub trait RnnConfig<F> {
    // Provided method
    fn workspace_size(&self) -> usize { ... }
}
Expand description

Provides Rnn Config functionality.

Needs to be implemented for Operation specific configurations.

Provided Methods§

source

fn workspace_size(&self) -> usize

Workspace Size - Overwritten by each plugin method except native, which doesn’t require a workspace size.

Implementations on Foreign Types§

source§

impl<T> RnnConfig<T> for RnnConfig
where T: Float + DataTypeInfo,

Implementors§

source§

impl<'a, T> RnnConfig<T> for coaster_nn::frameworks::native::helper::RnnConfig
where T: Add<T, Output = T> + Mul<T, Output = T> + Default + Copy,