Trait burn::module::ADModule

source · 
pub trait ADModule<B>: Module<B> + Send + Sync + Debugwhere
    B: ADBackend,{
    type InnerModule: Module<<B as ADBackend>::InnerBackend>;

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

Module with auto-differentiation backend.

Required Associated Types§

source

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

Inner module without auto-differentiation.

Required Methods§

source

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

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

Implementations on Foreign Types§

source§

impl<B> ADModule<B> for usizewhere B: ADBackend,

§

type InnerModule = usize

source§

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

source§

impl<B> ADModule<B> for i16where B: ADBackend,

§

type InnerModule = i16

source§

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

source§

impl<T, B> ADModule<B> for Vec<T, Global>where T: ADModule<B> + Debug + Send + Sync + Clone, B: ADBackend,

§

type InnerModule = Vec<<T as ADModule<B>>::InnerModule, Global>

source§

fn valid(&self) -> <Vec<T, Global> as ADModule<B>>::InnerModule

source§

impl<B> ADModule<B> for f32where B: ADBackend,

§

type InnerModule = f32

source§

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

source§

impl<B> ADModule<B> for u32where B: ADBackend,

§

type InnerModule = u32

source§

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

source§

impl<B> ADModule<B> for i8where B: ADBackend,

§

type InnerModule = i8

source§

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

source§

impl<B> ADModule<B> for boolwhere B: ADBackend,

§

type InnerModule = bool

source§

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

source§

impl<B> ADModule<B> for u8where B: ADBackend,

§

type InnerModule = u8

source§

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

source§

impl<B> ADModule<B> for i32where B: ADBackend,

§

type InnerModule = i32

source§

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

source§

impl<const N: usize, T, B> ADModule<B> for [T; N]where T: ADModule<B> + Debug + Send + Sync + Clone + Copy, <T as ADModule<B>>::InnerModule: Copy + Debug, <<T as ADModule<B>>::InnerModule as Module<<B as ADBackend>::InnerBackend>>::Record: Debug, <T as Module<B>>::Record: Debug, B: ADBackend,

§

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

source§

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

source§

impl<T, B> ADModule<B> for Option<T>where T: ADModule<B> + Debug + Send + Sync + Clone, B: ADBackend,

§

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

source§

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

source§

impl<B> ADModule<B> for u64where B: ADBackend,

§

type InnerModule = u64

source§

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

source§

impl<B> ADModule<B> for Stringwhere B: ADBackend,

§

type InnerModule = String

source§

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

source§

impl<B> ADModule<B> for i64where B: ADBackend,

§

type InnerModule = i64

source§

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

source§

impl<B> ADModule<B> for f64where B: ADBackend,

§

type InnerModule = f64

source§

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

source§

impl<B> ADModule<B> for u16where B: ADBackend,

§

type InnerModule = u16

source§

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

source§

impl<B> ADModule<B> for PhantomData<B>where B: ADBackend,

§

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

source§

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

Implementors§

source§

impl<B> ADModule<B> for PaddingConfig1dwhere B: ADBackend,

§

type InnerModule = PaddingConfig1d

source§

impl<B> ADModule<B> for PaddingConfig2dwhere B: ADBackend,

§

type InnerModule = PaddingConfig2d

source§

impl<B> ADModule<B> for MultiHeadAttention<B>where B: Backend + ADBackend, <B as ADBackend>::InnerBackend: Backend,

§

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

source§

impl<B> ADModule<B> for Conv1d<B>where B: Backend + ADBackend, <B as ADBackend>::InnerBackend: Backend,

§

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

source§

impl<B> ADModule<B> for Conv2d<B>where B: Backend + ADBackend, <B as ADBackend>::InnerBackend: Backend,

§

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

source§

impl<B> ADModule<B> for ConvTranspose1d<B>where B: Backend + ADBackend, <B as ADBackend>::InnerBackend: Backend,

§

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

source§

impl<B> ADModule<B> for ConvTranspose2d<B>where B: Backend + ADBackend, <B as ADBackend>::InnerBackend: Backend,

§

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

source§

impl<B> ADModule<B> for Gru<B>where B: Backend + ADBackend, <B as ADBackend>::InnerBackend: Backend,

§

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

source§

impl<B> ADModule<B> for BinaryCrossEntropyLoss<B>where B: Backend + ADBackend, <B as ADBackend>::InnerBackend: Backend,

§

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

source§

impl<B> ADModule<B> for CrossEntropyLoss<B>where B: Backend + ADBackend, <B as ADBackend>::InnerBackend: Backend,

§

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

source§

impl<B> ADModule<B> for AdaptiveAvgPool1dwhere B: ADBackend,

§

type InnerModule = AdaptiveAvgPool1d

source§

impl<B> ADModule<B> for AdaptiveAvgPool2dwhere B: ADBackend,

§

type InnerModule = AdaptiveAvgPool2d

source§

impl<B> ADModule<B> for AvgPool1dwhere B: ADBackend,

§

type InnerModule = AvgPool1d

source§

impl<B> ADModule<B> for AvgPool2dwhere B: ADBackend,

§

type InnerModule = AvgPool2d

source§

impl<B> ADModule<B> for MaxPool1dwhere B: ADBackend,

§

type InnerModule = MaxPool1d

source§

impl<B> ADModule<B> for MaxPool2dwhere B: ADBackend,

§

type InnerModule = MaxPool2d

source§

impl<B> ADModule<B> for Dropoutwhere B: ADBackend,

§

type InnerModule = Dropout

source§

impl<B> ADModule<B> for Embedding<B>where B: Backend + ADBackend, <B as ADBackend>::InnerBackend: Backend,

§

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

source§

impl<B> ADModule<B> for GELUwhere B: ADBackend,

§

type InnerModule = GELU

source§

impl<B> ADModule<B> for GateController<B>where B: Backend + ADBackend, <B as ADBackend>::InnerBackend: Backend,

§

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

source§

impl<B> ADModule<B> for LayerNorm<B>where B: Backend + ADBackend, <B as ADBackend>::InnerBackend: Backend,

§

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

source§

impl<B> ADModule<B> for Linear<B>where B: Backend + ADBackend, <B as ADBackend>::InnerBackend: Backend,

§

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

source§

impl<B> ADModule<B> for Lstm<B>where B: Backend + ADBackend, <B as ADBackend>::InnerBackend: Backend,

§

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

source§

impl<B> ADModule<B> for PositionalEncoding<B>where B: Backend + ADBackend, <B as ADBackend>::InnerBackend: Backend,

§

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

source§

impl<B> ADModule<B> for ReLUwhere B: ADBackend,

§

type InnerModule = ReLU

source§

impl<B> ADModule<B> for PositionWiseFeedForward<B>where B: Backend + ADBackend, <B as ADBackend>::InnerBackend: Backend,

§

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

source§

impl<B> ADModule<B> for TransformerDecoder<B>where B: Backend + ADBackend, <B as ADBackend>::InnerBackend: Backend,

§

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

source§

impl<B> ADModule<B> for TransformerDecoderLayer<B>where B: Backend + ADBackend, <B as ADBackend>::InnerBackend: Backend,

§

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

source§

impl<B> ADModule<B> for TransformerEncoder<B>where B: Backend + ADBackend, <B as ADBackend>::InnerBackend: Backend,

§

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

source§

impl<B> ADModule<B> for TransformerEncoderLayer<B>where B: Backend + ADBackend, <B as ADBackend>::InnerBackend: Backend,

§

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

source§

impl<B> ADModule<B> for bf16where B: ADBackend,

§

type InnerModule = bf16

source§

impl<B> ADModule<B> for f16where B: ADBackend,

§

type InnerModule = f16

source§

impl<B, const D: usize> ADModule<B> for BatchNorm<B, D>where B: Backend + ADBackend, <B as ADBackend>::InnerBackend: Backend,

§

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

source§

impl<const D: usize, B> ADModule<B> for Tensor<B, D, Float>where B: ADBackend,

§

type InnerModule = Tensor<<B as ADBackend>::InnerBackend, D, Float>

source§

impl<const D: usize, B> ADModule<B> for Param<Tensor<B, D, Float>>where B: ADBackend,

§

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

source§

impl<const D: usize, B> ADModule<B> for RunningState<Tensor<B, D, Float>>where B: ADBackend,

§

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