Representation of ONNX subgraph. It is used for example by ONNX Loop op. It has access for initializers both from subgraph and from parent graph cache. More...
#include <graph.hpp>
Public Member Functions | |
| Subgraph (const ONNX_NAMESPACE::GraphProto &proto, Model &model, const Graph &parent_graph) | |
| Subgraph a GraphCache class object. More... | |
 Public Member Functions inherited from ngraph::onnx_import::Graph | |
| 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::Node > | get_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 GraphCache & | get_graph_cache () const |
Additional Inherited Members | |
| Protected Member Functions inherited from ngraph::onnx_import::Graph | |
| 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 inherited from ngraph::onnx_import::Graph | |
| ParameterVector | m_parameters |
Representation of ONNX subgraph. It is used for example by ONNX Loop op. It has access for initializers both from subgraph and from parent graph cache.
| ngraph::onnx_import::Subgraph::Subgraph | ( | const ONNX_NAMESPACE::GraphProto & | proto, |
| Model & | model, | ||
| const Graph & | parent_graph | ||
| ) |
Subgraph a GraphCache class object.
| [in] | proto | The ONNX protobuf graph representation. |
| [in] | model | The ONNX model object. |
| [in] | parent_graph | The reference to the parent graph. |
