Searched refs:GraphBuilder (Results 1 – 6 of 6) sorted by relevance
| /wasmtime-44.0.1/crates/test-programs/src/ |
| H A D | nn.rs | 89 use wasi_nn::{Graph, GraphBuilder, TensorType};
97 Ok(GraphBuilder::new(encoding, target).build_from_bytes(bytes)?) in load()
106 Ok(GraphBuilder::new(encoding, target).build_from_cache(name)?) in load_by_name()
|
| /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/wasi-nn/examples/classification-component-onnx/src/ |
| H A D | main.rs | 16 graph::{Graph, GraphBuilder, load, ExecutionTarget, GraphEncoding},
52 let model: GraphBuilder = fs::read("fixture/models/squeezenet1.1-7.onnx").unwrap(); 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/src/ |
| H A D | wit.rs | 141 pub use generated::graph::{ExecutionTarget, Graph, GraphBuilder, GraphEncoding};
145 pub use generated::graph::{ExecutionTarget, Graph, GraphBuilder, GraphEncoding};
172 builders: Vec<GraphBuilder>, in load() argument
|
| /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()
|