Enum burn::tensor::repr::FloatOperationDescription

pub enum FloatOperationDescription {
    Exp(UnaryOperationDescription),
    Log(UnaryOperationDescription),
    Log1p(UnaryOperationDescription),
    Erf(UnaryOperationDescription),
    PowfScalar(ScalarOperationDescription<f32>),
    Sqrt(UnaryOperationDescription),
    Cos(UnaryOperationDescription),
    Sin(UnaryOperationDescription),
    Tanh(UnaryOperationDescription),
    Round(UnaryOperationDescription),
    Floor(UnaryOperationDescription),
    Ceil(UnaryOperationDescription),
    IntoInt(UnaryOperationDescription),
    Matmul(BinaryOperationDescription),
    Random(RandomOperationDescription),
    Recip(UnaryOperationDescription),
    Quantize(QuantizeOperationDescription),
    Dequantize(DequantizeOperationDescription),
}

Operation description specific to a float tensor.

Variants

Exp(UnaryOperationDescription)

Operation corresponding to exp.

Log(UnaryOperationDescription)

Operation corresponding to log.

Log1p(UnaryOperationDescription)

Operation corresponding to log1p.

Erf(UnaryOperationDescription)

Operation corresponding to erf.

PowfScalar(ScalarOperationDescription<f32>)

Operation corresponding to powf_scalar.

Sqrt(UnaryOperationDescription)

Operation corresponding to sqrt.

Cos(UnaryOperationDescription)

Operation corresponding to cos.

Sin(UnaryOperationDescription)

Operation corresponding to sin.

Tanh(UnaryOperationDescription)

Operation corresponding to tanh.

Round(UnaryOperationDescription)

Operation corresponding to round.

Floor(UnaryOperationDescription)

Operation corresponding to floor.

Ceil(UnaryOperationDescription)

Operation corresponding to ceil.

IntoInt(UnaryOperationDescription)

Operation corresponding to into_int.

Matmul(BinaryOperationDescription)

Operation corresponding to matmul.

Random(RandomOperationDescription)

Operation corresponding to random.

Recip(UnaryOperationDescription)

Operation corresponding to recip.

Quantize(QuantizeOperationDescription)

Operation corresponding to quantize.

Dequantize(DequantizeOperationDescription)

Operation corresponding to dequantize.

Trait Implementations

impl Clone for FloatOperationDescription

fn clone(&self) -> FloatOperationDescription

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for FloatOperationDescription

fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for FloatOperationDescription

fn deserialize<__D>( __deserializer: __D, ) -> Result<FloatOperationDescription, <__D as Deserializer<'de>>::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Hash for FloatOperationDescription

fn hash<__H>(&self, state: &mut __H)

where __H: Hasher,

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for FloatOperationDescription

fn eq(&self, other: &FloatOperationDescription) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Serialize for FloatOperationDescription

fn serialize<__S>( &self, __serializer: __S, ) -> Result<<__S as Serializer>::Ok, <__S as Serializer>::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl StructuralPartialEq for FloatOperationDescription