#include <SolverHooks.h>
|
virtual void | onBeginSolver (const std::string &model_path, float q8error, float q16error)=0 |
| called on the start of mpq search
|
|
virtual void | onBeginIteration ()=0 |
| called on the start of current iteration
|
|
virtual void | onEndIteration (const LayerParams &layers, const std::string &def_dtype, float error)=0 |
| called at the end of current iteration
|
|
virtual void | onEndSolver (const LayerParams &layers, const std::string &def_dtype, float qerror)=0 |
| called at the end of mpq search
|
|
Definition at line 31 of file SolverHooks.h.
◆ onBeginIteration()
virtual void mpqsolver::core::SolverHooks::onBeginIteration |
( |
| ) |
|
|
pure virtual |
◆ onBeginSolver()
virtual void mpqsolver::core::SolverHooks::onBeginSolver |
( |
const std::string & |
model_path, |
|
|
float |
q8error, |
|
|
float |
q16error |
|
) |
| |
|
pure virtual |
called on the start of mpq search
- Parameters
-
model_path | path of original float model to quantize |
q8error | error of Q8 quantization (if NAN, then not applicable) |
q16error | error of Q16 quantization (if NAN, then not applicable) |
Implemented in mpqsolver::core::DumpingHooks.
◆ onEndIteration()
virtual void mpqsolver::core::SolverHooks::onEndIteration |
( |
const LayerParams & |
layers, |
|
|
const std::string & |
def_dtype, |
|
|
float |
error |
|
) |
| |
|
pure virtual |
called at the end of current iteration
- Parameters
-
layers | model nodes with specific quantization parameters |
def_dtype | default quantization dtype |
error | error of quantization for current iteration |
Implemented in mpqsolver::core::DumpingHooks.
◆ onEndSolver()
virtual void mpqsolver::core::SolverHooks::onEndSolver |
( |
const LayerParams & |
layers, |
|
|
const std::string & |
def_dtype, |
|
|
float |
qerror |
|
) |
| |
|
pure virtual |
called at the end of mpq search
- Parameters
-
layers | model nodes with specific quantization parameters |
def_dtype | default quantization dtype |
qerror | final error of quantization (if NAN, then not applicable) |
Implemented in mpqsolver::core::DumpingHooks.
The documentation for this class was generated from the following file: