#include <TrainingInfo.h>
Definition at line 35 of file TrainingInfo.h.
◆ TrainingInfo() [1/3]
onert::ir::train::TrainingInfo::TrainingInfo |
( |
| ) |
|
|
inline |
Definition at line 38 of file TrainingInfo.h.
39 : _version{0}, _loss_info(), _optimizer_info(), _batch_size(0), _training_step{0},
40 _trainable_ops{}
41 {
42 }
◆ TrainingInfo() [2/3]
onert::ir::train::TrainingInfo::TrainingInfo |
( |
const TrainingInfo & |
| ) |
|
|
default |
◆ TrainingInfo() [3/3]
onert::ir::train::TrainingInfo::TrainingInfo |
( |
TrainingInfo && |
| ) |
|
|
default |
◆ ~TrainingInfo()
onert::ir::train::TrainingInfo::~TrainingInfo |
( |
| ) |
|
|
default |
◆ batchSize()
uint32_t onert::ir::train::TrainingInfo::batchSize |
( |
| ) |
const |
|
inline |
◆ getTrainableOps()
const std::set< OperationIndex > & onert::ir::train::TrainingInfo::getTrainableOps |
( |
| ) |
const |
|
inline |
◆ isValid()
bool onert::ir::train::TrainingInfo::isValid |
( |
| ) |
const |
◆ lossInfo()
const LossInfo & onert::ir::train::TrainingInfo::lossInfo |
( |
| ) |
const |
|
inline |
◆ operator=() [1/2]
◆ operator=() [2/2]
◆ optimizerInfo()
const OptimizerInfo & onert::ir::train::TrainingInfo::optimizerInfo |
( |
| ) |
const |
|
inline |
◆ setBatchSize()
void onert::ir::train::TrainingInfo::setBatchSize |
( |
const uint32_t |
batch_size | ) |
|
|
inline |
◆ setLossInfo()
void onert::ir::train::TrainingInfo::setLossInfo |
( |
const LossInfo & |
loss_info | ) |
|
|
inline |
◆ setOptimizerInfo()
void onert::ir::train::TrainingInfo::setOptimizerInfo |
( |
const OptimizerInfo & |
optimizer_info | ) |
|
|
inline |
Definition at line 61 of file TrainingInfo.h.
61{ _optimizer_info = optimizer_info; }
◆ setTrainableOps()
void onert::ir::train::TrainingInfo::setTrainableOps |
( |
const std::set< OperationIndex > & |
trainable_ops | ) |
|
|
inline |
Definition at line 63 of file TrainingInfo.h.
64 {
65 _trainable_ops = trainable_ops;
66 }
◆ setVersion()
void onert::ir::train::TrainingInfo::setVersion |
( |
const uint32_t |
version | ) |
|
|
inline |
◆ trainingStep() [1/2]
uint32_t & onert::ir::train::TrainingInfo::trainingStep |
( |
| ) |
|
|
inline |
◆ trainingStep() [2/2]
const uint32_t & onert::ir::train::TrainingInfo::trainingStep |
( |
| ) |
const |
|
inline |
◆ version()
uint32_t onert::ir::train::TrainingInfo::version |
( |
| ) |
const |
|
inline |
The documentation for this class was generated from the following files: