mlir::dialects::linalg

Trait TransformShape

Source
pub trait TransformShape {
    // Required methods
    fn matmul(&self, rhs: &Self) -> Option<Vec<i64>>;
    fn matmul_transpose_a(&self, rhs: &Self) -> Option<Vec<i64>>;
    fn matmul_transpose_b(&self, rhs: &Self) -> Option<Vec<i64>>;
    fn matvec(&self, rhs: &Self) -> Option<Vec<i64>>;
    fn vecmat(&self, rhs: &Self) -> Option<Vec<i64>>;
    fn unpack_matmul(&self, rhs: &Self) -> Option<ShapeUnpacked>;
    fn unpack_matmul_transpose_a(&self, rhs: &Self) -> Option<ShapeUnpacked>;
    fn unpack_matmul_transpose_b(&self, rhs: &Self) -> Option<ShapeUnpacked>;
    fn unpack_matvec(&self, rhs: &Self) -> Option<ShapeUnpacked>;
    fn unpack_vecmat(&self, rhs: &Self) -> Option<ShapeUnpacked>;
}

Required Methods§

Source

fn matmul(&self, rhs: &Self) -> Option<Vec<i64>>

Source

fn matmul_transpose_a(&self, rhs: &Self) -> Option<Vec<i64>>

Source

fn matmul_transpose_b(&self, rhs: &Self) -> Option<Vec<i64>>

Source

fn matvec(&self, rhs: &Self) -> Option<Vec<i64>>

Source

fn vecmat(&self, rhs: &Self) -> Option<Vec<i64>>

Source

fn unpack_matmul(&self, rhs: &Self) -> Option<ShapeUnpacked>

Source

fn unpack_matmul_transpose_a(&self, rhs: &Self) -> Option<ShapeUnpacked>

Source

fn unpack_matmul_transpose_b(&self, rhs: &Self) -> Option<ShapeUnpacked>

Source

fn unpack_matvec(&self, rhs: &Self) -> Option<ShapeUnpacked>

Source

fn unpack_vecmat(&self, rhs: &Self) -> Option<ShapeUnpacked>

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.

Implementors§