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 specialized;
20pub mod types;
21
22pub use crate::control_flow::WhileBeforeResult;
23pub use crate::data::TensorData;
24pub use crate::error::{Error, Result};
25pub use crate::execution::{
26 deployment_platform, graph_options, optimization, optimization_profile,
27 reduced_precision_fast_math, CompilationDescriptor, ExecutableExecutionDescriptor,
28 ExecutableSerializationDescriptor, ExecutionDescriptor,
29};
30pub use crate::graph::{
31 data_type, data_type_size, padding_mode, padding_style, tensor_named_data_layout,
32 Convolution2DDescriptor, Convolution2DDescriptorInfo, Executable, Feed, FeedDescription, Graph,
33 Pooling2DDescriptor, Pooling2DDescriptorInfo, Tensor,
34};
35pub use crate::ops::{BinaryArithmeticOp, ReductionAxesOp, ReductionAxisOp, UnaryArithmeticOp};
36pub use crate::random::{random_distribution, random_normal_sampling_method, RandomOpDescriptor};
37pub use crate::rnn::{rnn_activation, GRUDescriptor, LSTMDescriptor, SingleGateRNNDescriptor};
38pub use crate::specialized::{
39 execution_stage, fft_scaling_mode, loss_reduction_type,
40 non_maximum_suppression_coordinate_mode, pooling_return_indices_mode, reduction_mode,
41 resize_mode, resize_nearest_rounding_mode, scatter_mode, sparse_storage_type,
42 Convolution3DDescriptor, Convolution3DDescriptorInfo, CreateSparseDescriptor,
43 DepthwiseConvolution2DDescriptor, DepthwiseConvolution2DDescriptorInfo,
44 DepthwiseConvolution3DDescriptor, DepthwiseConvolution3DDescriptorInfo, FftDescriptor,
45 FftDescriptorInfo, GraphType, ImToColDescriptor, ImToColDescriptorInfo, Object,
46 Pooling4DDescriptor, Pooling4DDescriptorInfo, StencilDescriptor, StencilDescriptorInfo,
47 VariableOp,
48};
49pub use crate::types::{graph_device_type, GraphDevice, Operation, ShapedType};