Struct tract_core::axes::AxesMapping

pub struct AxesMapping { /* private fields */ }

Implementations

impl AxesMapping

pub fn new(it: impl AsRef<[Axis]>) -> TractResult<AxesMapping>

pub fn new_no_inputs(it: impl AsRef<[Axis]>) -> TractResult<AxesMapping>

pub fn for_numpy_matmul( rank: usize, transposing_a: bool, transposing_b: bool, transposing_c: bool ) -> TractResult<AxesMapping>

pub fn disconnected( inputs: &[&TypedFact], outputs: &[&TypedFact] ) -> TractResult<AxesMapping>

pub fn disconnected_for_ranks( inputs: &[usize], outputs: &[usize] ) -> TractResult<AxesMapping>

pub fn natural( inputs: &[&TypedFact], outputs: &[&TypedFact] ) -> TractResult<AxesMapping>

pub fn iter_all_axes(&self) -> impl Iterator<Item = &Axis>

pub fn input_count(&self) -> usize

pub fn output_count(&self) -> usize

pub fn axis_by_repr(&self, c: char) -> Option<&Axis>

pub fn axis_positions_in_input(&self, input: usize, c: char) -> Option<&[usize]>

pub fn input_axis(&self, input: usize, position: usize) -> TractResult<&Axis>

pub fn input_rank(&self, input: usize) -> usize

pub fn input_axes(&self, input: usize) -> impl Iterator<Item = &Axis>

pub fn output_rank(&self, output: usize) -> usize

pub fn output_axis(&self, output: usize, position: usize) -> TractResult<&Axis>

pub fn output_axes(&self, output: usize) -> impl Iterator<Item = &Axis>

pub fn with_input_axis_named( self, input_id: usize, axis_pos: usize, name: char ) -> TractResult<AxesMapping>

pub fn with_output_axis_named( self, output_id: usize, axis_pos: usize, name: char ) -> TractResult<AxesMapping>

pub fn linking(self, a: char, b: char) -> TractResult<AxesMapping>

pub fn with_input_axis_linked_to( self, slot: usize, axis: usize, to: char ) -> TractResult<AxesMapping>

pub fn with_output_axis_linked_to( self, slot: usize, axis: usize, to: char ) -> TractResult<AxesMapping>

pub fn add_input(self, rank: usize) -> TractResult<AxesMapping>

pub fn check(self) -> TractResult<AxesMapping>

pub fn available_label(&self) -> char

pub fn is_element_wise_unary(&self) -> bool

pub fn extract_sub_mapping( &self, inputs: &[usize], outputs: &[usize] ) -> TractResult<AxesMapping>

pub fn relabel(self) -> TractResult<AxesMapping>

pub fn remove_axis(&self, repr: char) -> TractResult<AxesMapping>

pub fn with_extra_input(&self, slot: usize) -> TractResult<AxesMapping>

pub fn with_extra_input_axis( &self, repr: char, slot: usize, position: usize ) -> TractResult<AxesMapping>

pub fn with_extra_output_axis( &self, repr: char, slot: usize, position: usize ) -> TractResult<AxesMapping>

pub fn translate_to_axis_ops(&self) -> TractResult<Vec<AxisOp>>

pub fn from_strs( inputs: &[impl AsRef<str>], outputs: &[impl AsRef<str>] ) -> TractResult<AxesMapping>

pub fn to_strs(&self) -> (TVec<String>, TVec<String>)

Trait Implementations

impl Clone for AxesMapping

fn clone(&self) -> AxesMapping

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for AxesMapping

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

Formats the value using the given formatter.

impl Display for AxesMapping

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

Formats the value using the given formatter.

impl FromStr for AxesMapping

type Err = Error

The associated error which can be returned from parsing.

fn from_str(s: &str) -> Result<Self, Self::Err>

Parses a string s to return a value of this type.

impl Hash for AxesMapping

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<AxesMapping> for AxesMapping

fn eq(&self, other: &AxesMapping) -> bool

This method tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

This method tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Eq for AxesMapping

impl StructuralEq for AxesMapping

impl StructuralPartialEq for AxesMapping