Enum ElementwiseOp 

pub enum ElementwiseOp {
    Abs,
    Ceil,
    Cos,
    Erf,
    Exp,
    Floor,
    Log,
    Negate,
    Reciprocal,
    Rsqrt,
    Sin,
    Sigmoid,
    Tanh,
    Clamp(NanMode),
    Add,
    Sub,
    Mul,
    Pow,
    Maximum(NanMode),
    Minimum(NanMode),
    Equal,
    Greater,
    GreaterEqual,
    LogicalAnd,
    LogicalOr,
    LogicalXor,
    LogicalNot,
    Select,
    CopyBytes,
}

Elementwise operator lanes: the scalar function one invocation applies per output element.

Variants

Abs

Ceil

Cos

Erf

Exp

Floor

Log

Negate

Reciprocal

Rsqrt

Sin

Sigmoid

Tanh

Clamp(NanMode)

apply_min(apply_max(x, lo), hi); the bounds arrive as two trailing specialization constants holding FP32 bit patterns.

Add

Sub

Mul

Pow

Maximum(NanMode)

Minimum(NanMode)

Equal

Greater

GreaterEqual

LogicalAnd

LogicalOr

LogicalXor

LogicalNot

Select

CopyBytes

Byte-exact copy of a BOOL tensor (IDENTITY, RESHAPE on byte storage).

Implementations

impl ElementwiseOp

pub const fn inputs(self) -> &'static [Storage]

Storage of each tensor input, in operand order. Word marks a floating-point lane: the kernel resolves it to its own float storage (Word for FP32, Half for FP16); Byte lanes are always BOOL.

pub const fn output(self) -> Storage

Storage of the output tensor, with the same float-lane convention as Self::inputs.

pub const fn extra_spec_constants(self) -> u32

Trailing operator-specific specialization constants (after the operand and shape block).

Trait Implementations

impl Clone for ElementwiseOp

fn clone(&self) -> ElementwiseOp

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for ElementwiseOp

impl Debug for ElementwiseOp

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

Formats the value using the given formatter.

impl Eq for ElementwiseOp

impl Hash for ElementwiseOp

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 ElementwiseOp

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