PerAxisIterator< T, Base > Class Template Reference

PerAxisIterator for per-axis quantization. More...

#include <BaseIterator.hpp>

Inheritance diagram for PerAxisIterator< T, Base >:

Collaboration diagram for PerAxisIterator< T, Base >:

Public Member Functions
	PerAxisIterator (T *data=nullptr, unsigned int axisFactor=0, unsigned int axisDimensionality=0)
	PerAxisIterator (T *data=nullptr, const armnn::TensorShape &tensorShape=TensorShape(), const unsigned int axis=0)
void	Reset (void *data) override
PerAxisIterator &	operator++ () override
PerAxisIterator &	operator+= (const unsigned int increment) override
PerAxisIterator &	operator-= (const unsigned int decrement) override
PerAxisIterator &	SetIndexOnMem (const unsigned int index)
PerAxisIterator &	operator[] (const unsigned int index) override

Protected Attributes
T *	m_Iterator
T *	m_Start
unsigned int	m_AxisIndex
unsigned int	m_AxisDimensionality
unsigned int	m_AxisFactor
unsigned int	m_Index

Detailed Description

template<typename T, typename Base>
class armnn::PerAxisIterator< T, Base >

PerAxisIterator for per-axis quantization.

Iterates over a tensor as layed out in memory and keeps track of the axis index.

Definition at line 694 of file BaseIterator.hpp.

Constructor & Destructor Documentation

PerAxisIterator() [1/2]

PerAxisIterator ( T *  data = nullptr, unsigned int  axisFactor = 0, unsigned int  axisDimensionality = 0  )

inline

Definition at line 697 of file BaseIterator.hpp.

        : m_Iterator(data),
          m_Start(data),
          m_AxisIndex(0), // iterates over the dimension of axis
          m_AxisDimensionality(axisDimensionality), // tensorShape[quantization_dim]
          m_AxisFactor(axisFactor),
          m_Index(0)
    {}

PerAxisIterator() [2/2]

PerAxisIterator ( T *  data = nullptr, const armnn::TensorShape &  tensorShape = TensorShape(), const unsigned int  axis = 0  )

inline

Definition at line 708 of file BaseIterator.hpp.

        : m_Iterator(data),
          m_Start(data),
          m_AxisIndex(0),
          m_Index(0)
    {
        m_AxisDimensionality = tensorShape[axis];
        m_AxisFactor = armnnUtils::GetNumElementsAfter(tensorShape, axis);
    }

Member Function Documentation

operator++()

PerAxisIterator& operator++ ( )

inlineoverride

Definition at line 728 of file BaseIterator.hpp.

    {
        ++m_Index;
        this -> operator[](m_Index);
        return *this;
    }

operator+=()

PerAxisIterator& operator+= ( const unsigned int  increment )

inlineoverride

Definition at line 735 of file BaseIterator.hpp.

    {
        m_Index += increment;
        this -> operator[](m_Index);
        return *this;
    }

operator-=()

PerAxisIterator& operator-= ( const unsigned int  decrement )

inlineoverride

Definition at line 742 of file BaseIterator.hpp.

    {
        m_Index -= decrement;
        this -> operator[](m_Index);
        return *this;
    }

operator[]()

PerAxisIterator& operator[] ( const unsigned int  index )

inlineoverride

Definition at line 766 of file BaseIterator.hpp.

    {
        SetIndexOnMem(index);
        return *this;
    }

Referenced by PerAxisIterator< const int8_t, Decoder< float > >::operator++(), PerAxisIterator< const int8_t, Decoder< float > >::operator+=(), and PerAxisIterator< const int8_t, Decoder< float > >::operator-=().

Reset()

void Reset ( void *  data )

inlineoverride

Definition at line 720 of file BaseIterator.hpp.

    {
        m_Iterator = reinterpret_cast<T*>(data);
        m_Start = m_Iterator;
        m_AxisIndex = 0;
        m_Index = 0;
    }

SetIndexOnMem()

PerAxisIterator& SetIndexOnMem ( const unsigned int  index )

inline

Definition at line 750 of file BaseIterator.hpp.

    {
        ARMNN_ASSERT(m_Iterator);
        m_Iterator = m_Start + index;
        if (index < m_AxisFactor)
        {
            m_AxisIndex = 0;
        }
        else
        {
            m_AxisIndex = (index / m_AxisFactor) % m_AxisDimensionality;
        }
        m_Index = index;
        return *this;
    }

Referenced by PerAxisIterator< const int8_t, Decoder< float > >::operator[]().

Member Data Documentation

m_AxisDimensionality

unsigned int m_AxisDimensionality

protected

Definition at line 776 of file BaseIterator.hpp.

Referenced by PerAxisIterator< const int8_t, Decoder< float > >::PerAxisIterator(), and PerAxisIterator< const int8_t, Decoder< float > >::SetIndexOnMem().

m_AxisFactor

unsigned int m_AxisFactor

protected

Definition at line 777 of file BaseIterator.hpp.

Referenced by PerAxisIterator< const int8_t, Decoder< float > >::PerAxisIterator(), and PerAxisIterator< const int8_t, Decoder< float > >::SetIndexOnMem().

m_AxisIndex

unsigned int m_AxisIndex

protected

Definition at line 775 of file BaseIterator.hpp.

Referenced by PerAxisIterator< const int8_t, Decoder< float > >::Reset(), and PerAxisIterator< const int8_t, Decoder< float > >::SetIndexOnMem().

m_Index

unsigned int m_Index

protected

Definition at line 778 of file BaseIterator.hpp.

Referenced by PerAxisIterator< const int8_t, Decoder< float > >::operator++(), PerAxisIterator< const int8_t, Decoder< float > >::operator+=(), PerAxisIterator< const int8_t, Decoder< float > >::operator-=(), PerAxisIterator< const int8_t, Decoder< float > >::Reset(), and PerAxisIterator< const int8_t, Decoder< float > >::SetIndexOnMem().

m_Iterator

T* m_Iterator

protected

Definition at line 773 of file BaseIterator.hpp.

Referenced by PerAxisIterator< const int8_t, Decoder< float > >::Reset(), and PerAxisIterator< const int8_t, Decoder< float > >::SetIndexOnMem().

m_Start

T* m_Start

protected

Definition at line 774 of file BaseIterator.hpp.

Referenced by PerAxisIterator< const int8_t, Decoder< float > >::Reset(), and PerAxisIterator< const int8_t, Decoder< float > >::SetIndexOnMem().

---

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

• src/backends/reference/workloads/BaseIterator.hpp