Trait coaster_blas::plugin::Gemm

pub trait Gemm<F> {
    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.

Implementors

• impl Gemm<f32> for Backend<Native>
• impl Gemm<f64> for Backend<Native>
• impl Gemm<f32> for Backend<Cuda>