apple-mpsgraph 0.2.1

Safe Rust bindings for Apple's MetalPerformanceShadersGraph framework on macOS, backed by a Swift bridge
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

#![doc = include_str!("../README.md")]
#![cfg_attr(docsrs, feature(doc_cfg))]
#![allow(clippy::missing_const_for_fn)]
#![allow(clippy::missing_errors_doc)]
#![allow(clippy::module_name_repetitions)]
#![allow(clippy::must_use_candidate)]

pub mod call;
pub mod control_flow;
pub mod data;
pub mod error;
pub mod execution;
pub mod ffi;
pub mod gather;
pub mod graph;
pub mod ops;
pub mod random;
pub mod rnn;
pub mod types;

pub use crate::data::TensorData;
pub use crate::error::{Error, Result};
pub use crate::control_flow::WhileBeforeResult;
pub use crate::execution::{
    deployment_platform, graph_options, optimization, optimization_profile,
    reduced_precision_fast_math, CompilationDescriptor, ExecutableExecutionDescriptor,
    ExecutableSerializationDescriptor, ExecutionDescriptor,
};
pub use crate::graph::{
    data_type, data_type_size, padding_style, tensor_named_data_layout, Convolution2DDescriptor,
    Convolution2DDescriptorInfo, Executable, Feed, FeedDescription, Graph, Pooling2DDescriptor,
    Pooling2DDescriptorInfo, Tensor,
};
pub use crate::ops::{BinaryArithmeticOp, ReductionAxesOp, ReductionAxisOp, UnaryArithmeticOp};
pub use crate::random::{
    random_distribution, random_normal_sampling_method, RandomOpDescriptor,
};
pub use crate::rnn::{
    rnn_activation, GRUDescriptor, LSTMDescriptor, SingleGateRNNDescriptor,
};
pub use crate::types::{graph_device_type, GraphDevice, Operation, ShapedType};