Struct menoh::Builder [−][src]
pub struct Builder { /* fields omitted */ }
Helper to build Model
.
Methods
impl Builder
[src]
impl Builder
pub fn from_onnx<P>(path: P) -> Result<Self, Error> where
P: AsRef<Path>,
[src]
pub fn from_onnx<P>(path: P) -> Result<Self, Error> where
P: AsRef<Path>,
Create a builder from a ONNX file.
let builder = Builder::from_onnx("MLP.onnx")?;
pub fn add_input<T>(self, name: &str, dims: &[usize]) -> Result<Self, Error> where
T: Dtype,
[src]
pub fn add_input<T>(self, name: &str, dims: &[usize]) -> Result<Self, Error> where
T: Dtype,
Register a variable as input.
let builder = builder.add_input::<f32>("input", &[2, 3])?;
pub fn add_output<T>(self, name: &str) -> Result<Self, Error> where
T: Dtype,
[src]
pub fn add_output<T>(self, name: &str) -> Result<Self, Error> where
T: Dtype,
Register a variable as output.
let builder = builder.add_output::<f32>("fc2")?;
pub fn build(self, backend: &str, backend_config: &str) -> Result<Model, Error>
[src]
pub fn build(self, backend: &str, backend_config: &str) -> Result<Model, Error>
Build a Model
.
let model = builder.build("mkldnn", "")?;