Trait coaster_blas::operation::IOperationGemm

pub trait IOperationGemm<F> {
    // Required method
    fn compute(
        &self,
        alpha: &SharedTensor<F>,
        at: Transpose,
        a_dims: &[usize],
        a: &SharedTensor<F>,
        bt: Transpose,
        b_dims: &[usize],
        b: &SharedTensor<F>,
        beta: &SharedTensor<F>,
        c_dims: &[usize],
        c: &mut SharedTensor<F>
    ) -> Result<(), Error>;
}

Describes a Gemm Operation.

Required Methods

fn compute( &self, alpha: &SharedTensor<F>, at: Transpose, a_dims: &[usize], a: &SharedTensor<F>, bt: Transpose, b_dims: &[usize], b: &SharedTensor<F>, beta: &SharedTensor<F>, c_dims: &[usize], c: &mut SharedTensor<F> ) -> Result<(), Error>

Computes the Gemm operation.

Implementors