ONE - On-device Neural Engine
|
Class to iterate over operations and calls callback() method. More...
#include <OperationPass.h>
Public Member Functions | |
virtual | ~OperationPass ()=default |
std::string | id () override=0 |
Returns string id for this pass. Same with class name. | |
virtual void | callback (const ir::OperationIndex &index, ir::IOperation &node)=0 |
Be called for all nodes of graph. | |
void | run () final |
Run the pass. | |
Pass (ir::Graph &graph) | |
Public Member Functions inherited from onert::compiler::pass::Pass | |
Pass (ir::Graph &graph) | |
virtual | ~Pass ()=default |
Public Member Functions inherited from onert::compiler::pass::IPass | |
virtual | ~IPass ()=default |
Additional Inherited Members | |
Protected Attributes inherited from onert::compiler::pass::Pass | |
ir::Graph & | _graph |
Class to iterate over operations and calls callback() method.
Definition at line 46 of file OperationPass.h.
|
virtualdefault |
|
pure virtual |
Be called for all nodes of graph.
index | is the index of a node in graph |
node | is the node in graph |
Implemented in onert::compiler::pass::PermutationEliminationPass, onert::compiler::pass::LoweredOperationPass, onert::compiler::pass::ConstantInsertionPass, onert::compiler::pass::ConstantLoweringPass, and onert::compiler::train::pass::TrainableConstantInsertionPass.
Referenced by run().
|
overridepure virtual |
Returns string id for this pass. Same with class name.
Implements onert::compiler::pass::Pass.
Implemented in onert::compiler::pass::ConstantInsertionPass, onert::compiler::pass::ConstantLoweringPass, onert::compiler::pass::PermutationEliminationPass, onert::compiler::train::pass::TrainableConstantInsertionPass, and onert::compiler::pass::LoweredOperationPass.
|
finalvirtual |
Run the pass.
Implements onert::compiler::pass::Pass.
Definition at line 30 of file OperationPass.cc.
References onert::compiler::pass::Pass::_graph, callback(), onert::util::ObjectManager< Index, Object >::iterate(), and onert::ir::Graph::operations().
Referenced by package.infer.session::inference().