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.enable_tract_core();
13 self.registries.push(tract_extra_registry());
14 }
15
16 fn with_tract_extra(mut self) -> Self {
17 self.enable_tract_extra();
18 self
19 }
20}
21
22pub fn tract_extra_registry() -> Registry {
23 let mut reg = Registry::new("tract_extra");
24 exp_unit_norm::register(&mut reg);
25 reg
26}
27
28pub fn register_pulsifiers() {
29 let _ = tract_extra_registry();
30}