Struct menoh::Builder [−][src]
pub struct Builder { /* fields omitted */ }Helper to build Model.
Methods
impl Builder[src]
impl Builderpub 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", "")?;