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

mod compilation_descriptor;
mod deployment_platform;
mod executable;
mod executable_completion_handler;
mod executable_execution_descriptor;
mod executable_scheduled_handler;
mod executable_serialization_descriptor;
mod execution_descriptor;
mod execution_stage;
mod optimization;
mod optimization_profile;

use super::{Tensor, TensorData};
use objc2::rc::Retained;
use std::collections::HashMap;

pub use compilation_descriptor::CompilationDescriptor;
pub use deployment_platform::DeploymentPlatform;
pub use executable::Executable;
pub use executable_completion_handler::ExecutableCompletionHandler;
pub use executable_execution_descriptor::ExecutableExecutionDescriptor;
pub use executable_scheduled_handler::ExecutableScheduledHandler;
pub use executable_serialization_descriptor::ExecutableSerializationDescriptor;
pub use execution_descriptor::ExecutionDescriptor;
pub use execution_stage::ExecutionStage;
pub use optimization::Optimization;
pub use optimization_profile::OptimizationProfile;

/// Result type for graph execution
pub type ExecutionResult = HashMap<Retained<Tensor>, Retained<TensorData>>;