Trait coaster_blas::plugin::Gbmv

source ·
pub trait Gbmv<F> {
    // Required method
    fn gbmv(
        &self,
        alpha: &SharedTensor<F>,
        at: Transpose,
        a: &SharedTensor<F>,
        kl: &SharedTensor<u32>,
        ku: &SharedTensor<u32>,
        x: &SharedTensor<F>,
        beta: &SharedTensor<F>,
        c: &mut SharedTensor<F>
    ) -> Result<(), Error>;
}
Expand description

Provides the gbmv operation

Required Methods§

source

fn gbmv( &self, alpha: &SharedTensor<F>, at: Transpose, a: &SharedTensor<F>, kl: &SharedTensor<u32>, ku: &SharedTensor<u32>, x: &SharedTensor<F>, beta: &SharedTensor<F>, c: &mut SharedTensor<F> ) -> Result<(), Error>

Computes a matrix-vector product with a band matrix

Saves the resulting vector into c. This is a Level 2 BLAS operation.

Implementations on Foreign Types§

source§

impl Gbmv<f32> for Backend<Cuda>

source§

fn gbmv( &self, _alpha: &SharedTensor<f32>, _at: Transpose, _a: &SharedTensor<f32>, _kl: &SharedTensor<u32>, _ku: &SharedTensor<u32>, _x: &SharedTensor<f32>, _beta: &SharedTensor<f32>, _c: &mut SharedTensor<f32> ) -> Result<(), Error>

source§

impl Gbmv<f32> for Backend<Native>

source§

fn gbmv( &self, alpha: &SharedTensor<f32>, at: Transpose, a: &SharedTensor<f32>, kl: &SharedTensor<u32>, ku: &SharedTensor<u32>, x: &SharedTensor<f32>, beta: &SharedTensor<f32>, c: &mut SharedTensor<f32> ) -> Result<(), Error>

source§

impl Gbmv<f64> for Backend<Native>

source§

fn gbmv( &self, alpha: &SharedTensor<f64>, at: Transpose, a: &SharedTensor<f64>, kl: &SharedTensor<u32>, ku: &SharedTensor<u32>, x: &SharedTensor<f64>, beta: &SharedTensor<f64>, c: &mut SharedTensor<f64> ) -> Result<(), Error>

Implementors§