Struct tract_hir::internal::Invariants

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

Fields

axes: SmallVec<[AxisInfo; 4]>

Implementations

impl Invariants

pub fn none() -> Invariants

pub fn new_element_wise(
    model: &ModelImpl<TypedFact, Box<dyn TypedOp + 'static>>,
    node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>
) -> Result<Invariants, TractError>

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

Performs the conversion.

impl FromIterator<AxisInfo> for Invariants

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

Creates a value from an iterator.