ONE - On-device Neural Engine
|
Data Structures | |
struct | DataBuffer |
class | DataSetIterator |
class | DirectoryIterator |
class | HDF5Iterator |
class | ListFileIterator |
class | MinMaxComputer |
class | MinMaxMap |
class | MinMaxObserver |
struct | MinMaxVectors |
class | MovingAvgComputer |
class | PercentileComputer |
class | RandomIterator |
class | RecordMinMax |
Typedefs | |
using | Buffer = std::vector< char > |
using | Output = std::vector< Buffer > |
using | WholeOutput = std::vector< Output > |
Functions | |
std::unique_ptr< MinMaxComputer > | make_percentile_computer (float min_percentile, float max_percentile) |
std::unique_ptr< MinMaxComputer > | make_moving_avg_computer (uint32_t batch_size, float moving_avg_const) |
float | getNthPercentile (std::vector< float > &vector, float percentile) |
getNthPercentile calculates the n-th percentile of input vector (0.0 <= n <= 100.0) linear interpolation is used when the desired percentile lies between two data points | |
float | getMovingAverage (const std::vector< float > &vector, const float alpha, const uint8_t batch_size, bool is_min) |
getMovingAverage calculates the weighted moving average of input vector The initial value is the minimum (or maximum) value of the first batch of the vector | |
uint32_t | numElements (const luci::CircleNode *node) |
size_t | getTensorSize (const luci::CircleNode *node) |
void | readDataFromFile (const std::string &filename, std::vector< char > &data, size_t data_size) |
void | checkInputDimension (const luci::CircleInput *input) |
using record_minmax::Buffer = typedef std::vector<char> |
Definition at line 33 of file RecordMinMax.h.
using record_minmax::Output = typedef std::vector<Buffer> |
Definition at line 34 of file RecordMinMax.h.
using record_minmax::WholeOutput = typedef std::vector<Output> |
Definition at line 35 of file RecordMinMax.h.
void record_minmax::checkInputDimension | ( | const luci::CircleInput * | input | ) |
Definition at line 29 of file Utils.cpp.
References numElements().
Referenced by record_minmax::RecordMinMax::profileData().
float record_minmax::getMovingAverage | ( | const std::vector< float > & | vector, |
const float | alpha, | ||
const uint8_t | batch_size, | ||
bool | is_min | ||
) |
getMovingAverage calculates the weighted moving average of input vector The initial value is the minimum (or maximum) value of the first batch of the vector
Definition at line 60 of file RecordFunction.cpp.
References offset().
Referenced by record_minmax::MovingAvgComputer::update_qparam().
float record_minmax::getNthPercentile | ( | std::vector< float > & | vector, |
float | percentile | ||
) |
getNthPercentile calculates the n-th percentile of input vector (0.0 <= n <= 100.0) linear interpolation is used when the desired percentile lies between two data points
Definition at line 30 of file RecordFunction.cpp.
Referenced by record_minmax::PercentileComputer::update_qparam().
size_t record_minmax::getTensorSize | ( | const luci::CircleNode * | node | ) |
Definition at line 57 of file Utils.cpp.
References numElements(), and luci::size().
Referenced by record_minmax::DirectoryIterator::next(), record_minmax::HDF5Iterator::next(), record_minmax::ListFileIterator::next(), and record_minmax::RandomIterator::next().
std::unique_ptr< MinMaxComputer > record_minmax::make_moving_avg_computer | ( | uint32_t | batch_size, |
float | moving_avg_const | ||
) |
Definition at line 80 of file MinMaxComputer.cpp.
std::unique_ptr< MinMaxComputer > record_minmax::make_percentile_computer | ( | float | min_percentile, |
float | max_percentile | ||
) |
Definition at line 75 of file MinMaxComputer.cpp.
uint32_t record_minmax::numElements | ( | const luci::CircleNode * | node | ) |
Definition at line 41 of file Utils.cpp.
References luci::CircleNode::name().
Referenced by checkInputDimension(), getTensorSize(), and record_minmax::RandomIterator::next().
void record_minmax::readDataFromFile | ( | const std::string & | filename, |
std::vector< char > & | data, | ||
size_t | data_size | ||
) |
Definition at line 65 of file Utils.cpp.
Referenced by record_minmax::DirectoryIterator::next(), and record_minmax::ListFileIterator::next().