Trait AutodiffModule

pub trait AutodiffModule<B>:
    Module<B>
    + Send
    + Debug
where
    B: AutodiffBackend,
{
    type InnerModule: Module<<B as AutodiffBackend>::InnerBackend>;

    // Required method
    fn valid(&self) -> Self::InnerModule;
}

Module with auto-differentiation backend.

Required Associated Types

type InnerModule: Module<<B as AutodiffBackend>::InnerBackend>

Inner module without auto-differentiation.

Required Methods

fn valid(&self) -> Self::InnerModule

Get the same module, but on the inner backend without auto-differentiation.

Dyn Compatibility

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementations on Foreign Types

impl<B> AutodiffModule<B> for bool

where B: AutodiffBackend,

type InnerModule = bool

fn valid(&self) -> <bool as AutodiffModule<B>>::InnerModule

impl<B> AutodiffModule<B> for f32

where B: AutodiffBackend,

type InnerModule = f32

fn valid(&self) -> <f32 as AutodiffModule<B>>::InnerModule

impl<B> AutodiffModule<B> for f64

where B: AutodiffBackend,

type InnerModule = f64

fn valid(&self) -> <f64 as AutodiffModule<B>>::InnerModule

impl<B> AutodiffModule<B> for i8

where B: AutodiffBackend,

type InnerModule = i8

fn valid(&self) -> <i8 as AutodiffModule<B>>::InnerModule

impl<B> AutodiffModule<B> for i16

where B: AutodiffBackend,

type InnerModule = i16

fn valid(&self) -> <i16 as AutodiffModule<B>>::InnerModule

impl<B> AutodiffModule<B> for i32

where B: AutodiffBackend,

type InnerModule = i32

fn valid(&self) -> <i32 as AutodiffModule<B>>::InnerModule

impl<B> AutodiffModule<B> for i64

where B: AutodiffBackend,

type InnerModule = i64

fn valid(&self) -> <i64 as AutodiffModule<B>>::InnerModule

impl<B> AutodiffModule<B> for u8

where B: AutodiffBackend,

type InnerModule = u8

fn valid(&self) -> <u8 as AutodiffModule<B>>::InnerModule

impl<B> AutodiffModule<B> for u16

where B: AutodiffBackend,

type InnerModule = u16

fn valid(&self) -> <u16 as AutodiffModule<B>>::InnerModule

impl<B> AutodiffModule<B> for u32

where B: AutodiffBackend,

type InnerModule = u32

fn valid(&self) -> <u32 as AutodiffModule<B>>::InnerModule

impl<B> AutodiffModule<B> for u64

where B: AutodiffBackend,

type InnerModule = u64

fn valid(&self) -> <u64 as AutodiffModule<B>>::InnerModule

impl<B> AutodiffModule<B> for usize

where B: AutodiffBackend,

type InnerModule = usize

fn valid(&self) -> <usize as AutodiffModule<B>>::InnerModule

impl<B> AutodiffModule<B> for String

where B: AutodiffBackend,

type InnerModule = String

fn valid(&self) -> <String as AutodiffModule<B>>::InnerModule

impl<B> AutodiffModule<B> for PhantomData<B>

where B: AutodiffBackend,

type InnerModule = PhantomData<<B as AutodiffBackend>::InnerBackend>

fn valid(&self) -> <PhantomData<B> as AutodiffModule<B>>::InnerModule

impl<B, L0, L1> AutodiffModule<B> for (L0, L1)

where B: AutodiffBackend, L0: AutodiffModule<B> + Debug + Send + Clone, L1: AutodiffModule<B> + Debug + Send + Clone,

type InnerModule = (<L0 as AutodiffModule<B>>::InnerModule, <L1 as AutodiffModule<B>>::InnerModule)

fn valid(&self) -> <(L0, L1) as AutodiffModule<B>>::InnerModule

impl<B, L0, L1, L2> AutodiffModule<B> for (L0, L1, L2)

