Public Member Functions | Protected Member Functions | Protected Attributes
ngraph::onnx_import::Graph Class Reference
Inheritance diagram for ngraph::onnx_import::Graph:
Inheritance graph
[legend]

Public Member Functions

 Graph (const ONNX_NAMESPACE::GraphProto &proto, Model &model)
 
const std::vector< Node > & get_nodes () const
 
const std::vector< ValueInfo > & get_inputs () const
 
const std::vector< ValueInfo > & get_outputs () const
 
OutputVector get_ng_outputs () const
 
const ParameterVector & get_ng_parameters () const
 
bool is_node_in_cache (const std::string &name) const
 
Output< ngraph::Nodeget_ng_node_from_cache (const std::string &name) const
 
const std::string & get_name () const
 
OutputVector make_ng_nodes (const Node &onnx_node) const
 
const GraphCacheget_graph_cache () const
 

Protected Member Functions

 Graph (const ONNX_NAMESPACE::GraphProto &proto, Model &model, std::unique_ptr< GraphCache > &&cache)
 
void set_friendly_names (const Node &onnx_node, const OutputVector &ng_node_vector) const
 
void add_provenance_tag_to_initializer (const Tensor &initializer, std::shared_ptr< default_opset::Constant > node) const
 
void add_provenance_tag_to_input (const ValueInfo &input, std::shared_ptr< ngraph::Node > node) const
 
void add_provenance_tags (const Node &onnx_node, const OutputVector &ng_node_vector) const
 

Protected Attributes

ParameterVector m_parameters
 

The documentation for this class was generated from the following file: