Trait host::wasi_ephemeral_nn::WasiEphemeralNn[−][src]

pub trait WasiEphemeralNn {
    fn load<'a>(
        &mut self, 
        builder: &GraphBuilderArray<'a>, 
        encoding: GraphEncoding, 
        target: ExecutionTarget
    ) -> Result<Graph, NnErrno>;

    fn init_execution_context(
        &mut self, 
        graph: Graph
    ) -> Result<GraphExecutionContext, NnErrno>;

    fn set_input<'a>(
        &mut self, 
        context: GraphExecutionContext, 
        index: u32, 
        tensor: &Tensor<'a>
    ) -> Result<(), NnErrno>;

    fn get_output<'a>(
        &mut self, 
        context: GraphExecutionContext, 
        index: u32, 
        out_buffer: &GuestPtr<'a, u8>, 
        out_buffer_max_size: BufferSize
    ) -> Result<BufferSize, NnErrno>;

    fn compute(&mut self, context: GraphExecutionContext) -> Result<(), NnErrno>;
}

Required methods

Implementors