Struct melior::dialect::ods::tensor::ExpandShapeOperation

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

An expand_shape operation. Operation to produce a tensor with a higher rank.

The tensor.expand_shape op produces a new tensor with a higher rank whose sizes are a reassociation of the original src.

A reassociation is defined as a continuous grouping of dimensions and is represented with an array of DenseI64ArrayAttr attribute.

The verification rule is that the reassociation maps are applied to the result tensor with the higher rank to obtain the operand tensor with the smaller rank.

The operand tensor type of a reshape can be zero-ranked if the result tensor type is statically shaped with all dimensions being unit extent. In such cases the reassociation map is empty.

Examples:

// Dimension expansion i -> (i', j') and (k) -> (k')
%b = tensor.expand_shape %a [[0, 1], [2]]
    : tensor<?x?xf32> into tensor<?x?x?xf32>

Implementations

impl<'c> ExpandShapeOperation<'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> ) -> ExpandShapeOperationBuilder<'c, Unset, Unset, Unset>

Creates a builder.

pub fn result(&self) -> Result<OperationResult<'c, '_>, Error>

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

pub fn reassociation(&self) -> Result<ArrayAttribute<'c>, Error>

pub fn set_reassociation(&mut self, value: ArrayAttribute<'c>)

Trait Implementations

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

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

Converts to this type from the input type.

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