where B: AutodiffBackend, L0: AutodiffModule<B> + Debug + Send + Clone, L1: AutodiffModule<B> + Debug + Send + Clone, L2: AutodiffModule<B> + Debug + Send + Clone,

type InnerModule = (<L0 as AutodiffModule<B>>::InnerModule, <L1 as AutodiffModule<B>>::InnerModule, <L2 as AutodiffModule<B>>::InnerModule)

fn valid(&self) -> <(L0, L1, L2) as AutodiffModule<B>>::InnerModule

impl<B, L0, L1, L2, L3> AutodiffModule<B> for (L0, L1, L2, L3)

where B: AutodiffBackend, L0: AutodiffModule<B> + Debug + Send + Clone, L1: AutodiffModule<B> + Debug + Send + Clone, L2: AutodiffModule<B> + Debug + Send + Clone, L3: AutodiffModule<B> + Debug + Send + Clone,

type InnerModule = (<L0 as AutodiffModule<B>>::InnerModule, <L1 as AutodiffModule<B>>::InnerModule, <L2 as AutodiffModule<B>>::InnerModule, <L3 as AutodiffModule<B>>::InnerModule)

fn valid(&self) -> <(L0, L1, L2, L3) as AutodiffModule<B>>::InnerModule

impl<B, L0, L1, L2, L3, L4> AutodiffModule<B> for (L0, L1, L2, L3, L4)

where B: AutodiffBackend, L0: AutodiffModule<B> + Debug + Send + Clone, L1: AutodiffModule<B> + Debug + Send + Clone, L2: AutodiffModule<B> + Debug + Send + Clone, L3: AutodiffModule<B> + Debug + Send + Clone, L4: AutodiffModule<B> + Debug + Send + Clone,

type InnerModule = (<L0 as AutodiffModule<B>>::InnerModule, <L1 as AutodiffModule<B>>::InnerModule, <L2 as AutodiffModule<B>>::InnerModule, <L3 as AutodiffModule<B>>::InnerModule, <L4 as AutodiffModule<B>>::InnerModule)

fn valid(&self) -> <(L0, L1, L2, L3, L4) as AutodiffModule<B>>::InnerModule

impl<B, L0, L1, L2, L3, L4, L5> AutodiffModule<B> for (L0, L1, L2, L3, L4, L5)

where B: AutodiffBackend, L0: AutodiffModule<B> + Debug + Send + Clone, L1: AutodiffModule<B> + Debug + Send + Clone, L2: AutodiffModule<B> + Debug + Send + Clone, L3: AutodiffModule<B> + Debug + Send + Clone, L4: AutodiffModule<B> + Debug + Send + Clone, L5: AutodiffModule<B> + Debug + Send + Clone,

type InnerModule = (<L0 as AutodiffModule<B>>::InnerModule, <L1 as AutodiffModule<B>>::InnerModule, <L2 as AutodiffModule<B>>::InnerModule, <L3 as AutodiffModule<B>>::InnerModule, <L4 as AutodiffModule<B>>::InnerModule, <L5 as AutodiffModule<B>>::InnerModule)

fn valid(&self) -> <(L0, L1, L2, L3, L4, L5) as AutodiffModule<B>>::InnerModule

impl<B, L0, L1, L2, L3, L4, L5, L6> AutodiffModule<B> for (L0, L1, L2, L3, L4, L5, L6)

where B: AutodiffBackend, L0: AutodiffModule<B> + Debug + Send + Clone, L1: AutodiffModule<B> + Debug + Send + Clone, L2: AutodiffModule<B> + Debug + Send + Clone, L3: AutodiffModule<B> + Debug + Send + Clone, L4: AutodiffModule<B> + Debug + Send + Clone, L5: AutodiffModule<B> + Debug + Send + Clone, L6: AutodiffModule<B> + Debug + Send + Clone,

