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 call;
9pub mod control_flow;
10pub mod data;
11pub mod error;
12pub mod execution;
13pub mod ffi;
14pub mod gather;
15pub mod graph;
16pub mod ops;
17pub mod random;
18pub mod rnn;
19pub mod types;
20
21pub use crate::data::TensorData;
22pub use crate::error::{Error, Result};
23pub use crate::control_flow::WhileBeforeResult;
24pub use crate::execution::{
25 deployment_platform, graph_options, optimization, optimization_profile,
26 reduced_precision_fast_math, CompilationDescriptor, ExecutableExecutionDescriptor,
27 ExecutableSerializationDescriptor, ExecutionDescriptor,
28};
29pub use crate::graph::{
30 data_type, data_type_size, padding_style, tensor_named_data_layout, Convolution2DDescriptor,
31 Convolution2DDescriptorInfo, Executable, Feed, FeedDescription, Graph, Pooling2DDescriptor,
32 Pooling2DDescriptorInfo, Tensor,
33};
34pub use crate::ops::{BinaryArithmeticOp, ReductionAxesOp, ReductionAxisOp, UnaryArithmeticOp};
35pub use crate::random::{
36 random_distribution, random_normal_sampling_method, RandomOpDescriptor,
37};
38pub use crate::rnn::{
39 rnn_activation, GRUDescriptor, LSTMDescriptor, SingleGateRNNDescriptor,
40};
41pub use crate::types::{graph_device_type, GraphDevice, Operation, ShapedType};