tenferro-runtime 0.2.0

Concrete tensor helpers, traced tensors, graph execution, and extension runtime support for tenferro.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

//! Graph compilation APIs.
//!
//! # Examples
//!
//! ```
//! use tenferro_runtime::{GraphCompiler, TracedTensor};
//!
//! let x = TracedTensor::from_vec_col_major(vec![2], vec![1.0_f64, 2.0]).unwrap();
//! let y = (&x + &x).unwrap();
//! let mut compiler = GraphCompiler::new();
//! let program = compiler.compile(&y).unwrap();
//! assert_eq!(program.output_count(), 1);
//! ```

pub(crate) mod cache;
mod compiler;
mod executor;
mod lowering_view;
mod program;

pub use cache::{GraphCompilerCacheStats, GraphExecutorCacheStats};
pub use compiler::GraphCompiler;
pub use executor::GraphExecutor;
pub use lowering_view::{
    GraphInstructionView, GraphOpView, GraphProgramLoweringShapeError, GraphProgramLoweringView,
};
pub use program::{GraphProgram, GraphProgramInput};