Struct ElementwiseSpec 

pub struct ElementwiseSpec<'a> {
    pub count: u32,
    pub inputs: &'a [Operand],
    pub output: Operand,
    pub dims: [u32; 6],
    pub strides: &'a [[u32; 6]],
    pub clamp: Option<[u32; 2]>,
}

Specialization payload of an elementwise dispatch. The declaration order in the elementwise kernel is: count, each input operand, the output operand, then (broadcast only) the output dims and each input’s element strides, then the clamp bounds.

Fields

count: u32

inputs: &'a [Operand]

output: Operand

dims: [u32; 6]

Output shape padded with leading ones to MAX_RANK.

strides: &'a [[u32; 6]]

Per input, element strides over dims (0 along a broadcast dimension).

clamp: Option<[u32; 2]>

CLAMP bounds as FP32 bit patterns.

Implementations

impl ElementwiseSpec<'_>

pub fn words(&self, broadcast: bool) -> Vec<u32>

Trait Implementations

impl<'a> Clone for ElementwiseSpec<'a>

fn clone(&self) -> ElementwiseSpec<'a>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<'a> Debug for ElementwiseSpec<'a>

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

Formats the value using the given formatter.

impl<'a> Eq for ElementwiseSpec<'a>

impl<'a> PartialEq for ElementwiseSpec<'a>

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