Enum ActivationRecord

pub enum ActivationRecord<B>
where
    B: Backend,
{
    Gelu(<Gelu as Module<B>>::Record),
    PRelu(<PRelu<B> as Module<B>>::Record),
    Relu(<Relu as Module<B>>::Record),
    LeakyRelu(<LeakyRelu as Module<B>>::Record),
    SwiGlu(<SwiGlu<B> as Module<B>>::Record),
    Sigmoid(<Sigmoid as Module<B>>::Record),
    Tanh(<Tanh as Module<B>>::Record),
    HardSigmoid(<HardSigmoid as Module<B>>::Record),
}

The record type for the module.

Variants

Gelu(<Gelu as Module<B>>::Record)

The module record associative type.

PRelu(<PRelu<B> as Module<B>>::Record)

The module record associative type.

Relu(<Relu as Module<B>>::Record)

The module record associative type.

LeakyRelu(<LeakyRelu as Module<B>>::Record)

The module record associative type.

SwiGlu(<SwiGlu<B> as Module<B>>::Record)

The module record associative type.

Sigmoid(<Sigmoid as Module<B>>::Record)

The module record associative type.

Tanh(<Tanh as Module<B>>::Record)

The module record associative type.

HardSigmoid(<HardSigmoid as Module<B>>::Record)

The module record associative type.

Trait Implementations

impl<B> Record<B> for ActivationRecord<B>

where B: Backend,

type Item<S: PrecisionSettings> = ActivationRecordItem<B, S>

Type of the item that can be serialized and deserialized.

fn into_item<S>(self) -> <ActivationRecord<B> as Record<B>>::Item<S>

where S: PrecisionSettings,

Convert the current record into the corresponding item that follows the given settings.

fn from_item<S>( item: <ActivationRecord<B> as Record<B>>::Item<S>, device: &<B as Backend>::Device, ) -> ActivationRecord<B>

where S: PrecisionSettings,

Convert the given item into a record.