tenferro-runtime 0.3.0

Concrete tensor helpers, traced tensors, graph execution, and extension runtime support for tenferro.
Documentation 
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

use std::fmt;

use super::{ProgramBindings, ProgramValue, SemanticProgram};

/// Read-only request to import the dependency closure of ordered roots.
pub struct ProgramImport<'a> {
    /// Immutable source semantic program.
    pub program: &'a SemanticProgram,
    /// Tensor bindings frozen with the source program.
    pub bindings: &'a ProgramBindings,
    /// Ordered source roots to remap; empty and duplicate roots are valid.
    pub roots: &'a [ProgramValue],
}

impl fmt::Debug for ProgramImport<'_> {
    fn fmt(&self, formatter: &mut fmt::Formatter<'_>) -> fmt::Result {
        formatter
            .debug_struct("ProgramImport")
            .field("program_inputs", &self.program.inputs().len())
            .field("program_outputs", &self.program.outputs().len())
            .field("bindings", &self.bindings.len())
            .field("roots", &self.roots.len())
            .finish()
    }
}

/// Destination-builder values corresponding to requested import roots.
pub struct ImportedProgramValues {
    roots: Box<[ProgramValue]>,
}

impl ImportedProgramValues {
    pub(crate) fn new(roots: Box<[ProgramValue]>) -> Self {
        Self { roots }
    }

    /// Borrow imported roots in request order, including duplicates.
    pub fn roots(&self) -> &[ProgramValue] {
        &self.roots
    }
}

impl fmt::Debug for ImportedProgramValues {
    fn fmt(&self, formatter: &mut fmt::Formatter<'_>) -> fmt::Result {
        formatter
            .debug_struct("ImportedProgramValues")
            .field("roots", &self.roots.len())
            .finish()
    }
}