PerAxisIterator< T, Base > Class Template Reference

#include <BaseIterator.hpp>

Inheritance diagram for PerAxisIterator< T, Base >:

Public Member Functions
	PerAxisIterator (T *data=nullptr, unsigned int axisFactor=0)
PerAxisIterator &	SetIndex (unsigned int index, unsigned int axisIndex) override
void	Reset (void *data) override
PerAxisIterator &	operator++ () override
PerAxisIterator &	operator+= (const unsigned int increment) override
PerAxisIterator &	operator-= (const unsigned int decrement) override
PerAxisIterator &	operator[] (const unsigned int index) override

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

Detailed Description

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

Definition at line 495 of file BaseIterator.hpp.

Constructor & Destructor Documentation

PerAxisIterator()

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

inline

Definition at line 499 of file BaseIterator.hpp.

        : m_Iterator(data), m_Start(data), m_AxisIndex(0), m_AxisFactor(axisFactor)
    {}

Member Function Documentation

operator++()

PerAxisIterator& operator++ ( )

inlineoverride

Definition at line 519 of file BaseIterator.hpp.

    {
        ARMNN_ASSERT(m_Iterator);
        ++m_Iterator;
        m_AxisIndex = static_cast<unsigned int>(*m_Iterator) % m_AxisFactor;
        return *this;
    }

operator+=()

PerAxisIterator& operator+= ( const unsigned int  increment )

inlineoverride

Definition at line 527 of file BaseIterator.hpp.

    {
        ARMNN_ASSERT(m_Iterator);
        m_Iterator += increment;
        m_AxisIndex = static_cast<unsigned int>(*m_Iterator) % m_AxisFactor;
        return *this;
    }

operator-=()

PerAxisIterator& operator-= ( const unsigned int  decrement )

inlineoverride

Definition at line 535 of file BaseIterator.hpp.

    {
        ARMNN_ASSERT(m_Iterator);
        m_Iterator -= decrement;
        m_AxisIndex = static_cast<unsigned int>(*m_Iterator) % m_AxisFactor;
        return *this;
    }

operator[]()

PerAxisIterator& operator[] ( const unsigned int  index )

inlineoverride

Definition at line 543 of file BaseIterator.hpp.

    {
        ARMNN_ASSERT(m_Iterator);
        m_Iterator = m_Start + index;
        m_AxisIndex = static_cast<unsigned int>(*m_Iterator) % m_AxisFactor;
        return *this;
    }

Reset()

void Reset ( void *  data )

inlineoverride

Definition at line 512 of file BaseIterator.hpp.

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

SetIndex()

PerAxisIterator& SetIndex ( unsigned int  index, unsigned int  axisIndex  )

inlineoverride

Definition at line 504 of file BaseIterator.hpp.

    {
         ARMNN_ASSERT(m_Iterator);
         m_Iterator = m_Start + index;
         m_AxisIndex = axisIndex;
         return *this;
    }

Member Data Documentation

m_AxisFactor

unsigned int m_AxisFactor

protected

Definition at line 555 of file BaseIterator.hpp.

m_AxisIndex

unsigned int m_AxisIndex

protected

Definition at line 554 of file BaseIterator.hpp.

m_Iterator

T* m_Iterator

protected

Definition at line 552 of file BaseIterator.hpp.

m_Start

T* m_Start

protected

Definition at line 553 of file BaseIterator.hpp.

---

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

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