record_minmax::MinMaxMap Class Reference

#include <MinMaxObserver.h>

Public Member Functions
void	recordMinMax (const luci::CircleNode *node, float min, float max)
void	appendMinMaxVector (const luci::CircleNode *node, const MinMaxVectors &minmax_vector)
const std::unordered_map< const luci::CircleNode *, MinMaxVectors > *	getMap () const

Detailed Description

Definition at line 31 of file MinMaxObserver.h.

Member Function Documentation

appendMinMaxVector()

void record_minmax::MinMaxMap::appendMinMaxVector ( const luci::CircleNode *  node, const MinMaxVectors &  minmax_vector  )

inline

Definition at line 42 of file MinMaxObserver.h.

  {
    MinMaxVectors &vectors = _minmax_map[node];
    vectors.min_vector.insert(vectors.min_vector.end(), minmax_vector.min_vector.begin(),
                              minmax_vector.min_vector.end());
    vectors.max_vector.insert(vectors.max_vector.end(), minmax_vector.max_vector.begin(),
                              minmax_vector.max_vector.end());
  }

References record_minmax::MinMaxVectors::max_vector, and record_minmax::MinMaxVectors::min_vector.

Referenced by record_minmax::RecordMinMax::profileDataInParallel().

getMap()

const std::unordered_map< const luci::CircleNode *, MinMaxVectors > * record_minmax::MinMaxMap::getMap ( ) const

inline

Definition at line 51 of file MinMaxObserver.h.

  {
    return &_minmax_map;
  }

Referenced by record_minmax::RecordMinMax::profileDataInParallel().

recordMinMax()

void record_minmax::MinMaxMap::recordMinMax ( const luci::CircleNode *  node, float  min, float  max  )

inline

Definition at line 35 of file MinMaxObserver.h.

  {
    MinMaxVectors &vectors = _minmax_map[node];
    vectors.min_vector.push_back(min);
    vectors.max_vector.push_back(max);
  }

References record_minmax::MinMaxVectors::max_vector, and record_minmax::MinMaxVectors::min_vector.

Referenced by record_minmax::MinMaxObserver::postTensorWrite().

---

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

• compiler/record-minmax/include/MinMaxObserver.h