Trait coaster_blas::plugin::Axpy

pub trait Axpy<F> {
    // Required method
    fn axpy(
        &self,
        a: &SharedTensor<F>,
        x: &SharedTensor<F>,
        y: &mut SharedTensor<F>
    ) -> Result<(), Error>;
}

Provides the axpy operation.

Required Methods

fn axpy( &self, a: &SharedTensor<F>, x: &SharedTensor<F>, y: &mut SharedTensor<F> ) -> Result<(), Error>

Computes a vector x times a constant a plus a vector y aka. a * x + y.

Saves the resulting vector back into y. This is a Level 1 BLAS operation.

Implementations on Foreign Types

impl Axpy<f32> for Backend<Cuda>

fn axpy( &self, a: &SharedTensor<f32>, x: &SharedTensor<f32>, y: &mut SharedTensor<f32> ) -> Result<(), Error>

impl Axpy<f32> for Backend<Native>

fn axpy( &self, a: &SharedTensor<f32>, x: &SharedTensor<f32>, y: &mut SharedTensor<f32> ) -> Result<(), Error>

impl Axpy<f64> for Backend<Native>

fn axpy( &self, a: &SharedTensor<f64>, x: &SharedTensor<f64>, y: &mut SharedTensor<f64> ) -> Result<(), Error>

Implementors