Struct GraphBuilder 

pub struct GraphBuilder { /* private fields */ }

Implementations

impl GraphBuilder

pub fn new(limits: Limits) -> Self

pub fn lookup(&self, name: &str) -> Option<ValueId>

pub fn input(&mut self, name: &str, shape: Vec<usize>) -> Result<ValueId, Error>

pub fn constant( &mut self, name: &str, shape: Vec<usize>, data: Vec<f32>, ) -> Result<ValueId, Error>

pub fn constant_shared( &mut self, name: &str, shape: Vec<usize>, data: Arc<[f32]>, ) -> Result<ValueId, Error>

Like constant, but shares the buffer, so rebuilding a graph never copies weights.

pub fn node( &mut self, name: &str, op: Op, inputs: &[ValueId], ) -> Result<ValueId, Error>

pub fn finish(self, outputs: &[ValueId]) -> Result<Graph, Error>

pub fn finish_named( self, outputs: Vec<(String, ValueId)>, ) -> Result<Graph, Error>