HDK
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
onnxruntime::GraphTransformer Class Referenceabstract

#include <graph_transformer.h>

+ Inheritance diagram for onnxruntime::GraphTransformer:

Public Member Functions

 GraphTransformer (const std::string &name, const InlinedHashSet< std::string_view > &compatible_execution_providers={}) noexcept
 
virtual ~GraphTransformer ()=default
 
const std::stringName () const noexcept
 
const InlinedHashSet
< std::string_view > & 
GetCompatibleExecutionProviders () const noexcept
 
common::Status Apply (Graph &graph, bool &modified, const logging::Logger &logger) const
 
virtual bool ShouldOnlyApplyOnce () const
 

Protected Member Functions

common::Status Recurse (Node &node, bool &modified, int graph_level, const logging::Logger &logger) const
 

Detailed Description

The interface for in-place transformation of a Graph.

Definition at line 19 of file graph_transformer.h.

Constructor & Destructor Documentation

onnxruntime::GraphTransformer::GraphTransformer ( const std::string name,
const InlinedHashSet< std::string_view > &  compatible_execution_providers = {} 
)
inlinenoexcept

Definition at line 21 of file graph_transformer.h.

virtual onnxruntime::GraphTransformer::~GraphTransformer ( )
virtualdefault

Member Function Documentation

common::Status onnxruntime::GraphTransformer::Apply ( Graph graph,
bool &  modified,
const logging::Logger logger 
) const

Apply the in-place transformation defined by this transformer to the provided Graph instance.

Parameters
[out]modifiedSet to true if the Graph was modified.
Returns
Status with success or error information.
const InlinedHashSet<std::string_view>& onnxruntime::GraphTransformer::GetCompatibleExecutionProviders ( ) const
inlinenoexcept

Definition at line 33 of file graph_transformer.h.

const std::string& onnxruntime::GraphTransformer::Name ( ) const
inlinenoexcept

Gets the name of this graph transformer.

Definition at line 29 of file graph_transformer.h.

common::Status onnxruntime::GraphTransformer::Recurse ( Node node,
bool &  modified,
int  graph_level,
const logging::Logger logger 
) const
inlineprotected

Helper method to call ApplyImpl on any subgraphs in the Node.

Definition at line 47 of file graph_transformer.h.

virtual bool onnxruntime::GraphTransformer::ShouldOnlyApplyOnce ( ) const
inlinevirtual

Definition at line 43 of file graph_transformer.h.


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