ONE - On-device Neural Engine
•All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
Public Member Functions | Protected Attributes
onert::compiler::train::UntrainableOperationConverter Class Reference

#include <UntrainableOperationConverter.h>

Collaboration diagram for onert::compiler::train::UntrainableOperationConverter:
Collaboration graph
[legend]

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
 

Detailed Description

Definition at line 29 of file UntrainableOperationConverter.h.

Constructor & Destructor Documentation

◆ UntrainableOperationConverter()

onert::compiler::train::UntrainableOperationConverter::UntrainableOperationConverter ( ir::train::TrainableGraph &  tgraph)

Definition at line 24 of file UntrainableOperationConverter.cc.

25 : _tgraph{tgraph}, _return_op{nullptr}
26{
27}
onert::compiler::train::UntrainableOperationConverter::_return_op
std::unique_ptr< ir::train::ITrainableOperation > _return_op
Definition UntrainableOperationConverter.h:41

Member Function Documentation

◆ operator()()

std::unique_ptr< ir::train::ITrainableOperation > onert::compiler::train::UntrainableOperationConverter::operator() ( const ir::IOperation &  op)

Definition at line 30 of file UntrainableOperationConverter.cc.

31{
32 op.accept(*this);
33
34 return std::move(_return_op);
35}

References _return_op, and onert::ir::IOperation::accept().

Field Documentation

◆ _return_op

std::unique_ptr<ir::train::ITrainableOperation> onert::compiler::train::UntrainableOperationConverter::_return_op
protected

Definition at line 41 of file UntrainableOperationConverter.h.

Referenced by operator()().

◆ _tgraph

ir::train::TrainableGraph& onert::compiler::train::UntrainableOperationConverter::_tgraph
protected

Definition at line 40 of file UntrainableOperationConverter.h.

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