| /wasmtime-44.0.1/crates/wasi-nn/src/ |
| H A D | witx.rs | 133 target: generated::types::ExecutionTarget, in load() argument
246 impl From<generated::types::ExecutionTarget> for crate::wit::types::ExecutionTarget { implementation
247 fn from(value: generated::types::ExecutionTarget) -> Self { in from()
249 generated::types::ExecutionTarget::Cpu => crate::wit::types::ExecutionTarget::Cpu, in from()
250 generated::types::ExecutionTarget::Gpu => crate::wit::types::ExecutionTarget::Gpu, in from()
251 generated::types::ExecutionTarget::Tpu => crate::wit::types::ExecutionTarget::Tpu, in from()
|
| H A D | wit.rs | 141 pub use generated::graph::{ExecutionTarget, Graph, GraphBuilder, GraphEncoding};
145 pub use generated::graph::{ExecutionTarget, Graph, GraphBuilder, GraphEncoding};
174 target: ExecutionTarget, in load() argument
|
| /wasmtime-44.0.1/crates/wasi-nn/src/backend/ |
| H A D | openvino.rs | 7 use crate::wit::{ExecutionTarget, GraphEncoding, Tensor, TensorType};
23 fn load(&mut self, builders: &[&[u8]], target: ExecutionTarget) -> Result<Graph, BackendError> { in load()
66 target: ExecutionTarget, in load_from_dir() argument
213 impl From<ExecutionTarget> for DeviceType<'static> {
214 fn from(target: ExecutionTarget) -> Self { in from()
216 ExecutionTarget::Cpu => DeviceType::CPU, in from()
217 ExecutionTarget::Gpu => DeviceType::GPU, in from()
218 ExecutionTarget::Tpu => { in from()
|
| H A D | pytorch.rs | 7 use crate::wit::types::{ExecutionTarget, GraphEncoding, Tensor, TensorType};
23 fn load(&mut self, builders: &[&[u8]], target: ExecutionTarget) -> Result<Graph, BackendError> { in load()
56 target: ExecutionTarget, in load_from_dir() argument
74 target: ExecutionTarget,
100 target: ExecutionTarget,
253 fn map_execution_target_to_string(target: ExecutionTarget) -> Device { in map_execution_target_to_string()
255 ExecutionTarget::Cpu => Device::Cpu, in map_execution_target_to_string()
256 ExecutionTarget::Gpu => Device::Cuda(0), in map_execution_target_to_string()
257 ExecutionTarget::Tpu => { in map_execution_target_to_string()
|
| H A D | onnx.rs | 7 use crate::wit::types::{ExecutionTarget, GraphEncoding, Tensor, TensorType};
34 fn load(&mut self, builders: &[&[u8]], target: ExecutionTarget) -> Result<Graph, BackendError> { in load()
59 target: ExecutionTarget, in configure_execution_providers() argument
62 ExecutionTarget::Cpu => { in configure_execution_providers()
67 ExecutionTarget::Gpu => { in configure_execution_providers()
80 ExecutionTarget::Tpu => { in configure_execution_providers()
93 target: ExecutionTarget, in load_from_dir() argument
100 struct OnnxGraph(Arc<Mutex<Session>>, #[allow(dead_code)] ExecutionTarget);
|
| H A D | mod.rs | 23 use crate::wit::{ExecutionTarget, GraphEncoding, Tensor};
57 fn load(&mut self, builders: &[&[u8]], target: ExecutionTarget) -> Result<Graph, BackendError>; in load()
68 target: ExecutionTarget, in load_from_dir() argument
|
| H A D | winml.rs | 14 use crate::wit::{ExecutionTarget, GraphEncoding, Tensor, TensorType};
36 fn load(&mut self, builders: &[&[u8]], target: ExecutionTarget) -> Result<Graph, BackendError> { in load()
50 ExecutionTarget::Cpu => LearningModelDeviceKind::Cpu, in load()
51 ExecutionTarget::Gpu => LearningModelDeviceKind::DirectX, in load()
52 ExecutionTarget::Tpu => unimplemented!(), in load()
69 target: ExecutionTarget, in load_from_dir() argument
|
| /wasmtime-44.0.1/crates/test-programs/src/ |
| H A D | nn.rs | 24 pub use self::wasi::nn::graph::{ExecutionTarget, GraphEncoding}; // Used by tests.
31 target: ExecutionTarget, in load() argument
88 pub use wasi_nn::{ExecutionTarget, GraphEncoding};
95 target: ExecutionTarget, in load() argument
104 target: ExecutionTarget, in load_by_name() argument
|
| /wasmtime-44.0.1/crates/wasi-nn/examples/classification-component-onnx/src/ |
| H A D | main.rs | 16 graph::{Graph, GraphBuilder, load, ExecutionTarget, GraphEncoding},
22 fn get_execution_target() -> ExecutionTarget { in get_execution_target()
31 return ExecutionTarget::Gpu; in get_execution_target()
35 return ExecutionTarget::Cpu; in get_execution_target()
46 ExecutionTarget::Cpu in get_execution_target()
|
| /wasmtime-44.0.1/crates/wasi-nn/src/registry/ |
| H A D | in_memory.rs | 5 use crate::wit::ExecutionTarget;
33 let graph = backend.load_from_dir(path, ExecutionTarget::Cpu)?; in load()
|
| /wasmtime-44.0.1/crates/wasi-nn/examples/classification-example-pytorch/src/ |
| H A D | main.rs | 4 use wasi_nn::{self, ExecutionTarget, GraphBuilder, GraphEncoding};
9 let graph = GraphBuilder::new(GraphEncoding::Pytorch, ExecutionTarget::CPU) in main()
|
| /wasmtime-44.0.1/crates/test-programs/src/bin/ |
| H A D | nn_witx_image_classification_openvino_named.rs | 9 witx::ExecutionTarget::CPU, in main()
|
| H A D | nn_witx_image_classification_winml_named.rs | 9 witx::ExecutionTarget::CPU, in main()
|
| H A D | nn_wit_image_classification_onnx.rs | 11 wit::ExecutionTarget::Cpu, in main()
|
| H A D | nn_witx_image_classification_onnx.rs | 11 witx::ExecutionTarget::CPU, in main()
|
| H A D | nn_wit_image_classification_openvino.rs | 13 wit::ExecutionTarget::Cpu, in main()
|
| H A D | nn_witx_image_classification_openvino.rs | 13 witx::ExecutionTarget::CPU, in main()
|
| H A D | nn_wit_image_classification_pytorch.rs | 11 wit::ExecutionTarget::Cpu, in main()
|
| H A D | nn_witx_image_classification_pytorch.rs | 11 witx::ExecutionTarget::CPU, in main()
|
| /wasmtime-44.0.1/crates/wasi-nn/examples/classification-example-named/src/ |
| H A D | main.rs | 5 let graph = GraphBuilder::new(GraphEncoding::Openvino, ExecutionTarget::CPU) in main()
|
| /wasmtime-44.0.1/crates/wasi-nn/examples/classification-example-winml/src/ |
| H A D | main.rs | 8 wasi_nn::GraphBuilder::new(wasi_nn::GraphEncoding::Onnx, wasi_nn::ExecutionTarget::CPU) in main()
|