Struct Stream 

pub struct Stream { /* private fields */ }

A computation stream binding a graph to a backend.

Operations on tensors add nodes to the stream’s graph lazily. Calling eval() topologically sorts and evaluates pending nodes.

Implementations

impl Stream

pub fn new(backend: Box<dyn Backend>) -> Self

Create a new stream with the given backend.

pub fn add_constant(&self, data: Vec<f32>, meta: TensorMeta) -> NodeId

Add a constant node (data already known).

pub fn add_op( &self, op: OpKind, inputs: SmallVec<[NodeId; 2]>, meta: TensorMeta, ) -> NodeId

Add an operation node to the graph.

pub fn eval(&self, output: NodeId) -> Result<()>

Evaluate all nodes needed to materialize the given output.

pub fn get_buffer(&self, id: NodeId) -> Option<Vec<f32>>

Get materialized buffer data for a node (must call eval first).

pub fn get_node(&self, id: NodeId) -> Option<Node>

Get a clone of a graph node by ID.

pub fn topo_sort(&self, outputs: &[NodeId]) -> Vec<NodeId>

Topological sort of the subgraph rooted at the given outputs.

pub fn eval_count(&self) -> u64

Number of times the backend’s eval_node has been called.

pub fn cache_len(&self) -> usize

Number of materialized buffers currently cached.

Trait Implementations

impl Debug for Stream

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.