ONE - On-device Neural Engine
|
#include <UntrainableOperationConverter.h>
Public Member Functions | |
UntrainableOperationConverter (ir::train::TrainableGraph &tgraph) | |
std::unique_ptr< ir::train::ITrainableOperation > | operator() (const ir::IOperation &op) |
Public Member Functions inherited from onert::ir::OperationVisitor | |
virtual | ~OperationVisitor ()=default |
Protected Attributes | |
ir::train::TrainableGraph & | _tgraph |
std::unique_ptr< ir::train::ITrainableOperation > | _return_op |
Definition at line 33 of file UntrainableOperationConverter.h.
onert::compiler::train::UntrainableOperationConverter::UntrainableOperationConverter | ( | ir::train::TrainableGraph & | tgraph | ) |
Definition at line 28 of file UntrainableOperationConverter.cc.
std::unique_ptr< ir::train::ITrainableOperation > onert::compiler::train::UntrainableOperationConverter::operator() | ( | const ir::IOperation & | op | ) |
Definition at line 34 of file UntrainableOperationConverter.cc.
References _return_op, and onert::ir::IOperation::accept().
|
protected |
Definition at line 45 of file UntrainableOperationConverter.h.
Referenced by operator()().
|
protected |
Definition at line 44 of file UntrainableOperationConverter.h.