Skip to main content

Crate nxpu_analysis

Crate nxpu_analysis 

Source

Re-exports§

pub use analyze::ActivationOp;
pub use analyze::AnalysisError;
pub use analyze::ChainOperand;
pub use analyze::ChainStep;
pub use analyze::Conv2DShape;
pub use analyze::ElementWiseOp;
pub use analyze::EmbeddedWeight;
pub use analyze::KernelPattern;
pub use analyze::MatMulShape;
pub use analyze::NormType;
pub use analyze::PoolKind;
pub use analyze::PoolShape;
pub use analyze::ReduceOp;
pub use analyze::ScalarBinding;
pub use analyze::TensorBinding;
pub use analyze::TensorRole;
pub use analyze::chain_op_names;
pub use analyze::chain_summary;
pub use analyze::classify_entry_point;
pub use analyze::extract_embedded_weights;
pub use analyze::normalize_axis;
pub use cost::Bottleneck;
pub use cost::HardwareProfile;
pub use cost::OpCost;
pub use cost::default_profiles;
pub use cost::estimate_activation_cost;
pub use cost::estimate_conv2d_cost;
pub use cost::estimate_elementwise_cost;
pub use cost::estimate_kernel_cost;
pub use cost::estimate_matmul_cost;
pub use dataflow::CriticalPathResult;
pub use dataflow::DataflowError;
pub use dataflow::DataflowGraph;
pub use dataflow::DependencyKind;
pub use dataflow::DfgEdge;
pub use dataflow::DfgNode;
pub use dataflow::DfgNodeKind;
pub use fusion::FusedActivation;
pub use fusion::FusedPattern;
pub use fusion::fuse_patterns;
pub use fusion::input_tensor_names;
pub use fusion::output_tensor_names;
pub use fusion::tensors_connect;

Modules§

analyze
IR pattern classification for ONNX lowering.
cost
Cost model and latency prediction for kernel operations.
dataflow
Dataflow graph construction and analysis.
fusion
Kernel fusion: merges adjacent classified patterns into fused operations.