Struct melior::dialect::ods::linalg::SoftmaxOperation

pub struct SoftmaxOperation<'c> { /* private fields */ }

A softmax operation. Softmax operator.

linalg.softmax computes a numerically stable version of softmax.

For a given input tensor and a specified dimension d, compute:

1. the max m along that dimension d
2. f(x) = exp(x - m)
3. sum f(x) along dimension d to get l(x).
4. compute the final result f(x) / l(x).

This is an aggregate linalg operation that further reduces to a small DAG of structured operations.

Warning: Regarding the tiling capabilities, the implementation doesn’t check that the provided dimensions make sense. This is the responsability of the transformation calling the tiling to ensure that the provided sizes for each dimension make sense with respect to the semantic of softmax.

Implementations

impl<'c> SoftmaxOperation<'c>

pub fn name() -> &'static str

Returns a name.

pub fn as_operation(&self) -> &Operation<'c>

Returns a generic operation.

pub fn builder( context: &'c Context, location: Location<'c> ) -> SoftmaxOperationBuilder<'c, Unset, Unset, Unset, Unset>

Creates a builder.

pub fn result(&self) -> impl Iterator<Item = OperationResult<'c, '_>>

pub fn input(&self) -> Result<Value<'c, '_>, Error>

pub fn output(&self) -> Result<Value<'c, '_>, Error>

pub fn dimension(&self) -> Result<IntegerAttribute<'c>, Error>

pub fn set_dimension(&mut self, value: IntegerAttribute<'c>)

Trait Implementations

impl<'c> From<SoftmaxOperation<'c>> for Operation<'c>

fn from(operation: SoftmaxOperation<'c>) -> Self

Converts to this type from the input type.

impl<'c> TryFrom<Operation<'c>> for SoftmaxOperation<'c>

type Error = Error

The type returned in the event of a conversion error.

fn try_from(operation: Operation<'c>) -> Result<Self, Self::Error>

Performs the conversion.