HasAutodiffModule

burn::module

Trait HasAutodiffModule 

pub trait HasAutodiffModule<B>
where
    B: AutodiffBackend,
{
    type TrainModule: AutodiffModule<B, InnerModule = Self>;
}
Expand description

Helper trait to associate a module with its autodiff version.

Required Associated Types§

type TrainModule: AutodiffModule<B, InnerModule = Self>

The module with auto-differentiation.

Implementors§

§

impl<B> HasAutodiffModule<B> for Activation<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = Activation<B>

§

impl<B> HasAutodiffModule<B> for Normalization<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = Normalization<B>

§

impl<B> HasAutodiffModule<B> for BinaryCrossEntropyLoss<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = BinaryCrossEntropyLoss<B>

§

impl<B> HasAutodiffModule<B> for CrossEntropyLoss<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = CrossEntropyLoss<B>

§

impl<B> HasAutodiffModule<B> for CrossAttention<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = CrossAttention<B>

§

impl<B> HasAutodiffModule<B> for MultiHeadAttention<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = MultiHeadAttention<B>

§

impl<B> HasAutodiffModule<B> for Conv1d<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = Conv1d<B>

§

impl<B> HasAutodiffModule<B> for Conv2d<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = Conv2d<B>

§

impl<B> HasAutodiffModule<B> for Conv3d<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = Conv3d<B>

§

impl<B> HasAutodiffModule<B> for ConvTranspose1d<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = ConvTranspose1d<B>

§

impl<B> HasAutodiffModule<B> for ConvTranspose2d<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = ConvTranspose2d<B>

§

impl<B> HasAutodiffModule<B> for ConvTranspose3d<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = ConvTranspose3d<B>

§

impl<B> HasAutodiffModule<B> for DeformConv2d<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = DeformConv2d<B>

§

impl<B> HasAutodiffModule<B> for PositionWiseFeedForward<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = PositionWiseFeedForward<B>

§

impl<B> HasAutodiffModule<B> for TransformerDecoder<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = TransformerDecoder<B>

§

impl<B> HasAutodiffModule<B> for TransformerDecoderLayer<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = TransformerDecoderLayer<B>

§

impl<B> HasAutodiffModule<B> for TransformerEncoder<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = TransformerEncoder<B>

§

impl<B> HasAutodiffModule<B> for TransformerEncoderLayer<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = TransformerEncoderLayer<B>

§

impl<B> HasAutodiffModule<B> for BatchNorm<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = BatchNorm<B>

§

impl<B> HasAutodiffModule<B> for BiGru<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = BiGru<B>

§

impl<B> HasAutodiffModule<B> for BiLstm<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = BiLstm<B>

§

impl<B> HasAutodiffModule<B> for BiRnn<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = BiRnn<B>

§

impl<B> HasAutodiffModule<B> for Embedding<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = Embedding<B>

§

impl<B> HasAutodiffModule<B> for GateController<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = GateController<B>

§

impl<B> HasAutodiffModule<B> for GroupNorm<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = GroupNorm<B>

§

impl<B> HasAutodiffModule<B> for Gru<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = Gru<B>

§

impl<B> HasAutodiffModule<B> for InstanceNorm<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = InstanceNorm<B>

§

impl<B> HasAutodiffModule<B> for LayerNorm<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = LayerNorm<B>

§

impl<B> HasAutodiffModule<B> for Linear<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = Linear<B>

§

impl<B> HasAutodiffModule<B> for Lstm<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = Lstm<B>

§

impl<B> HasAutodiffModule<B> for PRelu<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = PRelu<B>

§

impl<B> HasAutodiffModule<B> for PositionalEncoding<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = PositionalEncoding<B>

§

impl<B> HasAutodiffModule<B> for RmsNorm<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = RmsNorm<B>

§

impl<B> HasAutodiffModule<B> for Rnn<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = Rnn<B>

§

impl<B> HasAutodiffModule<B> for RotaryEncoding<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = RotaryEncoding<B>

§

impl<B> HasAutodiffModule<B> for SwiGlu<<B as AutodiffBackend>::InnerBackend>
where B: Backend + AutodiffBackend, <B as AutodiffBackend>::InnerBackend: Backend,

§

type TrainModule = SwiGlu<B>

§

impl<const D: usize, B> HasAutodiffModule<B> for Param<Tensor<<B as AutodiffBackend>::InnerBackend, D>>
where B: AutodiffBackend,

§

type TrainModule = Param<Tensor<B, D>>