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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
#[macro_export]
macro_rules! wrap {
($($x:expr),*) => ({
vec![$( $crate::analyser::rules::expr::IntoExp::bex($x) ),*]
});
($($x:expr,)*) => (wrap![$($x),*]);
}
use crate::internal::*;
mod cache;
pub mod expr;
mod path;
mod proxies;
mod solver;
pub use self::proxies::*;
pub use self::solver::Solver;
pub type InferenceResult = TractResult<()>;
pub trait InferenceRulesOp {
fn rules<'r, 'p: 'r, 's: 'r>(
&'s self,
solver: &mut Solver<'r>,
inputs: &'p [TensorProxy],
outputs: &'p [TensorProxy],
) -> InferenceResult;
}
impl<O: InferenceRulesOp> crate::ops::InferenceOp for O {
fn infer_facts(
&self,
inputs: TVec<&TensorFact>,
outputs: TVec<&TensorFact>,
) -> TractResult<(TVec<TensorFact>, TVec<TensorFact>)> {
let inputs_proxy: TVec<TensorProxy> =
(0..inputs.len()).map(|ix| TensorProxy::new(tvec!(0, ix as isize).into())).collect();
let outputs_proxy: TVec<TensorProxy> =
(0..outputs.len()).map(|ix| TensorProxy::new(tvec!(1, ix as isize).into())).collect();
let mut solver = Solver::default();
self.rules(&mut solver, &inputs_proxy, &outputs_proxy)?;
solver.infer_facts((inputs, outputs))
}
}