Struct ExpandShapeOperation 

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

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

The memref.expand_shape op produces a new view with a higher rank whose sizes are a reassociation of the original view. The operation is limited to such reassociations, where a dimension is expanded into one or multiple contiguous dimensions. Such reassociations never require additional allocs or copies.

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

Example:

%r = memref.expand_shape %0 [[0, 1], [2]] output_shape [%sz0, %sz1, 32]
    : memref<?x32xf32> into memref<?x?x32xf32>

If an op can be statically proven to be invalid (e.g, an expansion from memref<10xf32> to memref<2x6xf32>), it is rejected by the verifier. If it cannot statically be proven invalid (e.g., the full example above; it is unclear whether the first source dimension is divisible by 5), the op is accepted by the verifier. However, if the op is in fact invalid at runtime, the behavior is undefined.

The source memref can be zero-ranked. In that case, the reassociation indices must be empty and the result shape may only consist of unit dimensions.

For simplicity, this op may not be used to cast dynamicity of dimension sizes and/or strides. I.e., if and only if a source dimension is dynamic, there must be a dynamic result dimension in the corresponding reassociation group. Same for strides.

The representation for the output shape supports a partially-static specification via attributes specified through the static_output_shape argument. A special sentinel value ShapedType::kDynamic encodes that the corresponding entry has a dynamic value. There must be exactly as many SSA inputs in output_shape as there are ShapedType::kDynamic entries in static_output_shape.

Note: This op currently assumes that the inner strides are of the source/result layout map are the faster-varying ones.

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, Unset, Unset>

Creates a builder.

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

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

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

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

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

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

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

Trait Implementations

impl<'c> Clone for ExpandShapeOperation<'c>

fn clone(&self) -> ExpandShapeOperation<'c>

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<'c> Debug for ExpandShapeOperation<'c>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<'b> From<ExpandShapeOperation<'b>> for MemRefDialectOperation<'b>

fn from(op: ExpandShapeOperation<'b>) -> Self

Converts to this type from the input type.

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

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

Converts to this type from the input type.

impl<'c> PartialEq for ExpandShapeOperation<'c>

fn eq(&self, other: &ExpandShapeOperation<'c>) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

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.

impl<'c> Eq for ExpandShapeOperation<'c>

impl<'c> StructuralPartialEq for ExpandShapeOperation<'c>