Struct melior::dialect::ods::tensor::UnPackOperation

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

An unpack operation. Tensor unpack operation.

The unpack operation converts a tensor with a tiled and packed layout to a lower-dimensional tensor. Similar to pack, the mandatory attributes inner_dims_pos specifies a permutation for the inner data dimensions, while inner_tiles is the tiling factor. The attribute outer_dims_perm has the exact behavior as the one described in pack. In unpack, it is UB if the tile does not perfectly divide the dimension.

Example NCnc_to_NC:

%0 = tensor.unpack %source inner_dims_pos = [0, 1]
  inner_tiles = [8, 32] into %dest : tensor<16x8x8x32xf32> -> tensor<128x256xf32>

Example CK to KCck:

%0 = tensor.unapck %source outer_dims_perm = [1, 0] inner_dims_pos = [0, 1]
  inner_tiles = [8, 32] into %dest : tensor<8x16x8x32xf32> -> tensor<128x256xf32>

Implementations

impl<'c> UnPackOperation<'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> ) -> UnPackOperationBuilder<'c, Unset, Unset, Unset, Unset, Unset, Unset>

Creates a builder.

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

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

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

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

pub fn outer_dims_perm(&self) -> Result<Attribute<'c>, Error>

pub fn set_outer_dims_perm(&mut self, value: Attribute<'c>)

pub fn remove_outer_dims_perm(&mut self) -> Result<(), Error>

pub fn inner_dims_pos(&self) -> Result<Attribute<'c>, Error>

pub fn set_inner_dims_pos(&mut self, value: Attribute<'c>)

pub fn static_inner_tiles(&self) -> Result<Attribute<'c>, Error>

pub fn set_static_inner_tiles(&mut self, value: Attribute<'c>)

Trait Implementations

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

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

Converts to this type from the input type.

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