tenferro-internal-ops 0.8.0

Graph op vocabulary and AD rule implementations for the tenferro workspace.
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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

//! Internal tensor operation vocabulary shared by tenferro graph crates.
//!
//! This crate owns symbolic dimension expressions, extension-op traits, shape
//! guards, and the standard tensor op enum used by traced execution and AD
//! lowering. End users normally interact through `tenferro-runtime` and
//! operation-family crates rather than importing this crate directly.
//!
//! # Examples
//!
//! ```
//! use tenferro_ops::{ShapeExtent, SymDim};
//! use tenferro_ops::std_tensor_op::StdTensorOp;
//!
//! let op = StdTensorOp::constant(2.0_f64);
//! let extent = ShapeExtent::exact(SymDim::from(3usize));
//! assert!(matches!(op, StdTensorOp::Constant { .. }));
//! assert!(extent.is_exact());
//! ```
//!
//! Collected shape constraints are inference-driver internals rather than a
//! crate-root extension-author API:
//!
//! ```compile_fail
//! use tenferro_ops::ExtensionShapeConstraint;
//! ```
//!
pub mod ad;
pub mod axis;
pub mod broadcast;
pub mod dim_expr;
pub mod ext_op;
pub mod input_key;
pub mod reduction;
#[doc(hidden)]
pub mod shape_constraint;
pub mod shape_extent;
pub mod std_tensor_op;
pub mod sym_dim;

pub use ad::context::{
    ShapeGuard, ShapeGuardContext, ShapeGuardError, ShapeGuardFailure, ShapeGuardResult, TensorMeta,
};
pub use ext_op::ExtensionOp;
pub use shape_constraint::{ExtensionShapeContext, ExtensionShapeError, ShapeRelation};
pub use shape_extent::ShapeExtent;
pub use sym_dim::{SymDim, SymDimConversionError};
pub use tenferro_extension_macros::ExtensionFamilyId;
pub use tenferro_tensor::config;

#[cfg(test)]
mod tests;