ONE - On-device Neural Engine
|
#include <TrainableOperationConverter.h>
Public Member Functions | |
TrainableOperationConverter (ir::train::TrainableGraph &trainable_graph, const ir::train::TrainingInfo *training_info) | |
![]() | |
UntrainableOperationConverter (ir::train::TrainableGraph &tgraph) | |
std::unique_ptr< ir::train::ITrainableOperation > | operator() (const ir::IOperation &op) |
![]() | |
virtual | ~OperationVisitor ()=default |
Additional Inherited Members | |
![]() | |
ir::train::TrainableGraph & | _tgraph |
std::unique_ptr< ir::train::ITrainableOperation > | _return_op |
Definition at line 27 of file TrainableOperationConverter.h.
onert::compiler::train::TrainableOperationConverter::TrainableOperationConverter | ( | ir::train::TrainableGraph & | trainable_graph, |
const ir::train::TrainingInfo * | training_info | ||
) |
Definition at line 24 of file TrainableOperationConverter.cc.