ONE - On-device Neural Engine
Loading...
Searching...
No Matches
onert::compiler::train::UntrainableOperationConverter Class Reference

#include <UntrainableOperationConverter.h>

Collaboration diagram for onert::compiler::train::UntrainableOperationConverter:

Public Member Functions

 UntrainableOperationConverter (ir::train::TrainableGraph &tgraph)
 
std::unique_ptr< ir::train::ITrainableOperationoperator() (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 33 of file UntrainableOperationConverter.h.

Constructor & Destructor Documentation

◆ UntrainableOperationConverter()

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

Definition at line 28 of file UntrainableOperationConverter.cc.

29 : _tgraph{tgraph}, _return_op{nullptr}
30{
31}
std::unique_ptr< ir::train::ITrainableOperation > _return_op

Member Function Documentation

◆ operator()()

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

Definition at line 34 of file UntrainableOperationConverter.cc.

35{
36 op.accept(*this);
37
38 return std::move(_return_op);
39}

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 45 of file UntrainableOperationConverter.h.

Referenced by operator()().

◆ _tgraph

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

Definition at line 44 of file UntrainableOperationConverter.h.


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