tract-metal 0.22.2

Tiny, no-nonsense, self contained, TensorFlow and ONNX inference
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

mod command_buffer;
mod context;
mod encoder;
mod func_constants;
pub mod kernels;
pub mod ops;
mod rewrite_rules;
mod tensor;
mod tests;
mod transform;
mod utils;

use crate::func_constants::{ConstantValues, Value};
pub use crate::kernels::matmul::MetalGemmImplKind;
use crate::kernels::LibraryName;

pub use crate::context::{MetalContext, MetalStream, METAL_STREAM};
pub use crate::transform::MetalTransform;