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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50

use ffi::TF_Code;
use std::ffi::CStr;
use std::{error, fmt};

use status::Status;

/// An error.
#[derive(Clone, Debug)]
pub struct Error {
    message: String,
}

impl Error {
    #[doc(hidden)]
    pub fn from_status(status: &Status) -> Option<Self> {
        if ffi!(TF_GetCode(status.as_raw())) == TF_Code::TF_OK {
            return None;
        }
        let message = ffi!(TF_Message(status.as_raw()));
        if message.is_null() {
            return None;
        }
        let message = match unsafe { CStr::from_ptr(message).to_str() } {
            Ok(message) => message.into(),
            _ => String::new(),
        };
        Some(Error { message: message })
    }
}

impl fmt::Display for Error {
    #[inline]
    fn fmt(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
        self.message.fmt(formatter)
    }
}

impl error::Error for Error {
    #[inline]
    fn description(&self) -> &str {
        &self.message
    }
}

impl<T> From<T> for Error where T: Into<String> {
    #[inline]
    fn from(message: T) -> Error {
        Error { message: message.into() }
    }
}