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
8pub mod data;
9pub mod error;
10pub mod execution;
11pub mod ffi;
12pub mod graph;
13pub mod ops;
14pub mod types;
15
16pub use crate::data::TensorData;
17pub use crate::error::{Error, Result};
18pub use crate::execution::{
19 deployment_platform, graph_options, optimization, optimization_profile,
20 reduced_precision_fast_math, CompilationDescriptor, ExecutableExecutionDescriptor,
21 ExecutableSerializationDescriptor, ExecutionDescriptor,
22};
23pub use crate::graph::{
24 data_type, data_type_size, padding_style, tensor_named_data_layout, Convolution2DDescriptor,
25 Convolution2DDescriptorInfo, Executable, Feed, FeedDescription, Graph, Pooling2DDescriptor,
26 Pooling2DDescriptorInfo, Tensor,
27};
28pub use crate::ops::{BinaryArithmeticOp, ReductionAxesOp, ReductionAxisOp, UnaryArithmeticOp};
29pub use crate::types::{graph_device_type, GraphDevice, Operation, ShapedType};