tensorflow_proto 0.1.0

Rust structs for Tensorflow Protocol buffers.
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

#![allow(clippy::large_enum_variant)]
include!(concat!(env!("OUT_DIR"), "/tensorflow_proto_gen.rs"));

#[derive(thiserror::Error, Debug)]
pub enum Error {
    #[error("Failed to encode protobuf to bytes")]
    Encode(#[source] prost::EncodeError),
}

/// Serialize a protobuf message into a vector of bytes.
pub fn into_bytes(msg: impl prost::Message) -> Result<Vec<u8>, Error> {
    let mut bytes = vec![];
    msg.encode(&mut bytes).map_err(Error::Encode)?;
    Ok(bytes)
}

#[cfg(test)]
mod tests {
    use super::{into_bytes, tensorflow};

    #[test]
    fn gpu_options() {
        let config_proto = tensorflow::ConfigProto {
            gpu_options: Some(tensorflow::GpuOptions {
                allow_growth: true,
                ..Default::default()
            }),
            ..Default::default()
        };
        let bytes = into_bytes(config_proto).unwrap();
        assert!(!bytes.is_empty());
    }
}