tokitai-operator 0.1.0

Verified DL kernel compiler: formally-checked GEMM, p-adic, sheaf, contract-carrying ops. Paper-artifact grade.
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

//! Operator surface: the public types and families.
//!
//! The op module tree is the public contract for the operator
//! library. Each `pub mod` declares an op family; the re-exports at
//! the bottom of this file flatten the families into a single
//! `tokitai_operator::op` namespace.
//!
//! - `arithmetic` (P335) — binary / scalar / unary integer ops.
//! - `shape` (P336) — `Reshape`, `Transpose`, `Slice`, `Concat`,
//!   `Broadcast`, `Flatten`, `Squeeze`, `Unsqueeze`, `Permute`.
//! - `nn` (P337) — activations and normalization (`Relu`,
//!   `Sigmoid`, `Tanh`, `Gelu`, `Softmax`, `LayerNorm`).
//! - `index` (P338) — `Gather`, `Scatter`, `IndexSelect`,
//!   `IndexAdd`, `Nonzero`.
//! - `reductions` (P339) — `ReduceOp` variants.
//! - `registry` — the `OperatorRegistry` and `LoweringRule` types
//!   consumed by the planner.
//! - `signature` — the `Operator` trait and `OpSignature` data class.
//! - `contract` — re-exports of the domain-level contract primitives.
//!
pub mod arithmetic;
pub mod contract;
pub mod index;
pub mod nn;
pub mod reductions;
pub mod registry;
pub mod shape;
pub mod signature;

pub use arithmetic::{
    AbsOp, AddOp, ClampOp, DivOp, Exp2Op, FmaOp, Log2Op, MapOp, MatmulOp, MulByTwoOp, MulOp, NegOp,
    PAdicDotOp, PAdicMatmulFmaOp, PowOp, ReduceOp, ScalarAddOp, ScalarMulOp, SqrtOp, SquareOp,
    SubOp,
};
pub use index::{GatherOp, IndexAddOp, IndexSelectOp, NonzeroOp, ScatterOp};
pub use nn::{GeluOp, LayerNormOp, ReluOp, SigmoidOp, SoftmaxOp, TanhOp};
pub use reductions::{
    AllOp, AnyOp, ArgMaxOp, ArgMinOp, MaxOp, MeanOp, MinOp, ProdOp, ReductionKind, SumOp,
};
pub use registry::{
    BackendLoweringRegistry, DeviceRequirement, LayoutRequirement, LocalityRequirement,
    LoweringCapability, LoweringDomainRequirement, LoweringEvidence, LoweringEvidenceKind,
    LoweringObligation, LoweringRule, LoweringRuleId, OperatorEntry, OperatorRegistry,
    PrecisionRequirement, UnsupportedBackendReason, ValuationRequirement,
};
pub use shape::{
    BroadcastOp, ConcatOp, FlattenOp, PermuteOp, ReshapeOp, SliceOp, SqueezeOp, TransposeOp,
    UnsqueezeOp,
};
pub use signature::{LayerBehavior, OpInput, OpOutput, OpSignature, Operator};