Struct tract_hir::internal::Invariants

pub struct Invariants {
    pub axes: SmallVec<[AxisInfo; 4]>,
    /* private fields */
}

Fields

axes: SmallVec<[AxisInfo; 4]>

Implementations

impl Invariants

pub fn none() -> Invariants

pub fn new_element_wise(
    inputs: &[&TypedFact],
    outputs: &[&TypedFact]
) -> Result<Invariants, Error>

pub fn element_wise(&self) -> bool

impl Invariants

pub fn track_input_axis(&self, input: usize, axis: usize) -> Option<&AxisInfo>

pub fn track_output_axis(&self, output: usize, axis: usize) -> Option<&AxisInfo>

pub fn unary_track_axis_up(
    &self,
    axis: usize,
    only_disposable: bool
) -> Option<usize>

pub fn unary_track_axis_down(
    &self,
    axis: usize,
    only_disposable: bool
) -> Option<usize>

Trait Implementations

impl Clone for Invariants

fn clone(&self) -> Invariants

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for Invariants

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

Formats the value using the given formatter.

impl Default for Invariants

fn default() -> Invariants

Returns the “default value” for a type.

impl From<SmallVec<[AxisInfo; 4]>> for Invariants

fn from(axes: SmallVec<[AxisInfo; 4]>) -> Invariants

Converts to this type from the input type.

impl FromIterator<AxisInfo> for Invariants

fn from_iter<T>(iter: T) -> Invariantswhere
    T: IntoIterator<Item = AxisInfo>,

Creates a value from an iterator.