Skip to main content

native_onnx/
lib.rs

1#![deny(unsafe_code)]
2#![warn(missing_docs)]
3#![doc = include_str!("../README.md")]
4
5mod api;
6mod buffers;
7// The only engine module allowed to call the dynamically loaded CUDA driver.
8#[allow(unsafe_code)]
9mod cuda_transfer;
10mod diagnostics;
11mod engine;
12mod options;
13mod session;
14mod tensor;
15
16pub use anyhow::{Error, Result};
17pub use api::{Compilation, CompiledFormat, FallbackEvent, ModelInfo, OnnxRuntime};
18pub use options::{
19    Backend, BackendSelection, CompileOptions, CompileTarget, CudaOptions, OnnxOptions,
20    TensorRtOptions,
21};
22/// Configure additional installed ONNX Runtime providers through [`Backend::Custom`].
23pub use ort::ep;
24pub use tensor::{
25    DType, Tensor, TensorBuffer, TensorData, TensorDataMut, TensorSpec, TensorView, TensorViewMut,
26    bf16, element_count, f16,
27};