mpsgraph 0.2.0

Modern Rust bindings for Apple's Metal Performance Shaders Graph framework
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

use objc2::Encoding;
use objc2::{Encode, RefEncode};
use objc2_foundation::NSUInteger;

#[repr(u64)]
#[derive(Copy, Clone, Debug, Eq, PartialEq, Hash, PartialOrd, Ord)]
pub enum TensorNamedDataLayout {
    /// LayoutNCHW
    NCHW = 0,
    /// LayoutNHWC
    NHWC = 1,
    /// LayoutOIHW
    OIHW = 2,
    /// LayoutHWIO
    HWIO = 3,
    /// LayoutCHW
    CHW = 4,
    /// LayoutHWC
    HWC = 5,
    /// LayoutHW
    HW = 6,
    /// LayoutNCDHW
    NCDHW = 7,
    /// LayoutNDHWC
    NDHWC = 8,
    /// LayoutOIDHW
    OIDHW = 9,
    /// LayoutDHWIO
    DHWIO = 10,
}

impl From<TensorNamedDataLayout> for NSUInteger {
    fn from(layout: TensorNamedDataLayout) -> Self {
        layout as NSUInteger
    }
}

unsafe impl Encode for TensorNamedDataLayout {
    const ENCODING: Encoding = NSUInteger::ENCODING;
}

unsafe impl RefEncode for TensorNamedDataLayout {
    const ENCODING_REF: Encoding = Encoding::Pointer(&Self::ENCODING);
}