Struct wasi_nn::GraphBuilder

pub struct GraphBuilder { /* private fields */ }

A graph factory for configuring the properties of a new graph.

Methods can be chained on it in order to configure it.

• Default Graph Encoding: Openvino.
• Default Execution Target: CPU.

Examples

Build a graph with default config ( CPU + Openvino )

let xml = "./mobilenet.xml";
let weight = "./mobilenet.bin";
let graph = GraphBuilder::default().build_from_files([xml, weight]).unwrap();

Build a graph with onnx backend and gpu device target

let graph = GraphBuilder::new(GraphEncoding::Onnx, ExecutionTarget::GPU)
    .build_from_files(["./test.onnx"]).unwrap();

Implementations

impl GraphBuilder

pub fn new(encoding: GraphEncoding, target: ExecutionTarget) -> Self

Create a new GraphBuilder.

pub fn encoding(self, encoding: GraphEncoding) -> Self

Set graph encoding.

pub fn execution_target(self, execution_target: ExecutionTarget) -> Self

Set graph execution target.

pub fn cpu(self) -> Self

Set graph execution target to CPU.

pub fn gpu(self) -> Self

Set graph execution target to GPU.

pub fn tpu(self) -> Self

Set graph execution target to TPU.

pub fn build_from_bytes<B>( self, bytes_array: impl AsRef<[B]> ) -> Result<Graph, Error>where B: AsRef<[u8]>,

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

pub fn build_from_cache(self, name: &str) -> Result<Graph, Error>

Trait Implementations

impl Clone for GraphBuilder

fn clone(&self) -> GraphBuilder

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for GraphBuilder

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

Formats the value using the given formatter.

impl Default for GraphBuilder

fn default() -> Self

Returns the “default value” for a type.