Struct GraphBuilder

pub struct GraphBuilder { /* private fields */ }

Builder for creating a new computation graph.

Implementations

impl GraphBuilder

pub fn new(ty: EngineType) -> Result<Self, LlamaCoreError>

Create a new computation graph builder.

pub fn with_config(self, metadata: &Metadata) -> Result<Self, LlamaCoreError>

pub fn use_cpu(self) -> Self

pub fn use_gpu(self) -> Self

pub fn use_tpu(self) -> Self

pub fn build_from_buffer<B>( self, bytes_array: impl AsRef<[B]>, ) -> Result<Graph, LlamaCoreError>

where B: AsRef<[u8]>,

pub fn build_from_files<P>( self, files: impl AsRef<[P]>, ) -> Result<Graph, LlamaCoreError>

where P: AsRef<Path>,

pub fn build_from_cache(self) -> Result<Graph, LlamaCoreError>

Trait Implementations

impl Debug for GraphBuilder

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

Formats the value using the given formatter.