Struct onnx_helpers::builder::Graph

pub struct Graph { /* fields omitted */ }

Graph builder.

Implementations

impl Graph

pub fn new<S: Into<String>>(name: S) -> Self

Creates a new builder.

pub fn name<S: Into<String>>(self, name: S) -> Self

Sets graph name.

pub fn doc_string<S: Into<String>>(self, doc_string: S) -> Self

Sets graph doc_string.

pub fn constant<S: Into<String>, T: Into<TensorProto>>(
    &mut self,
    name: S,
    tensor: T
) -> Node

Creates constant node in a graph.

pub fn concat<I>(&mut self, axis: i64, inputs: I) -> Node where
    I: IntoIterator,
    I::Item: Into<String>, 

Creates a concat node in a graph.

pub fn nodes<T: Into<NodeProto>>(self, node: T) -> Self

Inserts graph nodes.

pub fn inputs<T: Into<ValueInfoProto>>(self, input: T) -> Self

Inserts graph inputs.

pub fn outputs<T: Into<ValueInfoProto>>(self, output: T) -> Self

Inserts graph outputs.

pub fn outputs_typed<T: Into<ValueInfoProto>, D: Into<TypeProto>>(
    self,
    output: T,
    typ: D
) -> Self

Inserts typed graph outputs.

pub fn initializer<T: Into<TensorProto>>(self, initializer: T) -> Self

Inserts graph initializers.

pub fn node<T: Into<String>>(&mut self, name: T) -> Node

Creates graph node builder.

pub fn input<T: Into<String>>(&mut self, name: T) -> Value

Creates graph input builder.

pub fn output<T: Into<String>>(&mut self, name: T) -> Value

Creates graph output builder.

pub fn model(self) -> Model

Builds a model builder from graph.

pub fn build(self) -> GraphProto

Builds the graph.

Trait Implementations

impl Clone for Graph

fn clone(&self) -> Graph

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Default for Graph

fn default() -> Graph

Returns the "default value" for a type.

impl Into<GraphProto> for Graph

fn into(self) -> GraphProto

Performs the conversion.