HDK
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
Classes | Public Types | Public Member Functions | Public Attributes | List of all members
onnxruntime::IndexedSubGraph Class Reference

#include <indexed_sub_graph.h>

Classes

struct  MetaDef
 

Public Types

enum  SourceOfSchema : uint8_t { SourceOfSchema::CREATE, SourceOfSchema::REUSE_OR_CREATE, SourceOfSchema::EXISTING }
 

Public Member Functions

void SetMetaDef (std::unique_ptr< MetaDef > &&meta_def)
 
const MetaDef * GetMetaDef () const
 

Public Attributes

std::vector
< onnxruntime::NodeIndex > 		nodes
 
SourceOfSchema schema_source {SourceOfSchema::CREATE}
 

Detailed Description

Class containing information about a subgraph of Nodes from a Graph. It contains a NodeIndex array of the Nodes covered by the subgraph, and the meta definition needed for representing this subgraph as a FunctionProto, which could be serialized/saved to a model file.

Definition at line 26 of file indexed_sub_graph.h.

Member Enumeration Documentation

enum onnxruntime::IndexedSubGraph::SourceOfSchema : uint8_t
strong
Enumerator
CREATE 
REUSE_OR_CREATE 

create new schema from info in IndexedSubGraph instance. schema instance will not be re-usable.

EXISTING 

re-use existing dynamically created schema with matching domain+name. create re-usable schema if one is not found.

Definition at line 49 of file indexed_sub_graph.h.

Member Function Documentation

const MetaDef* onnxruntime::IndexedSubGraph::GetMetaDef ( ) const
inline

Gets the meta definition needed to represent this subgraph as a FunctionProto.

Returns
MetaDef instance if it has been set. nullptr if not.

Definition at line 69 of file indexed_sub_graph.h.

void onnxruntime::IndexedSubGraph::SetMetaDef ( std::unique_ptr< MetaDef > &&  meta_def)
inline

Set the meta definition needed to represent this subgraph as a FunctionProto It's needed IF AND ONLY IF there are multiple indexes contained in nodes.

Definition at line 63 of file indexed_sub_graph.h.

Member Data Documentation

std::vector<onnxruntime::NodeIndex> onnxruntime::IndexedSubGraph::nodes

Nodes covered by this subgraph. The NodeIndex values are from the parent Graph.

Definition at line 47 of file indexed_sub_graph.h.

SourceOfSchema onnxruntime::IndexedSubGraph::schema_source {SourceOfSchema::CREATE}

Definition at line 59 of file indexed_sub_graph.h.

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