Struct tract_core::ops::matmul::MatMulAxes

pub struct MatMulAxes {
    pub a_m: usize,
    pub a_k: usize,
    pub b_k: usize,
    pub b_n: usize,
    pub c_m: usize,
    pub c_n: usize,
}

Fields

a_m: usize

a_k: usize

b_k: usize

b_n: usize

c_m: usize

c_n: usize

Implementations

impl MatMulAxes

pub fn default_for_rank(rank: usize) -> Self

pub fn default_for_ranks(a: usize, b: usize, c: usize) -> Self

pub fn transposing_a(self) -> Self

pub fn transposing_b(self) -> Self

pub fn transposing_c(self) -> Self

pub fn transposing(self, a: bool, b: bool, c: bool) -> Self

pub fn change_axis_from_b( &self, change: &AxisOp, bc_rank: usize ) -> TractResult<(MatMulAxes, Option<AxisOp>, Option<AxisOp>, Option<AxisOp>)>

pub fn change_axis_from_c( &self, change: &AxisOp, bc_rank: usize ) -> TractResult<(MatMulAxes, Option<AxisOp>, Option<AxisOp>, Option<AxisOp>)>

pub fn to_array(&self) -> [usize; 6]

pub fn from_array(array: &[usize]) -> TractResult<Self>

Trait Implementations

impl Clone for MatMulAxes

fn clone(&self) -> MatMulAxes

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for MatMulAxes

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

Formats the value using the given formatter.

impl Default for MatMulAxes

fn default() -> Self

Returns the “default value” for a type.

impl Hash for MatMulAxes

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

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

impl Eq for MatMulAxes

impl StructuralEq for MatMulAxes

impl StructuralPartialEq for MatMulAxes