Struct wasmedge_tensorflow_interface::Session

pub struct Session { /* fields omitted */ }

The session structure.

Implementations

impl Session

pub fn new<S: AsRef<[u8]>>(model_buf: S, mod_type: ModelType) -> Session

pub fn add_input<T: TensorType>(
    &mut self,
    name: &str,
    tensor_buf: &[T],
    shape: &[i64]
) -> &mut Session

Add input name, dimension, operation index, and input tensor into context.

pub fn add_output(&mut self, name: &str) -> &mut Session

Add output name and operation index into context.

pub fn clear_input(&mut self) -> &mut Session

Clear the set input tensors.

pub fn clear_output(&mut self) -> &mut Session

Clear the set output tensors.

pub fn run(&mut self) -> &mut Session

Run session.

pub fn get_output<T: TensorType>(&self, name: &str) -> Vec<T>

Get output tensor data by name.

Trait Implementations

impl Drop for Session

fn drop(&mut self)

Executes the destructor for this type.