PerAxisIterator< T, Base > Class Template Reference

PerAxisIterator for per-axis quantization. More...

#include <BaseIterator.hpp>

Inheritance 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 754 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 757 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 768 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 788 of file BaseIterator.hpp.

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

operator+=()

PerAxisIterator& operator+= ( const unsigned int  increment )

inlineoverride

Definition at line 795 of file BaseIterator.hpp.

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

operator-=()

PerAxisIterator& operator-= ( const unsigned int  decrement )

inlineoverride

Definition at line 802 of file BaseIterator.hpp.

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

operator[]()

PerAxisIterator& operator[] ( const unsigned int  index )

inlineoverride

Definition at line 826 of file BaseIterator.hpp.

    {
        SetIndexOnMem(index);
        return *this;
    }

Reset()

void Reset ( void *  data )

inlineoverride

Definition at line 780 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 810 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;
    }

Member Data Documentation

m_AxisDimensionality

unsigned int m_AxisDimensionality

protected

Definition at line 836 of file BaseIterator.hpp.

m_AxisFactor

unsigned int m_AxisFactor

protected

Definition at line 837 of file BaseIterator.hpp.

m_AxisIndex

unsigned int m_AxisIndex

protected

Definition at line 835 of file BaseIterator.hpp.

m_Index

unsigned int m_Index

protected

Definition at line 838 of file BaseIterator.hpp.

m_Iterator

T* m_Iterator

protected

Definition at line 833 of file BaseIterator.hpp.

m_Start

T* m_Start

protected

Definition at line 834 of file BaseIterator.hpp.

---

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

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