| /wasmtime-44.0.1/crates/wasi-nn/src/backend/ |
| H A D | onnx.rs | 60 ) -> Result<Vec<ExecutionProviderDispatch>, BackendError> { in configure_execution_providers() argument
94 ) -> Result<Graph, BackendError> { in load_from_dir() argument
185 ) -> Result<Option<Vec<NamedTensor>>, BackendError> { in compute() argument
293 Err(BackendError::BackendAccess(wasmtime::format_err!( in get_output()
301 impl From<ort::Error> for BackendError { implementation
303 BackendError::BackendAccess(wasmtime::format_err!("{e}")) in from()
324 fn from_onnx_input(input: &Input) -> Result<Self, BackendError> { in from_onnx_input() argument
381 _ => Err(BackendError::BackendAccess(wasmtime::format_err!( in convert_value_type()
387 fn convert_i64(i: &i64) -> Result<u32, BackendError> { in convert_i64() argument
388 u32::try_from(*i).map_err(|d| -> BackendError { in convert_i64()
[all …]
|
| H A D | pytorch.rs | 4 BackendError, BackendExecutionContext, BackendFromDir, BackendGraph, BackendInner, Id,
57 ) -> Result<Graph, BackendError> { in load_from_dir() argument
81 fn init_execution_context(&self) -> Result<ExecutionContext, BackendError> { in init_execution_context() argument
161 ) -> Result<Option<Vec<NamedTensor>>, BackendError> { in compute() argument
238 fn get_output(&mut self, _index: Id) -> Result<Tensor, BackendError> { in get_output() argument
263 fn kind_to_size(kind: Kind) -> Result<usize, BackendError> { in kind_to_size() argument
276 type Error = BackendError;
286 _ => Err(BackendError::UnsupportedTensorType(format!( in try_from()
296 type Error = BackendError;
311 impl From<TchError> for BackendError { implementation
[all …]
|
| H A D | openvino.rs | 4 BackendError, BackendExecutionContext, BackendFromDir, BackendGraph, BackendInner, Id,
25 return Err(BackendError::InvalidNumberOfBuilders(2, builders.len())); in load()
67 ) -> Result<Graph, BackendError> { in load_from_dir() argument
115 ) -> Result<Option<Vec<NamedTensor>>, BackendError> { in compute() argument
178 fn get_output(&mut self, id: Id) -> Result<Tensor, BackendError> { in get_output() argument
199 impl From<InferenceError> for BackendError { implementation
201 BackendError::BackendAccess(wasmtime::Error::new(e)) in from()
205 impl From<SetupError> for BackendError { implementation
207 BackendError::BackendAccess(wasmtime::Error::new(e)) in from()
243 type Error = BackendError;
[all …]
|
| H A D | winml.rs | 38 return Err(BackendError::InvalidNumberOfBuilders(1, builders.len())); in load()
70 ) -> Result<Graph, BackendError> { in load_from_dir() argument
115 ) -> Result<u32, BackendError> { in find() argument
131 BackendError::BackendAccess(wasmtime::format_err!( in find()
157 ) -> Result<Option<Vec<NamedTensor>>, BackendError> { in compute() argument
220 fn get_output(&mut self, id: Id) -> Result<Tensor, BackendError> { in get_output() argument
240 return Err(BackendError::BackendAccess(wasmtime::Error::msg( in get_output()
255 impl From<windows::core::Error> for BackendError { implementation
257 BackendError::BackendAccess(wasmtime::Error::new(e)) in from()
268 fn convert_i64(i: i64) -> Result<u32, BackendError> { in convert_i64() argument
[all …]
|
| H A D | mod.rs | 57 fn load(&mut self, builders: &[&[u8]], target: ExecutionTarget) -> Result<Graph, BackendError>; in load() argument
69 ) -> Result<Graph, BackendError>; in load_from_dir() argument
75 fn init_execution_context(&self) -> Result<ExecutionContext, BackendError>; in init_execution_context() argument
82 fn set_input(&mut self, id: Id, tensor: &Tensor) -> Result<(), BackendError>; in set_input() argument
83 fn get_output(&mut self, id: Id) -> Result<Tensor, BackendError>; in get_output() argument
89 ) -> Result<Option<Vec<NamedTensor>>, BackendError>; in compute() argument
116 pub enum BackendError { enum
|
| /wasmtime-44.0.1/crates/wasi-nn/src/ |
| H A D | lib.rs | 6 use crate::backend::{BackendError, Id, NamedTensor as BackendNamedTensor};
131 pub fn set_input(&mut self, id: Id, tensor: &Tensor) -> Result<(), BackendError> { in set_input() argument
135 pub fn compute(&mut self) -> Result<(), BackendError> { in compute() argument
139 pub fn get_output(&mut self, id: Id) -> Result<Tensor, BackendError> { in get_output() argument
146 ) -> Result<Vec<BackendNamedTensor>, BackendError> { in compute_with_io() argument
|
| H A D | witx.rs | 16 use crate::backend::BackendError;
117 WasiNnError::BackendError(_) => Ok(types::NnErrno::RuntimeError), in nn_errno_from_wasi_nn_error()
290 BackendError(#[from] BackendError), enumerator
|