MatmulPrecision

Trait MatmulPrecision 

Source
pub trait MatmulPrecision:
    Send
    + Sync
    + Copy
    + 'static {
    type Lhs: MatrixPrecision;
    type Rhs: MatrixPrecision;
    type Acc: MatrixPrecision;
}
Expand description

Matrix multiplication precisions.

Required Associated Types§

Source

type Lhs: MatrixPrecision

Element type of lhs input tensor of the kernel.

Source

type Rhs: MatrixPrecision

Element type of rhs input tensor of the kernel.

Source

type Acc: MatrixPrecision

Element type of acc input tensor of the kernel.

Dyn Compatibility§

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementations on Foreign Types§

Source§

impl MatmulPrecision for f32

Source§

type Lhs = (f32, f32)

Source§

type Rhs = (f32, f32)

Source§

type Acc = (f32, f32)

Source§

impl MatmulPrecision for f64

Source§

type Lhs = (f64, f32)

Source§

type Rhs = (f64, f32)

Source§

type Acc = (f64, f32)

Source§

impl MatmulPrecision for i8

Source§

type Lhs = (i8, i8)

Source§

type Rhs = (i8, i8)

Source§

type Acc = (i32, i32)

Source§

impl MatmulPrecision for i16

Source§

type Lhs = (i16, i16)

Source§

type Rhs = (i16, i16)

Source§

type Acc = (i32, i32)

Source§

impl MatmulPrecision for i32

Source§

type Lhs = (i32, i32)

Source§

type Rhs = (i32, i32)

Source§

type Acc = (i32, i32)

Source§

impl MatmulPrecision for i64

Source§

type Lhs = (i64, i64)

Source§

type Rhs = (i64, i64)

Source§

type Acc = (i64, i64)

Source§

impl MatmulPrecision for u8

Source§

type Lhs = (u8, u8)

Source§

type Rhs = (u8, u8)

Source§

type Acc = (i32, i32)

Source§

impl MatmulPrecision for u16

Source§

type Lhs = (u16, u16)

Source§

type Rhs = (u16, u16)

Source§

type Acc = (i32, i32)

Source§

impl MatmulPrecision for u32

Source§

type Lhs = (u32, u32)

Source§

type Rhs = (u32, u32)

Source§

type Acc = (u32, u32)

Source§

impl MatmulPrecision for u64

Source§

type Lhs = (u64, u64)

Source§

type Rhs = (u64, u64)

Source§

type Acc = (u64, u64)

Source§

impl MatmulPrecision for flex32

Source§

type Lhs = (f32, f16)

Source§

type Rhs = (f32, f16)

Source§

type Acc = (f32, f32)

Source§

impl MatmulPrecision for bf16

Source§

impl MatmulPrecision for f16

Source§

type Lhs = (f16, f16)

Source§

type Rhs = (f16, f16)

Source§

type Acc = (f16, f32)

Source§

impl<LhsG: Numeric, RhsG: Numeric, AccG: Numeric, LhsS: Numeric, RhsS: Numeric, AccS: Numeric> MatmulPrecision for (LhsG, RhsG, AccG, LhsS, RhsS, AccS)

Implementors§