Expand description
Inference-time optimization passes for vision models.
Re-exports§
pub use bn_folding::BnParams;pub use bn_folding::fold_bn_into_conv;pub use bn_folding::fold_bn_into_linear;pub use bn_folding::verify_bn_fold;
Modules§
- bn_
folding - Batch normalization folding into preceding linear / conv layers for inference.