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 AccurateGeLU

impl NonMutableModule for AvgPoolGlobal

impl NonMutableModule for Cos

impl NonMutableModule for Exp

impl NonMutableModule for FastGeLU

impl NonMutableModule for Flatten2D

impl NonMutableModule for GeLU

impl NonMutableModule for Ln

impl NonMutableModule for LogSoftmax

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<C: ConstDim, E: Dtype, D: Device<E>> NonMutableModule for PReLU1D<C, E, D>

impl<E: Dtype> NonMutableModule for LeakyReLU<E>

impl<E: Dtype, D: Device<E>> NonMutableModule for PReLU<E, D>

impl<S: ConstShape> NonMutableModule for Reshape<S>

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

impl<const H: usize, const W: usize, M: UpscaleMethod> NonMutableModule for Upscale2DBy<H, W, M>

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

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

impl<const I: usize, const O: usize, const K: usize, const S: usize, const P: usize, const L: usize, const G: usize, E, D> NonMutableModule for ConvTrans2D<I, O, K, S, P, L, G, E, D>where E: Dtype, D: Storage<E>, Const<{ _ }>: Sized,

impl<const I: usize, const O: usize, const K: usize, const S: usize, const P: usize, const L: usize, const G: usize, E: Dtype, D: Storage<E>> NonMutableModule for Conv2D<I, O, K, S, P, L, G, E, D>where Const<{ _ }>: Sized,

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

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

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

impl<const M: usize, E: Dtype, D: Storage<E>> 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 OH: usize, const OW: usize, M: UpscaleMethod> NonMutableModule for Upscale2D<OH, OW, M>

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