Enum BaseOperationIr 

pub enum BaseOperationIr {
    Reshape(ShapeOpIr),
    SwapDims(SwapDimsOpIr),
    Permute(PermuteOpIr),
    Flip(FlipOpIr),
    Expand(ShapeOpIr),
    Unfold(UnfoldOpIr),
    Slice(SliceOpIr),
    SliceAssign(SliceAssignOpIr),
    Select(SelectOpIr),
    SelectAssign(SelectAssignOpIr),
    MaskWhere(MaskWhereOpIr),
    MaskFill(MaskFillOpIr),
    Gather(GatherOpIr),
    Scatter(ScatterOpIr),
    ScatterNd(ScatterNdOpIr),
    GatherNd(GatherNdOpIr),
    Equal(BinaryOpIr),
    EqualElem(ScalarOpIr),
    RepeatDim(RepeatDimOpIr),
    Cat(CatOpIr),
    Cast(CastOpIr),
    Empty(CreationOpIr),
    Ones(CreationOpIr),
    Zeros(CreationOpIr),
    NotEqual(BinaryOpIr),
    NotEqualElem(ScalarOpIr),
    All(ReduceOpIr),
    Any(ReduceOpIr),
    AllDim(ReduceDimOpIr),
    AnyDim(ReduceDimOpIr),
}

Basic operations that can be done on any tensor type.

Variants

Reshape(ShapeOpIr)

Operation corresponding to:

Float => reshape. Int => reshape. Bool => reshape.

SwapDims(SwapDimsOpIr)

Operation corresponding to:

Float => swap_dims. Int => swap_dims. Bool => swap_dims.

Permute(PermuteOpIr)

Operation corresponding to:

Float => permute. Int => permute. Bool => permute.

Flip(FlipOpIr)

Operation corresponding to: Float => flip. Int => flip. Bool => flip.

Expand(ShapeOpIr)

Operation corresponding to:

Float => expand. Int => expand. Bool => expand.

Unfold(UnfoldOpIr)

Unfold windows along an axis.

Slice(SliceOpIr)

Operation corresponding to:

Float => slice. Int => slice. Bool => slice.

SliceAssign(SliceAssignOpIr)

Operation corresponding to:

Float => slice assign. Int => slice assign. Bool => slice assign.

Select(SelectOpIr)

Operation corresponding to:

Float => select. Int => select. Bool => select.

SelectAssign(SelectAssignOpIr)

Operation corresponding to:

Float => select assign. Int => select assign. Bool => select assign.

MaskWhere(MaskWhereOpIr)

Operation corresponding to:

Float => mask where. Int => mask where. Bool => mask where.

MaskFill(MaskFillOpIr)

Operation corresponding to:

Float => mask fill. Int => mask fill. Bool => mask fill.

Gather(GatherOpIr)

Operation corresponding to:

Float => gather. Int => gather. Bool => gather.

Scatter(ScatterOpIr)

Operation corresponding to:

Float => scatter. Int => scatter. Bool => scatter.

ScatterNd(ScatterNdOpIr)

Multi-dimensional scatter operation.

GatherNd(GatherNdOpIr)

Multi-dimensional gather operation.

Equal(BinaryOpIr)

Operation corresponding to:

Float => equal. Int => equal. Bool => equal.

EqualElem(ScalarOpIr)

Operation corresponding to:

Float => equal elem. Int => equal elem. Bool => equal elem.

RepeatDim(RepeatDimOpIr)

Operation corresponding to:

Float => repeat dim. Int => repeat dim. Bool => repeat dim.

Cat(CatOpIr)

Operation corresponding to:

Float => cat. Int => cat. Bool => cat.

Cast(CastOpIr)

Cast operation, no direct operation and should be supported by fusion backend.

Empty(CreationOpIr)

Operation corresponding to:

Float => empty. Int => empty. Bool => empty.

Ones(CreationOpIr)

Operation corresponding to:

Float => ones. Int => ones. Bool => ones.

Zeros(CreationOpIr)

Operation corresponding to:

Float => zeros. Int => zeros. Bool => zeros.

NotEqual(BinaryOpIr)

Operation corresponding to:

Float => not_equal. Int => not_equal. Bool => not_equal.

NotEqualElem(ScalarOpIr)

Operation corresponding to:

Float => not_equal_elem. Int => not_equal_elem. Bool => not_equal_elem.

All(ReduceOpIr)

Reduce-all over the input tensor.

Float/Int input is treated as a non-zero check; bool input is the value itself. Output is a single-element bool tensor.

Any(ReduceOpIr)

Reduce-any over the input tensor.

AllDim(ReduceDimOpIr)

Reduce-all along a dim.

AnyDim(ReduceDimOpIr)

Reduce-any along a dim.

Trait Implementations

impl Clone for BaseOperationIr

fn clone(&self) -> BaseOperationIr

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for BaseOperationIr

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for BaseOperationIr

fn deserialize<__D>( __deserializer: __D, ) -> Result<BaseOperationIr, <__D as Deserializer<'de>>::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Hash for BaseOperationIr

fn hash<__H>(&self, state: &mut __H)

where __H: Hasher,

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for BaseOperationIr

fn eq(&self, other: &BaseOperationIr) -> 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 Serialize for BaseOperationIr

fn serialize<__S>( &self, __serializer: __S, ) -> Result<<__S as Serializer>::Ok, <__S as Serializer>::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl StructuralPartialEq for BaseOperationIr