Home
last modified time | relevance | path

Searched refs:exec_context (Results 1 – 3 of 3) sorted by relevance

/wasmtime-44.0.1/crates/wasi-nn/src/
H A Dwit.rs227 Ok(exec_context) => { in init_execution_context()
228 let exec_context = self.table.push(exec_context)?; in init_execution_context() localVariable
229 Ok(Ok(exec_context)) in init_execution_context()
246 exec_context: Resource<GraphExecutionContext>, in compute()
257 let exec_context = &mut self.table.get_mut(&exec_context)?; in compute() localVariable
259 match exec_context.compute_with_io(named_tensors) { in compute()
281 fn drop(&mut self, exec_context: Resource<GraphExecutionContext>) -> wasmtime::Result<()> { in drop()
282 self.table.delete(exec_context)?; in drop()
H A Dwitx.rs175 let exec_context = if let Some(graph) = self.graphs.get_mut(graph_id.into()) { in init_execution_context() localVariable
181 let exec_context_id = self.executions.insert(exec_context); in init_execution_context()
192 if let Some(exec_context) = self.executions.get_mut(exec_context_id.into()) { in set_input()
198 Ok(exec_context.set_input(Id::Index(index), &tensor)?) in set_input()
209 if let Some(exec_context) = self.executions.get_mut(exec_context_id.into()) { in compute()
210 Ok(exec_context.compute()?) in compute()
224 if let Some(exec_context) = self.executions.get_mut(exec_context_id.into()) { in get_output()
225 let tensor = exec_context.get_output(Id::Index(index))?; in get_output()
/wasmtime-44.0.1/crates/wasi-nn/examples/classification-component-onnx/src/
H A Dmain.rs61 let exec_context = Graph::init_execution_context(&graph).unwrap(); in main() localVariable
79 let output_tensor_vec = exec_context.compute(input_tensor).unwrap(); in main()