apple_mpsgraph/
lib.rs

1#![doc = include_str!("../README.md")]
2#![cfg_attr(docsrs, feature(doc_cfg))]
3#![allow(clippy::missing_const_for_fn)]
4#![allow(clippy::missing_errors_doc)]
5#![allow(clippy::module_name_repetitions)]
6#![allow(clippy::must_use_candidate)]
7
8/// Groups `MPSGraph` framework constants for `call`.
9pub mod call;
10/// Groups `MPSGraph` framework constants for `control_flow`.
11pub mod control_flow;
12/// Groups `MPSGraph` framework constants for `data`.
13pub mod data;
14/// Groups `MPSGraph` framework constants for `error`.
15pub mod error;
16/// Groups `MPSGraph` framework constants for `execution`.
17pub mod execution;
18/// Groups `MPSGraph` framework constants for `ffi`.
19pub mod ffi;
20/// Groups `MPSGraph` framework constants for `gather`.
21pub mod gather;
22/// Groups `MPSGraph` framework constants for `graph`.
23pub mod graph;
24/// Groups `MPSGraph` framework constants for `ops`.
25pub mod ops;
26/// Groups `MPSGraph` framework constants for `random`.
27pub mod random;
28/// Groups `MPSGraph` framework constants for `rnn`.
29pub mod rnn;
30/// Groups `MPSGraph` framework constants for `specialized`.
31pub mod specialized;
32/// Groups `MPSGraph` framework constants for `types`.
33pub mod types;
34
35/// Re-exports the `MPSGraph` framework surface for this item.
36pub use crate::control_flow::WhileBeforeResult;
37/// Re-exports the `MPSGraph` framework surface for this item.
38pub use crate::data::TensorData;
39/// Re-exports the `MPSGraph` framework surface for this item.
40pub use crate::error::{Error, Result};
41/// Re-exports the `MPSGraph` framework surface for this item.
42pub use crate::execution::{
43    deployment_platform, graph_options, optimization, optimization_profile,
44    reduced_precision_fast_math, CompilationDescriptor, ExecutableExecutionDescriptor,
45    ExecutableSerializationDescriptor, ExecutionDescriptor,
46};
47/// Re-exports the `MPSGraph` framework surface for this item.
48pub use crate::graph::{
49    data_type, data_type_size, padding_mode, padding_style, tensor_named_data_layout,
50    Convolution2DDescriptor, Convolution2DDescriptorInfo, Executable, Feed, FeedDescription, Graph,
51    Pooling2DDescriptor, Pooling2DDescriptorInfo, Tensor,
52};
53/// Re-exports the `MPSGraph` framework surface for this item.
54pub use crate::ops::{BinaryArithmeticOp, ReductionAxesOp, ReductionAxisOp, UnaryArithmeticOp};
55/// Re-exports the `MPSGraph` framework surface for this item.
56pub use crate::random::{random_distribution, random_normal_sampling_method, RandomOpDescriptor};
57/// Re-exports the `MPSGraph` framework surface for this item.
58pub use crate::rnn::{rnn_activation, GRUDescriptor, LSTMDescriptor, SingleGateRNNDescriptor};
59/// Re-exports the `MPSGraph` framework surface for this item.
60pub use crate::specialized::{
61    execution_stage, fft_scaling_mode, loss_reduction_type,
62    non_maximum_suppression_coordinate_mode, pooling_return_indices_mode, reduction_mode,
63    resize_mode, resize_nearest_rounding_mode, scatter_mode, sparse_storage_type,
64    Convolution3DDescriptor, Convolution3DDescriptorInfo, CreateSparseDescriptor,
65    DepthwiseConvolution2DDescriptor, DepthwiseConvolution2DDescriptorInfo,
66    DepthwiseConvolution3DDescriptor, DepthwiseConvolution3DDescriptorInfo, FftDescriptor,
67    FftDescriptorInfo, GraphType, ImToColDescriptor, ImToColDescriptorInfo, Object,
68    Pooling4DDescriptor, Pooling4DDescriptorInfo, StencilDescriptor, StencilDescriptorInfo,
69    VariableOp,
70};
71/// Re-exports the `MPSGraph` framework surface for this item.
72pub use crate::types::{graph_device_type, GraphDevice, Operation, ShapedType};
apple_mpsgraph/lib.rs

apple_mpsgraph/
lib.rs
