Struct MatmulProblemSize

pub struct MatmulProblemSize {
    pub m: u32,
    pub n: u32,
    pub k: u32,
}

Fields

m: u32

n: u32

k: u32

Implementations

impl MatmulProblemSize

pub fn new(m: u32, n: u32, k: u32) -> Self

pub fn get(&self, dim: MatmulDim) -> u32

pub fn m(&self) -> u32

pub fn n(&self) -> u32

pub fn k(&self) -> u32

pub fn mn(&self) -> u32

pub fn mk(&self) -> u32

pub fn nk(&self) -> u32

pub fn mnk(&self) -> u32

Trait Implementations

impl Clone for MatmulProblemSize

fn clone(&self) -> MatmulProblemSize

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl CubeType for MatmulProblemSize

type ExpandType = MatmulProblemSizeExpand

fn into_mut(scope: &mut Scope, expand: Self::ExpandType) -> Self::ExpandType

Wrapper around the init method, necessary to type inference.

impl Debug for MatmulProblemSize

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

Formats the value using the given formatter.

impl From<(i32, i32, i32)> for MatmulProblemSize

fn from(value: (i32, i32, i32)) -> Self

Converts to this type from the input type.

impl From<(u16, u16, u16)> for MatmulProblemSize

fn from(value: (u16, u16, u16)) -> Self

Converts to this type from the input type.

impl From<(u32, u32, u32)> for MatmulProblemSize

fn from(value: (u32, u32, u32)) -> Self

Converts to this type from the input type.

impl From<(u8, u8, u8)> for MatmulProblemSize

fn from(value: (u8, u8, u8)) -> Self

Converts to this type from the input type.

impl From<(usize, usize, usize)> for MatmulProblemSize

fn from(value: (usize, usize, usize)) -> Self

Converts to this type from the input type.

impl From<MatmulProblem> for MatmulProblemSize

fn from(problem: MatmulProblem) -> Self

Converts to this type from the input type.

impl From<MatmulProblemSize> for (i32, i32, i32)

fn from(value: MatmulProblemSize) -> Self

Converts to this type from the input type.

impl From<MatmulProblemSize> for (u16, u16, u16)

fn from(value: MatmulProblemSize) -> Self

Converts to this type from the input type.

impl From<MatmulProblemSize> for (u32, u32, u32)

fn from(value: MatmulProblemSize) -> Self

Converts to this type from the input type.

impl From<MatmulProblemSize> for (u8, u8, u8)

fn from(value: MatmulProblemSize) -> Self

Converts to this type from the input type.

impl From<MatmulProblemSize> for (usize, usize, usize)

fn from(value: MatmulProblemSize) -> Self

Converts to this type from the input type.

impl From<MatmulProblemSize> for MatmulKind

fn from(matmul_size: MatmulProblemSize) -> Self

Converts to this type from the input type.

impl Hash for MatmulProblemSize

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

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

impl Eq for MatmulProblemSize

impl StructuralPartialEq for MatmulProblemSize