1pub mod analyze;
2pub mod cost;
3pub mod dataflow;
4pub mod fusion;
5
6pub use analyze::{
7 ActivationOp, AnalysisError, ChainOperand, ChainStep, Conv2DShape, ElementWiseOp,
8 EmbeddedWeight, KernelPattern, MatMulShape, NormType, PoolKind, PoolShape, ReduceOp,
9 ScalarBinding, TensorBinding, TensorRole, chain_op_names, chain_summary, classify_entry_point,
10 extract_embedded_weights, normalize_axis,
11};
12pub use cost::{
13 Bottleneck, HardwareProfile, OpCost, default_profiles, estimate_activation_cost,
14 estimate_conv2d_cost, estimate_elementwise_cost, estimate_kernel_cost, estimate_matmul_cost,
15};
16pub use dataflow::{
17 CriticalPathResult, DataflowError, DataflowGraph, DependencyKind, DfgEdge, DfgNode, DfgNodeKind,
18};
19pub use fusion::{
20 FusedActivation, FusedPattern, fuse_patterns, input_tensor_names, output_tensor_names,
21 tensors_connect,
22};