1use tract_nnef::internal::*;
2
3mod exp_unit_norm;
4
5pub trait WithTractExtra {
6 fn enable_tract_extra(&mut self);
7 fn with_tract_extra(self) -> Self;
8}
9
10impl WithTractExtra for tract_nnef::framework::Nnef {
11 fn enable_tract_extra(&mut self) {
12 self.registries.push(tract_extra_registry());
13 }
14
15 fn with_tract_extra(mut self) -> Self {
16 self.enable_tract_extra();
17 self
18 }
19}
20
21pub fn tract_extra_registry() -> Registry {
22 let mut reg = Registry::new("tract_extra");
23 exp_unit_norm::register(&mut reg);
24 reg
25}
26
27pub fn register_pulsifiers() {
28 let _ = tract_extra_registry();
29}