tfdeploy 0.0.10

Tiny, no-nonsense, self contained, TensorFlow inference
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61

use ops::prelude::*;

mod add_n;

pub fn register_all_ops(reg: &mut OpRegister) {
    reg.insert("Abs", abs);
    reg.insert("Add", add);
    reg.insert("AddN", add_n::add_n);
    reg.insert("BiasAdd", add);
    reg.insert("Div", div);
    reg.insert("FloorMod", rem);
    reg.insert("Mul", mul);
    reg.insert("Neg", neg);
    reg.insert("Rsqrt", rsqrt);
    reg.insert("Sub", sub);
    reg.insert("Tanh", tanh);
}

element_map_signed!(Abs, abs, |x| x.abs());
element_map_float!(Rsqrt, rsqrt, |x| x.sqrt().recip());
element_map_float!(Tanh, tanh, |x| x.tanh());

element_map!(Neg, neg, [i32, f32, TDim], |x| -x);

element_bin!(Add, add, [i32, f32, TDim], |mut a, b| {
    a += &b;
    a
});
element_bin!(Div, div, [i32, f32, TDim], |mut a, b| {
    a /= &b;
    a
});
element_bin!(Mul, mul, [i32, f32, TDim], |mut a, b| {
    a *= &b;
    a
});
element_bin!(Sub, sub, [i32, f32, TDim], |mut a, b| {
    a -= &b;
    a
});
element_bin!(Rem, rem, [i32, f32, TDim], |mut a, b| {
    a %= &b;
    a
});

#[cfg(test)]
mod tests {
    use ndarray::arr2;
    #[test]
    fn mul() {
        let a = arr2(&[[1., 2.], [3., 4.]]);
        let b = arr2(&[[1., 0.], [0., 0.]]);
        assert_eq!(a * b, arr2(&[[1., 0.], [0., 0.]]));
    }
    #[test]
    fn dot() {
        let a = arr2(&[[1., 2.], [3., 4.]]);
        let b = arr2(&[[1., 0.], [0., 0.]]);
        assert_eq!(a.dot(&b), arr2(&[[1., 0.], [3., 0.]]));
    }
}