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

use super::TypedPass;
use crate::internal::*;
use crate::model::*;
use crate::TractResult;

use crate::ops::change_axes::*;

#[derive(Clone, Debug)]
pub struct ChangeAxes;

impl TypedPass for ChangeAxes {
    fn reset(&mut self) -> TractResult<()> {
        Ok(())
    }
    fn next(&mut self, model: &TypedModel) -> TractResult<Option<TypedModelPatch>> {
        let mut interfaces = model.output_outlets()?.to_vec();
        interfaces.extend(model.input_outlets()?.iter());
        for n in model.eval_order()? {
            for suggestion in model.node(n).op.suggested_axis_changes()? {
                let outlet = suggestion.0.as_outlet(&model.node(n));
                let change = AxisChange { outlet, op: suggestion.1 };
                if let Some((patch, _)) = change_axes(model, &change, &interfaces, &[])
                    .with_context(|| {
                        format!("Making patch for {:?} from {}", change, model.node(n))
                    })?
                {
                    return Ok(Some(patch));
                }
            }
        }
        Ok(None)
        /*
        model.check_consistent_facts()?;
        let mut done_something = false;
        'top: loop {
            for n in model.eval_order()? {
                for suggestion in model.node(n).op.suggested_axis_changes()? {
                    let outlet = suggestion.0.as_outlet(&model.node(n));
                    let change = AxisChange { outlet, op: suggestion.1 };
                    if change_axes(model, &change, &interfaces, &[])
                        .and_then(|it| {
                            model.check_consistent_facts()?;
                            Ok(it)
                        })
                        .with_context(|| format!("Applying {:?} from {}", change, model.node(n)))?
                        .is_some()
                    {
                        done_something = true;
                        continue 'top;
                    }
                }
            }
            return Ok(done_something);
        }
        */
    }
}