ArmNN
 20.02
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)
 
PerAxisIteratorSetIndex (unsigned int index, unsigned int axisIndex) override
 
void Reset (void *data) override
 
PerAxisIteratoroperator++ () override
 
PerAxisIteratoroperator+= (const unsigned int increment) override
 
PerAxisIteratoroperator-= (const unsigned int decrement) override
 
PerAxisIteratoroperator[] (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 480 of file BaseIterator.hpp.

Constructor & Destructor Documentation

◆ PerAxisIterator()

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

Definition at line 484 of file BaseIterator.hpp.

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

Member Function Documentation

◆ operator++()

PerAxisIterator& operator++ ( )
inlineoverride

Definition at line 504 of file BaseIterator.hpp.

505  {
506  BOOST_ASSERT(m_Iterator);
507  ++m_Iterator;
508  m_AxisIndex = static_cast<unsigned int>(*m_Iterator) % m_AxisFactor;
509  return *this;
510  }

◆ operator+=()

PerAxisIterator& operator+= ( const unsigned int  increment)
inlineoverride

Definition at line 512 of file BaseIterator.hpp.

513  {
514  BOOST_ASSERT(m_Iterator);
515  m_Iterator += increment;
516  m_AxisIndex = static_cast<unsigned int>(*m_Iterator) % m_AxisFactor;
517  return *this;
518  }

◆ operator-=()

PerAxisIterator& operator-= ( const unsigned int  decrement)
inlineoverride

Definition at line 520 of file BaseIterator.hpp.

521  {
522  BOOST_ASSERT(m_Iterator);
523  m_Iterator -= decrement;
524  m_AxisIndex = static_cast<unsigned int>(*m_Iterator) % m_AxisFactor;
525  return *this;
526  }

◆ operator[]()

PerAxisIterator& operator[] ( const unsigned int  index)
inlineoverride

Definition at line 528 of file BaseIterator.hpp.

529  {
530  BOOST_ASSERT(m_Iterator);
531  m_Iterator = m_Start + index;
532  m_AxisIndex = static_cast<unsigned int>(*m_Iterator) % m_AxisFactor;
533  return *this;
534  }

◆ Reset()

void Reset ( void *  data)
inlineoverride

Definition at line 497 of file BaseIterator.hpp.

498  {
499  m_Iterator = reinterpret_cast<T*>(data);
501  m_AxisIndex = 0;
502  }

◆ SetIndex()

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

Definition at line 489 of file BaseIterator.hpp.

490  {
491  BOOST_ASSERT(m_Iterator);
492  m_Iterator = m_Start + index;
493  m_AxisIndex = axisIndex;
494  return *this;
495  }

Member Data Documentation

◆ m_AxisFactor

unsigned int m_AxisFactor
protected

Definition at line 540 of file BaseIterator.hpp.

◆ m_AxisIndex

unsigned int m_AxisIndex
protected

Definition at line 539 of file BaseIterator.hpp.

◆ m_Iterator

T* m_Iterator
protected

Definition at line 537 of file BaseIterator.hpp.

◆ m_Start

T* m_Start
protected

Definition at line 538 of file BaseIterator.hpp.


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