'iree_encoding' Dialect

Tensor encoding attributes and ops.

A dialect defining IREE tensor encoding attributes and related ops, used to implement data-tiling.

• 'iree_encoding' Dialect
  • Operations
    • iree_encoding.set_encoding (Encoding::SetEncodingOp)
    • iree_encoding.unset_encoding (Encoding::UnsetEncodingOp)

Operations

iree_encoding.set_encoding (Encoding::SetEncodingOp)

Perform pack and pad operation on source

Syntax:

operation ::= `iree_encoding.set_encoding` attr-dict $source `:` type($source) `->` type($result)

Operation to assign an encoding to a tensor. The operation does not change the rank or extent of a tensor. Instead it adds an encoding attribute to the tensor type to represent a change in layout.

Traits: AlwaysSpeculatableImplTrait

Interfaces: ConditionallySpeculatable, NoMemoryEffect (MemoryEffectOpInterface), ReifyRankedShapedTypeOpInterface

Effects: MemoryEffects::Effect{}

Operands:

Operand	Description
source	ranked tensor of any type values

Results:

Result	Description
result	ranked tensor of any type values

iree_encoding.unset_encoding (Encoding::UnsetEncodingOp)

Perfom unpack and extract operation on source

Syntax:

operation ::= `iree_encoding.unset_encoding` attr-dict $source `:` type($source) `->` type($result) (`` `{` $result_dims^ `}`)?

Operation to convert an tensor with encoding that represents its data layout into a tensor with default layout (i.e. no encoding). For now in IREE the default layout is row-major.

Traits: AlwaysSpeculatableImplTrait

Interfaces: ConditionallySpeculatable, NoMemoryEffect (MemoryEffectOpInterface), ReifyRankedShapedTypeOpInterface

Effects: MemoryEffects::Effect{}

Operands:

Operand	Description
source	ranked tensor of any type values
result_dims	variadic of index

Results:

Result	Description
result	ranked tensor of any type values