Trait burn::module::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.

Object Safety

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 + Copy, <T as AutodiffModule<B>>::InnerModule: Copy + Debug, <<T as AutodiffModule<B>>::InnerModule as Module<<B as AutodiffBackend>::InnerBackend>>::Record: Debug, <T as Module<B>>::Record: 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 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>