topos 0.13.1

An autodiff compiler stack in Rust.
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

use smallvec::SmallVec;

use super::ValueId;

/// The operand links of one recorded node, in the operation's positional
/// order.
///
/// It is one entry of the tape's operands column: the op column
/// holds what a node computes, this column holds which earlier nodes it
/// reads. The links are stored inline up to the arity of every current
/// operation, so scanning the column stays free of pointer chasing.
#[derive(Debug, Clone, PartialEq, Eq)]
pub(crate) struct Operands(SmallVec<[ValueId; 2]>);

impl Operands {
    /// Creates the empty operand list of a source node.
    pub(crate) fn none() -> Self {
        Self(SmallVec::new())
    }

    /// Creates the operand list holding `links` in order.
    pub(crate) fn from_slice(links: &[ValueId]) -> Self {
        Self(SmallVec::from_slice(links))
    }

    /// Returns the links as a positional slice.
    pub(crate) fn as_slice(&self) -> &[ValueId] {
        &self.0
    }
}