nnapi 0.2.0

Provides a safe abstraction of the Android NNAPI FFI bindings.
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 std::{
    ops::{Deref, DerefMut},
    ptr::{null_mut, NonNull},
};

use nnapi_sys::{
    ANeuralNetworksCompilation, ANeuralNetworksCompilation_create,
    ANeuralNetworksCompilation_finish, ANeuralNetworksCompilation_free, ResultCode,
};

use crate::{Execution, IntoResult, Model};

pub struct Compilation {
    inner: NonNull<ANeuralNetworksCompilation>,
}

impl Compilation {
    pub fn new(model: &mut Model) -> crate::Result<Self> {
        let mut compilation = null_mut();
        unsafe { ANeuralNetworksCompilation_create(&mut **model, &mut compilation) }
            .into_result()?;

        Ok(Compilation {
            inner: NonNull::new(compilation).ok_or(ResultCode::ANEURALNETWORKS_UNEXPECTED_NULL)?,
        })
    }

    #[inline]
    pub fn finish(&mut self) -> crate::Result<()> {
        unsafe { ANeuralNetworksCompilation_finish(&mut **self) }.into_result()
    }

    #[inline]
    pub fn create_execution(&mut self) -> crate::Result<Execution> {
        crate::Execution::new(self)
    }
}

impl Deref for Compilation {
    type Target = ANeuralNetworksCompilation;

    #[inline]
    fn deref(&self) -> &Self::Target {
        unsafe { self.inner.as_ref() }
    }
}

impl DerefMut for Compilation {
    #[inline]
    fn deref_mut(&mut self) -> &mut Self::Target {
        unsafe { self.inner.as_mut() }
    }
}

impl Drop for Compilation {
    fn drop(&mut self) {
        unsafe {
            ANeuralNetworksCompilation_free(self.inner.as_mut());
        }
    }
}