tensorflux 0.8.2

The package provides an interface to TensorFlow.
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

use ffi::TF_Library;

use Result;
use buffer::Buffer;
use status::Status;

/// A library.
pub struct Library {
    raw: *mut TF_Library,
}

impl Library {
    /// Load a library.
    pub fn load<T>(name: T) -> Result<Self> where T: Into<String> {
        let name = into_cstring!(name);
        let status = try!(Status::new());
        let raw = nonnull!(ffi!(TF_LoadLibrary(name.as_ptr(), status.as_raw())), &status);
        Ok(Library { raw: raw })
    }

    /// Return the operations defined in the library.
    pub fn operations(&self) -> Buffer {
        let buffer = ffi!(TF_GetOpList(self.raw));
        unsafe { Buffer::from_raw_parts(buffer.data as *mut _, buffer.length as usize) }
    }
}

impl Drop for Library {
    #[inline]
    fn drop(&mut self) {
        // Not available in the API.
    }
}