svod-tensor 0.1.0-alpha.3

High-level lazy tensor API for the Svod ML compiler
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

//! Reference implementations for correctness validation.
//!
//! These are simple, straightforward implementations used to verify
//! that the optimized tensor operations produce correct results.

/// Reference operations on f32 slices.
pub mod ops {
    // =========================================================================
    // Binary Operations
    // =========================================================================

    /// Reference add implementation: element-wise addition.
    pub fn add_f32(a: &[f32], b: &[f32]) -> Vec<f32> {
        a.iter().zip(b).map(|(x, y)| x + y).collect()
    }

    /// Reference subtract implementation: element-wise subtraction.
    pub fn sub_f32(a: &[f32], b: &[f32]) -> Vec<f32> {
        a.iter().zip(b).map(|(x, y)| x - y).collect()
    }

    /// Reference multiply implementation: element-wise multiplication.
    pub fn mul_f32(a: &[f32], b: &[f32]) -> Vec<f32> {
        a.iter().zip(b).map(|(x, y)| x * y).collect()
    }

    /// Reference divide implementation: element-wise division.
    pub fn div_f32(a: &[f32], b: &[f32]) -> Vec<f32> {
        a.iter().zip(b).map(|(x, y)| x / y).collect()
    }
}