Struct tract_linalg::Ops

pub struct Ops {
    pub leaky_relu_f16: Box<dyn Fn() -> Box<dyn ElementWise<f16, f16>> + Send + Sync>,
    pub leaky_relu_f32: Box<dyn Fn() -> Box<dyn ElementWise<f32, f32>> + Send + Sync>,
    pub sigmoid_f16: Box<dyn Fn() -> Box<dyn ElementWise<f16>> + Send + Sync>,
    pub sigmoid_f32: Box<dyn Fn() -> Box<dyn ElementWise<f32>> + Send + Sync>,
    pub tanh_f16: Box<dyn Fn() -> Box<dyn ElementWise<f16>> + Send + Sync>,
    pub tanh_f32: Box<dyn Fn() -> Box<dyn ElementWise<f32>> + Send + Sync>,
    pub erf_f32: Box<dyn Fn() -> Box<dyn ElementWise<f32>> + Send + Sync>,
    pub lut_u8: Box<dyn Fn(&[u8]) -> Box<dyn Lut> + Send + Sync>,
    /* private fields */
}

Fields

leaky_relu_f16: Box<dyn Fn() -> Box<dyn ElementWise<f16, f16>> + Send + Sync>

leaky_relu_f32: Box<dyn Fn() -> Box<dyn ElementWise<f32, f32>> + Send + Sync>

sigmoid_f16: Box<dyn Fn() -> Box<dyn ElementWise<f16>> + Send + Sync>

sigmoid_f32: Box<dyn Fn() -> Box<dyn ElementWise<f32>> + Send + Sync>

tanh_f16: Box<dyn Fn() -> Box<dyn ElementWise<f16>> + Send + Sync>

tanh_f32: Box<dyn Fn() -> Box<dyn ElementWise<f32>> + Send + Sync>

erf_f32: Box<dyn Fn() -> Box<dyn ElementWise<f32>> + Send + Sync>

lut_u8: Box<dyn Fn(&[u8]) -> Box<dyn Lut> + Send + Sync>

Implementations

impl Ops

pub fn mmm_f32_impls(&self) -> &[Box<dyn MatMatMul>]

pub fn mmm( &self, a: DatumType, b: DatumType, c: DatumType, m: Option<usize>, k: Option<usize>, n: Option<usize> ) -> Option<Box<dyn MatMatMul>>