Struct SIMDOperations

pub struct SIMDOperations;

SIMD operations for neural network computations (fallback implementation)

Implementations

impl SIMDOperations

pub fn simd_relu_f32_inplace(_input: &mut ArrayViewMut<'_, f32, IxDyn>)

Apply ReLU activation in-place using SIMD operations

pub fn simd_relu_f32(_input: &ArrayView<'_, f32, IxDyn>) -> ArrayD<f32>

Apply ReLU activation using SIMD operations

pub fn simd_sigmoid_f32_inplace(_input: &mut ArrayViewMut<'_, f32, IxDyn>)

Apply sigmoid activation in-place using SIMD operations

pub fn simd_sigmoid_f32(_input: &ArrayView<'_, f32, IxDyn>) -> ArrayD<f32>

Apply sigmoid activation using SIMD operations

pub fn simd_tanh_f32_inplace(_input: &mut ArrayViewMut<'_, f32, IxDyn>)

Apply tanh activation in-place using SIMD operations

pub fn simd_tanh_f32(_input: &ArrayView<'_, f32, IxDyn>) -> ArrayD<f32>

Apply tanh activation using SIMD operations

pub fn simd_gelu_f32(_input: &ArrayView<'_, f32, IxDyn>) -> ArrayD<f32>

Apply GELU activation using SIMD operations

pub fn simd_swish_f32(_input: &ArrayView<'_, f32, IxDyn>) -> ArrayD<f32>

Apply Swish activation using SIMD operations

pub fn simd_softmax_f32( _input: &ArrayD<f32>, _axis: Option<usize>, ) -> Result<ArrayD<f32>>

Apply softmax activation using SIMD operations

pub fn simd_cross_entropy_loss_f32( _predictions: &ArrayView<'_, f32, IxDyn>, _targets: &ArrayView<'_, f32, IxDyn>, _epsilon: f32, ) -> Result<f32>

Compute cross-entropy loss using SIMD operations

pub fn simd_matmul_f32( _a: &ArrayView<'_, f32, IxDyn>, _b: &ArrayView<'_, f32, IxDyn>, ) -> Result<ArrayD<f32>>

Perform matrix multiplication using SIMD operations

pub fn simd_add_f32( _a: &ArrayView<'_, f32, IxDyn>, _b: &ArrayView<'_, f32, IxDyn>, ) -> Result<ArrayD<f32>>

Perform element-wise addition using SIMD operations

pub fn simd_conv2d_f32( _input: &ArrayView<'_, f32, IxDyn>, _kernel: &ArrayView<'_, f32, IxDyn>, _bias: Option<&[f32]>, _stride: (usize, usize), _padding: (usize, usize), ) -> Result<ArrayD<f32>>

Perform 2D convolution using SIMD operations

pub fn simd_batch_norm_f32( _input: &ArrayView<'_, f32, IxDyn>, _gamma: Option<&[f32]>, _beta: Option<&[f32]>, _mean: &[f32], _variance: &[f32], _epsilon: f32, ) -> Result<ArrayD<f32>>

Perform batch normalization using SIMD operations