Trait dfdx::nn::NonMutableModule

pub trait NonMutableModule { }

Marker trait for modules that don’t have different behavior between mutable forwards and non-mutable forwards

Implementors

impl NonMutableModule for Abs

impl NonMutableModule for AvgPoolGlobal

impl NonMutableModule for Cos

impl NonMutableModule for Exp

impl NonMutableModule for Flatten2D

impl NonMutableModule for GeLU

impl NonMutableModule for Ln

impl NonMutableModule for MaxPoolGlobal

impl NonMutableModule for MinPoolGlobal

impl NonMutableModule for ReLU

impl NonMutableModule for Sigmoid

impl NonMutableModule for Sin

impl NonMutableModule for Softmax

impl NonMutableModule for Sqrt

impl NonMutableModule for Square

impl NonMutableModule for Tanh

impl<const C: usize, E: Dtype, D: DeviceStorage> NonMutableModule for Bias2D<C, E, D>

impl<const I: usize, const O: usize, E: Dtype, D: DeviceStorage> NonMutableModule for Linear<I, O, E, D>

impl<const I: usize, const O: usize, E: Dtype, D: DeviceStorage> NonMutableModule for UnbiasedLinear<I, O, E, D>

impl<const I: usize, const O: usize, const K: usize, const S: usize, const P: usize, E, D> NonMutableModule for Conv2D<I, O, K, S, P, E, D>where E: Dtype, D: DeviceStorage,

impl<const K: usize, const S: usize, const P: usize> NonMutableModule for AvgPool2D<K, S, P>

impl<const K: usize, const S: usize, const P: usize> NonMutableModule for MaxPool2D<K, S, P>

impl<const K: usize, const S: usize, const P: usize> NonMutableModule for MinPool2D<K, S, P>

impl<const M: usize, E: Dtype, D: DeviceStorage> NonMutableModule for LayerNorm1D<M, E, D>

impl<const M: usize, const H: usize, const A: usize, const B: usize, const F: usize, E, D> NonMutableModule for Transformer<M, H, A, B, F, E, D>where E: Dtype, D: Device<E>,

impl<const M: usize, const H: usize, const F: usize, E: Dtype, D: Device<E>> NonMutableModule for TransformerDecoderBlock<M, H, F, E, D>

impl<const M: usize, const H: usize, const F: usize, E: Dtype, D: Device<E>> NonMutableModule for TransformerEncoderBlock<M, H, F, E, D>

impl<const M: usize, const H: usize, const F: usize, const L: usize, E: Dtype, D: Device<E>> NonMutableModule for TransformerDecoder<M, H, F, L, E, D>

impl<const M: usize, const H: usize, const K: usize, const V: usize, E: Dtype, D: Device<E>> NonMutableModule for MultiHeadAttention<M, H, K, V, E, D>

impl<const V: usize, const M: usize, E: Dtype, D: DeviceStorage> NonMutableModule for Embedding<V, M, E, D>