Skip to main content

Module optimize

Module optimize 

Source
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.