type InnerModule = (<L0 as AutodiffModule<B>>::InnerModule, <L1 as AutodiffModule<B>>::InnerModule, <L2 as AutodiffModule<B>>::InnerModule, <L3 as AutodiffModule<B>>::InnerModule, <L4 as AutodiffModule<B>>::InnerModule, <L5 as AutodiffModule<B>>::InnerModule, <L6 as AutodiffModule<B>>::InnerModule)

fn valid( &self, ) -> <(L0, L1, L2, L3, L4, L5, L6) as AutodiffModule<B>>::InnerModule

impl<B, L0, L1, L2, L3, L4, L5, L6, L7> AutodiffModule<B> for (L0, L1, L2, L3, L4, L5, L6, L7)

where B: AutodiffBackend, L0: AutodiffModule<B> + Debug + Send + Clone, L1: AutodiffModule<B> + Debug + Send + Clone, L2: AutodiffModule<B> + Debug + Send + Clone, L3: AutodiffModule<B> + Debug + Send + Clone, L4: AutodiffModule<B> + Debug + Send + Clone, L5: AutodiffModule<B> + Debug + Send + Clone, L6: AutodiffModule<B> + Debug + Send + Clone, L7: AutodiffModule<B> + Debug + Send + Clone,

type InnerModule = (<L0 as AutodiffModule<B>>::InnerModule, <L1 as AutodiffModule<B>>::InnerModule, <L2 as AutodiffModule<B>>::InnerModule, <L3 as AutodiffModule<B>>::InnerModule, <L4 as AutodiffModule<B>>::InnerModule, <L5 as AutodiffModule<B>>::InnerModule, <L6 as AutodiffModule<B>>::InnerModule, <L7 as AutodiffModule<B>>::InnerModule)

fn valid( &self, ) -> <(L0, L1, L2, L3, L4, L5, L6, L7) as AutodiffModule<B>>::InnerModule

impl<B, L0, L1, L2, L3, L4, L5, L6, L7, L8> AutodiffModule<B> for (L0, L1, L2, L3, L4, L5, L6, L7, L8)

where B: AutodiffBackend, L0: AutodiffModule<B> + Debug + Send + Clone, L1: AutodiffModule<B> + Debug + Send + Clone, L2: AutodiffModule<B> + Debug + Send + Clone, L3: AutodiffModule<B> + Debug + Send + Clone, L4: AutodiffModule<B> + Debug + Send + Clone, L5: AutodiffModule<B> + Debug + Send + Clone, L6: AutodiffModule<B> + Debug + Send + Clone, L7: AutodiffModule<B> + Debug + Send + Clone, L8: AutodiffModule<B> + Debug + Send + Clone,

type InnerModule = (<L0 as AutodiffModule<B>>::InnerModule, <L1 as AutodiffModule<B>>::InnerModule, <L2 as AutodiffModule<B>>::InnerModule, <L3 as AutodiffModule<B>>::InnerModule, <L4 as AutodiffModule<B>>::InnerModule, <L5 as AutodiffModule<B>>::InnerModule, <L6 as AutodiffModule<B>>::InnerModule, <L7 as AutodiffModule<B>>::InnerModule, <L8 as AutodiffModule<B>>::InnerModule)

fn valid( &self, ) -> <(L0, L1, L2, L3, L4, L5, L6, L7, L8) as AutodiffModule<B>>::InnerModule

impl<B, L0, L1, L2, L3, L4, L5, L6, L7, L8, L9> AutodiffModule<B> for (L0, L1, L2, L3, L4, L5, L6, L7, L8, L9)

where B: AutodiffBackend, L0: AutodiffModule<B> + Debug + Send + Clone, L1: AutodiffModule<B> + Debug + Send + Clone, L2: AutodiffModule<B> + Debug + Send + Clone, L3: AutodiffModule<B> + Debug + Send + Clone, L4: AutodiffModule<B> + Debug + Send + Clone, L5: AutodiffModule<B> + Debug + Send + Clone, L6: AutodiffModule<B> + Debug + Send + Clone, L7: AutodiffModule<B> + Debug + Send + Clone, L8: AutodiffModule<B> + Debug + Send + Clone, L9: AutodiffModule<B> + Debug + Send + Clone,

