Struct melior::dialect::ods::tensor::InsertSliceOperation

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

An insert_slice operation. Insert_slice operation.

The “insert_slice” operation insert a tensor source into another tensor dest as specified by the operation’s offsets, sizes and strides arguments.

It returns a copy of dest with the proper slice updated with the value of source.

The insert_slice operation supports the following arguments:

• source: the tensor that is inserted.
• dest: the tensor into which the source tensor is inserted.
• offsets: tensor-rank number of offsets into the dest tensor into which the slice is inserted.
• sizes: tensor-rank number of sizes which specify the sizes of the source tensor type.
• strides: tensor-rank number of strides that specify subsampling in each dimension.

The representation based on offsets, sizes and strides support a partially-static specification via attributes specified through the static_offsets, static_sizes and static_strides arguments. A special sentinel value ShapedType::kDynamic encodes that the corresponding entry has a dynamic value.

After buffer allocation, the “insert_slice” op is expected to lower into a memref.subview op.

An insert_slice operation may additionally specify insertion into a tensor of higher rank than the source tensor, along dimensions that are statically known to be of size 1. This rank-altering behavior is not required by the op semantics: this flexibility allows to progressively drop unit dimensions while lowering between different flavors of ops on that operate on tensors. The rank-altering behavior of tensor.insert_slice matches the rank-reducing behavior of tensor.extract_slice.

Verification in the rank-reduced case

The same verification discussion and mechanisms apply as for ExtractSliceOp. Unlike ExtractSliceOp however, there is no need for a specific inference.

Example:

// Rank-altering insert_slice.
%1 = tensor.insert_slice %t into %0[0, 0, 0][1, 16, 4][1, 1, 1] :
  tensor<16x4xf32> into tensor<8x16x4xf32>
%3 = tensor.insert_slice %tt into %2[%o0, 4, %o2][1, %sz1, 1][1, %st1, 1] :
  tensor<1x?xf32> into tensor<8x16x4xf32>

Implementations

impl<'c> InsertSliceOperation<'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> ) -> InsertSliceOperationBuilder<'c, Unset, Unset, Unset, 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 offsets(&self) -> Result<impl Iterator<Item = Value<'c, '_>>, Error>

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

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

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

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

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

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

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

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

Trait Implementations

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

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

Converts to this type from the input type.

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