|
const std::string & | Name () const noexcept |
|
const std::string & | Description () const noexcept |
|
const Path & | ModelPath () const |
|
bool | IsSubgraph () const |
|
const Graph * | ParentGraph () const |
|
Graph * | MutableParentGraph () |
|
bool | StrictShapeTypeInference () const |
|
void | SetName (const std::string &name) |
|
void | SetDescription (const std::string &description) |
|
common::Status | ReplaceInitializedTensor (ONNX_NAMESPACE::TensorProto new_initializer) |
|
common::Status | InjectExternalInitializedTensors (const InlinedHashMap< std::string, OrtValue > &external_initializers) |
|
void | AddInitializedTensor (const ONNX_NAMESPACE::TensorProto &tensor_proto) |
|
void | RemoveInitializedTensor (const std::string &tensor_name) |
|
bool | IsInitializedTensor (const std::string &name) const |
|
bool | IsSparseInitializer (const std::string &name) const |
|
bool | GetInitializedTensor (const std::string &tensor_name, const ONNX_NAMESPACE::TensorProto *&value) const |
|
const InitializedTensorSet & | GetAllInitializedTensors () const noexcept |
|
void | CleanAllInitializedTensors () noexcept |
|
bool | CanOverrideInitializer () const noexcept |
|
const ONNX_NAMESPACE::TensorProto * | GetConstantInitializer (const std::string &name, bool check_outer_scope) const |
|
const ONNX_NAMESPACE::TensorProto * | GetInitializer (const std::string &name, bool check_outer_scope) const |
|
const std::vector< const
NodeArg * > & | GetInputs () const noexcept |
|
const std::vector< const
NodeArg * > & | GetInputsIncludingInitializers () const noexcept |
|
bool | IsInputsIncludingInitializers (const NodeArg *node_arg) const noexcept |
|
const std::vector< const
NodeArg * > & | GetOverridableInitializers () const |
|
const std::vector< const
NodeArg * > & | GetOutputs () const noexcept |
|
bool | IsOutput (const NodeArg *node_arg) const noexcept |
|
bool | NodeProducesGraphOutput (const Node &node) const |
|
std::vector< int > | GetNodeOutputsInGraphOutputs (const Node &node) const |
|
const std::unordered_set
< const NodeArg * > & | GetValueInfo () const noexcept |
|
void | AddValueInfo (const NodeArg *new_value_info) |
|
const Node * | GetNode (NodeIndex node_index) const |
|
Node * | GetNode (NodeIndex node_index) |
|
GraphNodes & | Nodes () noexcept |
|
const GraphNodes & | Nodes () const noexcept |
|
ConstGraphNodes | FilteredNodes (GraphNodes::NodeFilterFunc &&filter_func) const noexcept |
|
int | MaxNodeIndex () const noexcept |
|
int | NumberOfNodes () const noexcept |
|
NodeArg * | GetNodeArg (const std::string &name) |
|
const NodeArg * | GetNodeArg (const std::string &name) const |
|
NodeArg * | GetNodeArgIncludingParentGraphs (const std::string &node_arg_name) |
|
NodeArg & | GetOrCreateNodeArg (const std::string &name, const ONNX_NAMESPACE::TypeProto *p_arg_type) |
|
std::string | GenerateNodeArgName (const std::string &base_name) |
|
std::string | GenerateNodeName (const std::string &base_name) |
|
Node & | AddNode (const Node &other) |
|
Node & | AddNode (const std::string &name, const std::string &op_type, const std::string &description, gsl::span< NodeArg *const > input_args, gsl::span< NodeArg *const > output_args, const NodeAttributes *attributes=nullptr, const std::string &domain=kOnnxDomain) |
|
Node & | AddNode (const std::string &name, const std::string &op_type, const std::string &description, std::initializer_list< NodeArg * > input_args, std::initializer_list< NodeArg * > output_args, const NodeAttributes *attributes=nullptr, const std::string &domain=kOnnxDomain) |
|
Node & | AddNode (const std::string &name, const std::string &op_type, const std::string &description, gsl::span< NodeArg *const > input_args, std::initializer_list< NodeArg * > output_args, const NodeAttributes *attributes=nullptr, const std::string &domain=kOnnxDomain) |
|
Node & | AddNode (const std::string &name, const std::string &op_type, const std::string &description, std::initializer_list< NodeArg * > input_args, gsl::span< NodeArg *const > output_args, const NodeAttributes *attributes=nullptr, const std::string &domain=kOnnxDomain) |
|
bool | RemoveNode (NodeIndex node_index) |
|
void | AddEdge (NodeIndex src_node_index, NodeIndex dst_node_index, int src_arg_index, int dst_arg_index) |
|
void | RemoveEdge (NodeIndex src_node_index, NodeIndex dst_node_index, int src_arg_index, int dst_arg_index) |
|
bool | AddControlEdge (NodeIndex src_node_index, NodeIndex dst_node_index) |
|
Graph & | SetGraphResolveNeeded () noexcept |
|
bool | GraphResolveNeeded () const noexcept |
|
Graph & | SetGraphProtoSyncNeeded () noexcept |
|
bool | GraphProtoSyncNeeded () const noexcept |
|
void | ReverseDFSFrom (gsl::span< NodeIndex const > from, const std::function< void(const Node *)> &enter, const std::function< void(const Node *)> &leave, const std::function< bool(const Node *, const Node *)> &comp={}) const |
|
void | ReverseDFSFrom (gsl::span< const Node *const > from, const std::function< void(const Node *)> &enter, const std::function< void(const Node *)> &leave, const std::function< bool(const Node *, const Node *)> &comp={}) const |
|
void | ReverseDFSFrom (gsl::span< const Node *const > from, const std::function< void(const Node *)> &enter, const std::function< void(const Node *)> &leave, const std::function< bool(const Node *, const Node *)> &comp, const std::function< bool(const Node *, const Node *)> &stop) const |
|
void | KahnsTopologicalSort (const std::function< void(const Node *)> &enter, const std::function< bool(const Node *, const Node *)> &comp) const |
|
const std::unordered_map
< std::string, int > & | DomainToVersionMap () const noexcept |
|
Node & | BeginFuseSubGraph (const IndexedSubGraph &sub_graph, const std::string &fused_node_name) |
|
void | FinalizeFuseSubGraph (const IndexedSubGraph &sub_graph, Node &fused_node) |
|
const ONNX_NAMESPACE::GraphProto & | ToGraphProto () |
|
ONNX_NAMESPACE::GraphProto | ToGraphProto () const |
|
ONNX_NAMESPACE::GraphProto | ToGraphProtoWithExternalInitializers (const std::string &external_file_name, size_t initializer_size_threshold) const |
|
IOnnxRuntimeOpSchemaCollectionPtr | GetSchemaRegistry () const |
|
bool | SetOpSchemaFromRegistryForNode (Node &node) |
|
Node & | FuseSubGraph (const IndexedSubGraph &sub_graph, const std::string &fused_node_name) |
|
Status | InlineFunction (Node &node) |
|
void | AddOuterScopeNodeArg (const std::string &name) |
|
void | SetInputs (gsl::span< const NodeArg *const > inputs) |
|
void | SetInputs (std::initializer_list< const NodeArg * > inputs) |
|
const Model & | GetModel () const |
|
const logging::Logger & | GetLogger () const |
|
void | SetOutputs (gsl::span< const NodeArg *const > outputs) |
|
void | SetOutputs (std::initializer_list< const NodeArg * > outputs) |
|
void | SetNodeArgType (NodeArg &arg, const ONNX_NAMESPACE::TypeProto &type_proto) |
|
const Node * | GetProducerNode (const std::string &node_arg_name) const |
|
Node * | GetMutableProducerNode (const std::string &node_arg_name) |
|
void | UpdateProducerNode (const std::string &node_arg_name, NodeIndex node_index) |
|
std::vector< const Node * > | GetConsumerNodes (const std::string &node_arg_name) const |
|
void | AddConsumerNode (const std::string &node_arg_name, Node *consumer) |
|
void | RemoveConsumerNode (const std::string &node_arg_name, Node *consumer) |
|
std::vector< Node * > | GetMutableConsumerNodes (const std::string &node_arg_name) |
|
void | UpdateConsumerNodes (const std::string &node_arg_name, gsl::span< Node *const > nodes) |
|
void | UpdateConsumerNodes (const std::string &node_arg_name, std::initializer_list< Node * > nodes) |
|
Status | UpdateShapeInference (Node &node) |
|
common::Status | Resolve (const ResolveOptions &options) |
|
common::Status | Resolve () |
|
const std::unordered_set
< std::string > & | GetOuterScopeNodeArgNames () const noexcept |
|
common::Status | SaveToOrtFormat (flatbuffers::FlatBufferBuilder &builder, flatbuffers::Offset< onnxruntime::fbs::Graph > &fbs_graph) const |
|
const Node * | ParentNode () const |
|
bool | IsOuterScopeValue (const std::string &name) const |
|
| Graph (Graph &parent_graph, const Node &parent_node, ONNX_NAMESPACE::GraphProto &subgraph_proto) |
|
| Graph (const Model &owning_model, IOnnxRuntimeOpSchemaCollectionPtr schema_registry, ONNX_NAMESPACE::GraphProto &subgraph_proto, const std::unordered_map< std::string, int > &domain_version_map, const logging::Logger &logger, bool strict_shape_type_inference) |
|
virtual | ~Graph () |
|
const
RuntimeOptimizationRecordContainer & | RuntimeOptimizations () const |
|
RuntimeOptimizationRecordContainer & | MutableRuntimeOptimizations () |
|
| Graph ()=delete |
|
| Graph (const Model &owning_model, const std::unordered_map< std::string, int > &domain_to_version, IOnnxRuntimeOpSchemaCollectionPtr schema_registry, Graph *parent_graph, const Node *parent_node, const logging::Logger &logger, bool strict_shape_type_inference) |
|
Status | LoadFromOrtFormat (const onnxruntime::fbs::Graph &fbs_graph, const OrtFormatLoadOptions &load_options) |
|
| Graph (const Model &owning_model, ONNX_NAMESPACE::GraphProto *graph_proto, const std::unordered_map< std::string, int > &domain_to_version, Version ir_version, IOnnxRuntimeOpSchemaCollectionPtr schema_registry, const logging::Logger &logger, bool strict_shape_type_inference) |
|
| Graph (const Model &owning_model, ONNX_NAMESPACE::GraphProto *graph_proto, const std::unordered_map< std::string, int > &domain_to_version, Version ir_version, IOnnxRuntimeOpSchemaCollectionPtr schema_registry, Graph *parent_graph, const Node *parent_node, const logging::Logger &logger, bool strict_shape_type_inference) |
|
| ORT_DISALLOW_COPY_ASSIGNMENT_AND_MOVE (Graph) |
|
The Graph representation containing the graph inputs and outputs, the Node instances, and the edges connecting the nodes.
Definition at line 659 of file graph.h.