type InnerModule = (<L0 as AutodiffModule<B>>::InnerModule, <L1 as AutodiffModule<B>>::InnerModule, <L2 as AutodiffModule<B>>::InnerModule, <L3 as AutodiffModule<B>>::InnerModule, <L4 as AutodiffModule<B>>::InnerModule, <L5 as AutodiffModule<B>>::InnerModule, <L6 as AutodiffModule<B>>::InnerModule, <L7 as AutodiffModule<B>>::InnerModule, <L8 as AutodiffModule<B>>::InnerModule, <L9 as AutodiffModule<B>>::InnerModule)

fn valid( &self, ) -> <(L0, L1, L2, L3, L4, L5, L6, L7, L8, L9) as AutodiffModule<B>>::InnerModule

impl<T, B> AutodiffModule<B> for Option<T>

where T: AutodiffModule<B> + Debug + Send + Clone, B: AutodiffBackend,

type InnerModule = Option<<T as AutodiffModule<B>>::InnerModule>

fn valid(&self) -> <Option<T> as AutodiffModule<B>>::InnerModule

impl<T, B> AutodiffModule<B> for Vec<T>

where T: AutodiffModule<B> + Debug + Send + Clone, B: AutodiffBackend,

type InnerModule = Vec<<T as AutodiffModule<B>>::InnerModule>

fn valid(&self) -> <Vec<T> as AutodiffModule<B>>::InnerModule

impl<const N: usize, T, B> AutodiffModule<B> for [T; N]

where T: AutodiffModule<B> + Debug + Send + Clone, <T as AutodiffModule<B>>::InnerModule: Debug, B: AutodiffBackend,

type InnerModule = [<T as AutodiffModule<B>>::InnerModule; N]

fn valid(&self) -> <[T; N] as AutodiffModule<B>>::InnerModule

Implementors

impl<B> AutodiffModule<B> for MultiHeadAttention<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = MultiHeadAttention<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for Conv1d<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = Conv1d<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for Conv2d<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = Conv2d<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for Conv3d<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = Conv3d<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for ConvTranspose1d<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = ConvTranspose1d<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for ConvTranspose2d<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = ConvTranspose2d<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for ConvTranspose3d<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = ConvTranspose3d<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for DeformConv2d<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = DeformConv2d<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for Gru<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = Gru<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for Interpolate1d

where B: AutodiffBackend,

type InnerModule = Interpolate1d

impl<B> AutodiffModule<B> for Interpolate2d

where B: AutodiffBackend,

type InnerModule = Interpolate2d

impl<B> AutodiffModule<B> for BinaryCrossEntropyLoss<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = BinaryCrossEntropyLoss<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for CrossEntropyLoss<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = CrossEntropyLoss<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for HuberLoss

where B: AutodiffBackend,

type InnerModule = HuberLoss

impl<B> AutodiffModule<B> for MseLoss

where B: AutodiffBackend,

type InnerModule = MseLoss

impl<B> AutodiffModule<B> for AdaptiveAvgPool1d

where B: AutodiffBackend,

type InnerModule = AdaptiveAvgPool1d

impl<B> AutodiffModule<B> for AdaptiveAvgPool2d

where B: AutodiffBackend,

type InnerModule = AdaptiveAvgPool2d

impl<B> AutodiffModule<B> for AvgPool1d

where B: AutodiffBackend,

type InnerModule = AvgPool1d

impl<B> AutodiffModule<B> for AvgPool2d

where B: AutodiffBackend,

type InnerModule = AvgPool2d

impl<B> AutodiffModule<B> for MaxPool1d

where B: AutodiffBackend,

type InnerModule = MaxPool1d

