Struct wasmtime_wasi_nn::WasiNn

pub struct WasiNn {
    pub load: Func,
    pub init_execution_context: Func,
    pub set_input: Func,
    pub get_output: Func,
    pub compute: Func,
}

An instantiated instance of the wasi-nn exports.

Fields

load: Func

init_execution_context: Func

set_input: Func

get_output: Func

compute: Func

Implementations

impl WasiNn

pub fn new(store: &Store, ctx: Rc<RefCell<WasiNnCtx>>) -> Self

Creates a new WasiNn instance.

External values are allocated into the store provided and configuration of the instance itself should be all contained in the cx parameter.

pub fn get_export(&self, name: &str) -> Option<&Func>

Looks up a field called name in this structure, returning it if found.

This is often useful when instantiating a wasmtime instance where name resolution often happens with strings.

pub fn add_to_linker(&self, linker: &mut Linker) -> Result<()>

Adds all instance items to the specified Linker.

pub fn add_to_config(config: &mut Config)

Adds the host functions to the given wasmtime::Config.

Host functions added to the config expect [set_context] to be called.

Host functions will trap if the context is not set in the calling wasmtime::Store.

pub fn add_load_to_config(config: &mut Config, module: &str, field: &str)

Add the host function for load to a config under a given module and field name.

pub fn add_init_execution_context_to_config(
    config: &mut Config,
    module: &str,
    field: &str
)

Add the host function for init_execution_context to a config under a given module and field name.

pub fn add_set_input_to_config(config: &mut Config, module: &str, field: &str)

Add the host function for set_input to a config under a given module and field name.

pub fn add_get_output_to_config(config: &mut Config, module: &str, field: &str)

Add the host function for get_output to a config under a given module and field name.

pub fn add_compute_to_config(config: &mut Config, module: &str, field: &str)

Add the host function for compute to a config under a given module and field name.

pub fn set_context(store: &Store, ctx: WasiNnCtx) -> Result<(), WasiNnCtx>

Sets the context in the given store.

Context must be set in the store when using [add_to_config] and prior to any host function being called.

If the context is already set in the store, the given context is returned as an error.