1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
use tract_nnef::internal::*;
mod concat;
mod deconv_delay;
mod delay;
mod pad;
mod slice;
pub use tract_nnef;
pub use tract_nnef::tract_core;
pub mod prelude {
pub use crate::WithPulse;
pub use tract_nnef::tract_core::internal::DimLike;
}
pub mod ops {
pub use super::deconv_delay::DeconvDelay;
pub use super::delay::{ Delay, DelayState };
pub use super::pad::PulsePad;
pub use super::slice::PulsedAxisSlice;
}
pub trait WithPulse {
fn enable_pulse(&mut self);
fn with_pulse(self) -> Self;
}
impl WithPulse for tract_nnef::framework::Nnef {
fn enable_pulse(&mut self) {
self.enable_tract_core();
self.registries.push(tract_nnef_registry());
}
fn with_pulse(mut self) -> Self {
self.enable_pulse();
self
}
}
pub fn tract_nnef_registry() -> Registry {
let mut reg = Registry::new("tract_pulse");
reg.aliases.push("pulse".into());
delay::register(&mut reg);
pad::register(&mut reg);
reg
}