tenferro_ops/
lib.rs

1//! Internal tensor operation vocabulary shared by tenferro graph crates.
2//!
3//! This crate owns symbolic dimension expressions, extension-op traits, shape
4//! guards, and the standard tensor op enum used by traced execution and AD
5//! lowering. End users normally interact through `tenferro-runtime` and
6//! operation-family crates rather than importing this crate directly.
7//!
8//! # Examples
9//!
10//! ```
11//! use tenferro_ops::{ShapeExtent, SymDim};
12//! use tenferro_ops::std_tensor_op::StdTensorOp;
13//!
14//! let op = StdTensorOp::constant(2.0_f64);
15//! let extent = ShapeExtent::exact(SymDim::from(3usize));
16//! assert!(matches!(op, StdTensorOp::Constant { .. }));
17//! assert!(extent.is_exact());
18//! ```
19//!
20pub mod ad;
21pub mod axis;
22pub mod broadcast;
23pub mod dim_expr;
24pub mod ext_op;
25pub mod input_key;
26pub mod reduction;
27pub mod shape_extent;
28pub mod std_tensor_op;
29pub mod sym_dim;
30
31pub use ad::context::{ShapeGuard, ShapeGuardContext, TensorMeta};
32#[cfg(feature = "autodiff")]
33pub use ext_op::{
34    linearize_extension_rule, transpose_extension_rule, ExtensionLinearTransposeRule,
35    ExtensionLinearizeRule, ExtensionOp, ExtensionPrimalVjpRule, ExtensionRegistryError,
36    ExtensionRuleRole, ExtensionRuleSet, HostReference,
37};
38#[cfg(not(feature = "autodiff"))]
39pub use ext_op::{ExtensionOp, HostReference};
40pub use shape_extent::ShapeExtent;
41pub use sym_dim::SymDim;
42pub use tenferro_extension_macros::ExtensionFamilyId;
43pub use tenferro_tensor::config;
44
45#[cfg(test)]
46mod tests;
tenferro_ops/lib.rs

tenferro_ops/
lib.rs
