record_minmax::MovingAvgComputer Class Reference

#include <MinMaxComputer.h>

Collaboration diagram for record_minmax::MovingAvgComputer:

Public Member Functions
	MovingAvgComputer (uint32_t batch_size, float update_const)
virtual void	update_qparam (const std::unordered_map< const luci::CircleNode *, MinMaxVectors > *minmax_map)
Public Member Functions inherited from record_minmax::MinMaxComputer
	MinMaxComputer ()
virtual	~MinMaxComputer ()=default

Detailed Description

Definition at line 61 of file MinMaxComputer.h.

Constructor & Destructor Documentation

MovingAvgComputer()

record_minmax::MovingAvgComputer::MovingAvgComputer ( uint32_t  batch_size, float  update_const  )

inline

Definition at line 64 of file MinMaxComputer.h.

    : _batch_size(batch_size), _update_const(update_const)
  {
  }

Member Function Documentation

update_qparam()

void record_minmax::MovingAvgComputer::update_qparam ( const std::unordered_map< const luci::CircleNode *, MinMaxVectors > *  minmax_map )

virtual

Implements record_minmax::MinMaxComputer.

Definition at line 50 of file MinMaxComputer.cpp.

{
  if (minmax_map == nullptr)
    throw std::invalid_argument("minmax_map is nullptr");
 
  for (auto iter = minmax_map->begin(); iter != minmax_map->end(); ++iter)
  {
    auto node = iter->first;
    auto minmax = iter->second;
 
    auto min = getMovingAverage(minmax.min_vector, 1 - _update_const, _batch_size, true);
    auto max = getMovingAverage(minmax.max_vector, 1 - _update_const, _batch_size, false);
 
    auto quantparam = std::make_unique<luci::CircleQuantParam>();
    quantparam->min.push_back(min);
    quantparam->max.push_back(max);
 
    assert(node->quantparam() == nullptr);
 
    auto mutable_node = const_cast<luci::CircleNode *>(node);
    mutable_node->quantparam(std::move(quantparam));
  }
}

References record_minmax::getMovingAverage(), and luci::CircleNode::quantparam().

---

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

• compiler/record-minmax/include/MinMaxComputer.h
• compiler/record-minmax/src/MinMaxComputer.cpp