ONE - On-device Neural Engine
|
#include <CircleExporter.h>
Public Member Functions | |
CircleExporter (const std::string &source, const std::string &path) | |
~CircleExporter () | |
void | updateWeight (const std::unique_ptr< onert::exec::Execution > &exec) |
void | updateMetadata (const std::unique_ptr< onert::ir::train::TrainingInfo > &training_info) |
Definition at line 44 of file CircleExporter.h.
onert::exporter::CircleExporter::CircleExporter | ( | const std::string & | source, |
const std::string & | path | ||
) |
Definition at line 32 of file CircleExporter.cc.
onert::exporter::CircleExporter::~CircleExporter | ( | ) |
Definition at line 57 of file CircleExporter.cc.
void onert::exporter::CircleExporter::updateMetadata | ( | const std::unique_ptr< onert::ir::train::TrainingInfo > & | training_info | ) |
Definition at line 99 of file CircleExporter.cc.
References onert::exporter::TrainInfoBuilder::get(), and onert::exporter::TrainInfoBuilder::size().
Referenced by nnfw_session::train_export_circleplus().
void onert::exporter::CircleExporter::updateWeight | ( | const std::unique_ptr< onert::exec::Execution > & | exec | ) |
Definition at line 59 of file CircleExporter.cc.
References onert::util::Index< T, DummyTag >::valid(), and onert::util::Index< T, DummyTag >::value().
Referenced by nnfw_session::train_export_circleplus().