impl<B> AutodiffModule<B> for MaxPool2d

where B: AutodiffBackend,

type InnerModule = MaxPool2d

impl<B> AutodiffModule<B> for BiLstm<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = BiLstm<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for Dropout

where B: AutodiffBackend,

type InnerModule = Dropout

impl<B> AutodiffModule<B> for Embedding<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = Embedding<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for GateController<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = GateController<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for Gelu

where B: AutodiffBackend,

type InnerModule = Gelu

impl<B> AutodiffModule<B> for GroupNorm<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = GroupNorm<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for HardSigmoid

where B: AutodiffBackend,

type InnerModule = HardSigmoid

impl<B> AutodiffModule<B> for InstanceNorm<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = InstanceNorm<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for LayerNorm<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = LayerNorm<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for LeakyRelu

where B: AutodiffBackend,

type InnerModule = LeakyRelu

impl<B> AutodiffModule<B> for Linear<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = Linear<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for Lstm<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = Lstm<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for PRelu<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = PRelu<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for PositionalEncoding<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = PositionalEncoding<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for Relu

where B: AutodiffBackend,

type InnerModule = Relu

impl<B> AutodiffModule<B> for RmsNorm<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = RmsNorm<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for RotaryEncoding<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = RotaryEncoding<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for Sigmoid

where B: AutodiffBackend,

type InnerModule = Sigmoid

impl<B> AutodiffModule<B> for SwiGlu<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = SwiGlu<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for Tanh

where B: AutodiffBackend,

type InnerModule = Tanh

impl<B> AutodiffModule<B> for Unfold4d

where B: AutodiffBackend,

type InnerModule = Unfold4d

impl<B> AutodiffModule<B> for PositionWiseFeedForward<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = PositionWiseFeedForward<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for TransformerDecoder<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = TransformerDecoder<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for TransformerDecoderLayer<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = TransformerDecoderLayer<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for TransformerEncoder<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = TransformerEncoder<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for TransformerEncoderLayer<B>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = TransformerEncoderLayer<<B as AutodiffBackend>::InnerBackend>

impl<B> AutodiffModule<B> for bf16

where B: AutodiffBackend,

type InnerModule = bf16

impl<B> AutodiffModule<B> for f16

where B: AutodiffBackend,

type InnerModule = f16

impl<B, T> AutodiffModule<B> for Ignored<T>

where B: AutodiffBackend, T: Sync + Send + Debug + Clone,

type InnerModule = Ignored<T>

impl<B, const D: usize> AutodiffModule<B> for BatchNorm<B, D>

where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

type InnerModule = BatchNorm<<B as AutodiffBackend>::InnerBackend, D>

impl<const D: usize, B> AutodiffModule<B> for Param<Tensor<B, D>>

where B: AutodiffBackend,

type InnerModule = Param<Tensor<<B as AutodiffBackend>::InnerBackend, D>>

impl<const D: usize, B> AutodiffModule<B> for Param<Tensor<B, D, Bool>>

where B: AutodiffBackend,

type InnerModule = Param<Tensor<<B as AutodiffBackend>::InnerBackend, D, Bool>>

impl<const D: usize, B> AutodiffModule<B> for Param<Tensor<B, D, Int>>

where B: AutodiffBackend,

type InnerModule = Param<Tensor<<B as AutodiffBackend>::InnerBackend, D, Int>>

impl<const D: usize, B> AutodiffModule<B> for RunningState<Tensor<B, D>>

where B: AutodiffBackend,

type InnerModule = RunningState<Tensor<<B as AutodiffBackend>::InnerBackend, D>>

impl<const D: usize, B, K> AutodiffModule<B> for Tensor<B, D, K>

where B: AutodiffBackend, K: BasicAutodiffOps<B>,

type InnerModule = Tensor<<B as AutodiffBackend>::InnerBackend, D, <K as BasicAutodiffOps<B>>::InnerKind>