apple-accelerate 0.2.1

Safe Rust bindings for Apple's Accelerate framework on macOS using a Swift bridge with optional raw C FFI
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
45
46
47
48
49
50

#![allow(missing_docs, non_camel_case_types, non_upper_case_globals)]

use core::ffi::c_void;

pub type BNNSFilter = *mut c_void;
pub type BNNSGraphOptimizationPreference = u32;

#[repr(C)]
pub struct bnns_graph_compile_options_t {
    pub data: *mut c_void,
    pub size: usize,
}

#[link(name = "Accelerate", kind = "framework")]
unsafe extern "C" {
    pub fn BNNSFilterCreateLayerConvolution(
        layer_params: *const c_void,
        filter_params: *const c_void,
    ) -> BNNSFilter;
    pub fn BNNSFilterCreateLayerFullyConnected(
        layer_params: *const c_void,
        filter_params: *const c_void,
    ) -> BNNSFilter;
    pub fn BNNSFilterApply(filter: BNNSFilter, input: *const c_void, output: *mut c_void) -> i32;
    pub fn BNNSFilterDestroy(filter: BNNSFilter);

    pub fn BNNSGraphCompileOptionsMakeDefault() -> bnns_graph_compile_options_t;
    pub fn BNNSGraphCompileOptionsDestroy(options: bnns_graph_compile_options_t);
    pub fn BNNSGraphCompileOptionsSetTargetSingleThread(
        options: bnns_graph_compile_options_t,
        value: bool,
    );
    pub fn BNNSGraphCompileOptionsGetTargetSingleThread(
        options: bnns_graph_compile_options_t,
    ) -> bool;
    pub fn BNNSGraphCompileOptionsSetGenerateDebugInfo(
        options: bnns_graph_compile_options_t,
        value: bool,
    );
    pub fn BNNSGraphCompileOptionsGetGenerateDebugInfo(
        options: bnns_graph_compile_options_t,
    ) -> bool;
    pub fn BNNSGraphCompileOptionsSetOptimizationPreference(
        options: bnns_graph_compile_options_t,
        preference: BNNSGraphOptimizationPreference,
    );
    pub fn BNNSGraphCompileOptionsGetOptimizationPreference(
        options: bnns_graph_compile_options_t,
    ) -> BNNSGraphOptimizationPreference;
}