tensorflux 0.2.1

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
51
52
53
54

translate! {
    #[doc = "A type."]
    pub struct Type => TF_DataType,

    Bool => TF_BOOL,
    Half => TF_HALF,
    Float => TF_FLOAT,
    Double => TF_DOUBLE,
    Complex64 => TF_COMPLEX64,
    Complex128 => TF_COMPLEX128,
    Int8 => TF_INT8,
    Int16 => TF_INT16,
    Int32 => TF_INT32,
    Int64 => TF_INT64,
    QInt8 => TF_QINT8,
    QInt16 => TF_QINT16,
    QInt32 => TF_QINT32,
    QUInt8 => TF_QUINT8,
    QUInt16 => TF_QUINT16,
    UInt8 => TF_UINT8,
    UInt16 => TF_UINT16,
    String => TF_STRING,
    BFloat16 => TF_BFLOAT16,
}

/// A value.
pub trait Value: 'static {
    /// Return the type.
    fn kind() -> Type;
}

macro_rules! implement {
    ($($native:ident => $variant:ident,)*) => {
        $(impl Value for $native {
            #[inline]
            fn kind() -> Type {
                Type::$variant
            }
        })*
    }
}

implement! {
    bool => Bool,
    f32 => Float,
    f64 => Double,
    i8 => Int8,
    i16 => Int16,
    i32 => Int32,
    i64 => Int64,
    u8 => UInt8,
    u16 => UInt16,
    String => String,
}