Trait coaster_blas::plugin::Gemm

pub trait Gemm<F> {
    // Required method
    fn gemm(
        &self,
        alpha: &SharedTensor<F>,
        at: Transpose,
        a: &SharedTensor<F>,
        bt: Transpose,
        b: &SharedTensor<F>,
        beta: &SharedTensor<F>,
        c: &mut SharedTensor<F>
    ) -> Result<(), Error>;
}

Provides the gemm operation.

Required Methods

fn gemm( &self, alpha: &SharedTensor<F>, at: Transpose, a: &SharedTensor<F>, bt: Transpose, b: &SharedTensor<F>, beta: &SharedTensor<F>, c: &mut SharedTensor<F> ) -> Result<(), Error>

Computes a matrix-matrix product with general matrices.

Saves the result into c. This is a Level 3 BLAS operation.

Implementations on Foreign Types

impl Gemm<f32> for Backend<Cuda>

fn gemm( &self, alpha: &SharedTensor<f32>, at: Transpose, a: &SharedTensor<f32>, bt: Transpose, b: &SharedTensor<f32>, beta: &SharedTensor<f32>, c: &mut SharedTensor<f32> ) -> Result<(), Error>

impl Gemm<f32> for Backend<Native>

fn gemm( &self, alpha: &SharedTensor<f32>, at: Transpose, a: &SharedTensor<f32>, bt: Transpose, b: &SharedTensor<f32>, beta: &SharedTensor<f32>, c: &mut SharedTensor<f32> ) -> Result<(), Error>

impl Gemm<f64> for Backend<Native>

fn gemm( &self, alpha: &SharedTensor<f64>, at: Transpose, a: &SharedTensor<f64>, bt: Transpose, b: &SharedTensor<f64>, beta: &SharedTensor<f64>, c: &mut SharedTensor<f64> ) -> Result<(), Error>

Implementors