instmodel_inference 0.5.0

High-performance neural network inference library with instruction-based execution
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

//! Activation instruction implementation.

use crate::activation::Activation;
use crate::errors::InstructionModelError;
use crate::instructions::Instruction;

/// Represents an instruction that applies an activation function to values in place.
pub struct ActivationInstruction {
    activation: Activation,
    output_ptr: usize,
    data_size: usize,
}

impl ActivationInstruction {
    pub fn new(activation: Activation, output_ptr: usize, data_size: usize) -> Self {
        Self {
            activation,
            output_ptr,
            data_size,
        }
    }
}

impl Instruction for ActivationInstruction {
    fn output_ptr(&self) -> usize {
        self.output_ptr
    }

    fn data_size(&self) -> usize {
        self.data_size
    }

    fn apply(&self, unified_computation_buffer: &mut [f32]) -> Result<(), InstructionModelError> {
        let start = self.output_ptr;
        let end = start + self.data_size;
        let slice = &mut unified_computation_buffer[start..end];
        self.activation.apply_in_place(slice);
        Ok(())
    }
}