Enum ScalarPrimsDescriptor 

pub enum ScalarPrimsDescriptor {
    PointwiseUnary {
        op: ScalarUnaryOp,
    },
    PointwiseBinary {
        op: ScalarBinaryOp,
    },
    PointwiseTernary {
        op: ScalarTernaryOp,
    },
    Reduction {
        modes_a: Vec<u32>,
        modes_c: Vec<u32>,
        op: ScalarReductionOp,
    },
}

Descriptor for scalar-pointwise and scalar-reduction planning.

Examples

use tenferro_prims::{ScalarPrimsDescriptor, ScalarUnaryOp};

let desc = ScalarPrimsDescriptor::PointwiseUnary {
    op: ScalarUnaryOp::Reciprocal,
};
assert!(matches!(desc, ScalarPrimsDescriptor::PointwiseUnary { .. }));

Variants

PointwiseUnary

Apply a unary pointwise operation to one input tensor.

Fields

op: ScalarUnaryOp

The unary scalar operation to apply.

PointwiseBinary

Apply a binary pointwise operation to two input tensors.

Fields

op: ScalarBinaryOp

The binary scalar operation to apply.

PointwiseTernary

Apply a ternary pointwise operation to three input tensors.

Fields

op: ScalarTernaryOp

The ternary scalar operation to apply.

Reduction

Reduce one tensor into an output tensor over the dropped modes.

Fields

modes_a: Vec<u32>

Input modes associated with the source tensor.

modes_c: Vec<u32>

Output modes that remain after reduction.

op: ScalarReductionOp

Reduction operator to use.

Trait Implementations

impl Clone for ScalarPrimsDescriptor

fn clone(&self) -> ScalarPrimsDescriptor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ScalarPrimsDescriptor

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

Formats the value using the given formatter.

impl Hash for ScalarPrimsDescriptor

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

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 ScalarPrimsDescriptor

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

impl StructuralPartialEq for ScalarPrimsDescriptor