Struct wasi_nn::GraphExecutionContext

pub struct GraphExecutionContext<'a> { /* private fields */ }

Bind a Graph to the input and output for an inference.

Implementations

impl<'a> GraphExecutionContext<'a>

pub fn graph(&self) -> &Graph

Get the Graph instance for this GraphExecutionContext instance.

pub fn set_input<T: Sized>( &mut self, index: usize, tensor_type: TensorType, dimensions: &[usize], data: impl AsRef<[T]> ) -> Result<(), Error>

Set input uses the data, not only u8, but also f32, i32, etc.

pub fn compute(&mut self) -> Result<(), Error>

Compute the inference on the given inputs.

pub fn get_output<T: Sized>( &self, index: usize, out_buffer: &mut [T] ) -> Result<usize, Error>

Copy output tensor to out_buffer, return the output’s size in bytes.

Trait Implementations

impl<'a> Debug for GraphExecutionContext<'a>

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

Formats the value using the given formatter.

impl<'a> Display for GraphExecutionContext<'a>

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

Formats the value using the given formatter.