Enum KernelFunction 

pub enum KernelFunction {
    MatMul(fn(&[f32], &[f32], &mut [f32], usize, usize, usize) -> Result<()>),
    ElementWise(fn(&[f32], &[f32], &mut [f32], ElementWiseOp) -> Result<()>),
    Reduction(fn(&[f32], &mut f32, ReductionOp) -> Result<()>),
    Convolution(fn(&[f32], &[f32], &mut [f32], ConvolutionParams) -> Result<()>),
}

Function pointer type for SIMD kernels

Variants

MatMul(fn(&[f32], &[f32], &mut [f32], usize, usize, usize) -> Result<()>)

ElementWise(fn(&[f32], &[f32], &mut [f32], ElementWiseOp) -> Result<()>)

Reduction(fn(&[f32], &mut f32, ReductionOp) -> Result<()>)

Convolution(fn(&[f32], &[f32], &mut [f32], ConvolutionParams) -> Result<()>)

Trait Implementations

impl Clone for KernelFunction

fn clone(&self) -> KernelFunction

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for KernelFunction

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.