candela-tensor 0.2.0

A lazy, graph-based tensor engine 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

mod alias;
mod owned;
mod plan;
mod runtime;
mod sort;

pub(crate) use owned::{OwnedComputeKind, OwnedCorePlan, from_borrowed_core_to_owned};
pub(crate) use plan::ComputeKind;
pub(crate) use plan::OutputKind;
pub(crate) use plan::{core_plan_computation, plan_computation};

use crate::tensor::backend::Backend;
use crate::tensor::graph::NodeKind;

#[inline]
pub(crate) fn get_id<T, B: Backend>(node: &NodeKind<T, B>) -> usize {
    match node {
        NodeKind::Edge(edge) => edge.id,
        NodeKind::Slot(slot) => slot.id,
        NodeKind::Node(node) => node.id,
        NodeKind::Cache(cache) => cache.get_node().id,
        NodeKind::Baked(baked) => baked.id,
    }
}