Struct melior::dialect::ods::shape::ReduceOperation

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

A reduce operation. Returns an expression reduced over a shape or extent tensor.

An operation that takes as input a shape or extent tensor, and a number of initial values. This operation has a region that is applied repeatedly for every extent of the input. Starting with the initial values, the individual extents are then aggregated as defined by the associated region.

Conceptually this op performs the following reduction:

res[] = init;
for (int i = 0, i < shape.rank(); i++) {
  res = reduce(i, shape[i], res[0], ..., res[n]);
}

Where reduce represents the region attached and the result of the reduce op is the last computed output of the reduce region. As an example, the number of elements can be computed as follows:

func.func @reduce(%shape : !shape.shape, %init : !shape.size) ->
    !shape.size {
  %num_elements = shape.reduce(%shape, %init) -> !shape.size  {
    ^bb0(%index: index, %dim: !shape.size, %acc: !shape.size):
      %updated_acc = "shape.mul"(%acc, %dim) :
        (!shape.size, !shape.size) -> !shape.size
      shape.yield %updated_acc : !shape.size
  }
  return %num_elements : !shape.size
}

Implementations

impl<'c> ReduceOperation<'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> ) -> ReduceOperationBuilder<'c, Unset, Unset, Unset, Unset>

Creates a builder.

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

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

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

pub fn region(&self) -> Result<RegionRef<'c, '_>, Error>

Trait Implementations

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

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

Converts to this type from the input type.

impl<'c> TryFrom<Operation<'c>> for ReduceOperation<'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.