tflite 0.9.8

Rust bindings for TensorFlow Lite
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 std::marker::PhantomData;
use std::ops::Deref;

use super::bindings::root::rust::*;

#[repr(C)]
pub struct UniquePtr<T>(unique_ptr_of_void, PhantomData<T>);

unsafe impl<T> Sync for UniquePtr<T> {}
unsafe impl<T> Send for UniquePtr<T> {}

impl<T> PartialEq for UniquePtr<T>
where
    T: PartialEq,
    UniquePtr<T>: Deref<Target = T>,
{
    fn eq(&self, other: &Self) -> bool {
        self.deref() == other.deref()
    }
}

impl<T> Eq for UniquePtr<T>
where
    T: Eq,
    UniquePtr<T>: Deref<Target = T>,
{
}

impl<T> Drop for UniquePtr<T> {
    fn drop(&mut self) {
        #[allow(deprecated)]
        unsafe {
            cpp!([self as "std::unique_ptr<flatbuffers::NativeTable>*"] {
                self->reset();
            });
        }
    }
}

impl<T> UniquePtr<T> {
    pub fn is_valid(&self) -> bool {
        #[allow(deprecated)]
        unsafe {
            cpp!([self as "const std::unique_ptr<flatbuffers::NativeTable>*"] -> bool as "bool" {
                return static_cast<bool>(*self);
            })
        }
    }
}

#[cfg(test)]
mod tests {
    use crate::model::stl::vector::VectorExtract;
    use crate::model::Model;

    #[test]
    fn unittest_unique_ptr_drop() {
        let mut model = Model::from_file("data/MNISTnet_uint8_quant.tflite").unwrap();
        let _subgraph = model.subgraphs.extract(0);
    }
}