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

#include <Pass.h>

Collaboration diagram for onert::compiler::train::pass::Pass:

Public Member Functions

 Pass (ir::train::TrainableGraph &trainable_graph, const ir::train::TrainingInfo *training_info)
 
virtual ~Pass ()=default
 
- Public Member Functions inherited from onert::compiler::pass::IPass
virtual ~IPass ()=default
 
virtual std::string id ()=0
 
virtual void run ()=0
 

Protected Attributes

ir::train::TrainableGraph_trainable_graph
 
const ir::train::TrainingInfo_training_info
 

Detailed Description

Definition at line 43 of file Pass.h.

Constructor & Destructor Documentation

◆ Pass()

onert::compiler::train::pass::Pass::Pass ( ir::train::TrainableGraph trainable_graph,
const ir::train::TrainingInfo training_info 
)
inline

Definition at line 46 of file Pass.h.

47 : _trainable_graph{trainable_graph}, _training_info{training_info}
48 {
49 }
ir::train::TrainableGraph & _trainable_graph
Definition Pass.h:53
const ir::train::TrainingInfo * _training_info
Definition Pass.h:54

◆ ~Pass()

virtual onert::compiler::train::pass::Pass::~Pass ( )
virtualdefault

Field Documentation

◆ _trainable_graph

ir::train::TrainableGraph& onert::compiler::train::pass::Pass::_trainable_graph
protected

Definition at line 53 of file Pass.h.

Referenced by onert::compiler::train::pass::LossInsertionPass::run().

◆ _training_info

const ir::train::TrainingInfo* onert::compiler::train::pass::Pass::_training_info
protected

Definition at line 54 of file Pass.h.

Referenced by onert::compiler::train::pass::LossInsertionPass::run().


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