Skip to main content

tract_onnx/ops/ml/
mod.rs

1mod category_mapper;
2mod tree_ensemble_classifier;
3
4use crate::model::OnnxOpRegister;
5
6pub fn register_all_ops(reg: &mut OnnxOpRegister) {
7    category_mapper::register_all_ops(reg);
8    tree_ensemble_classifier::register_all_ops(reg);
9}