ArmNN  NotReleased
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 442 of file BaseIterator.hpp.

Constructor & Destructor Documentation

◆ PerAxisIterator()

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

Definition at line 446 of file BaseIterator.hpp.

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

Member Function Documentation

◆ operator++()

PerAxisIterator& operator++ ( )
inlineoverride

Definition at line 466 of file BaseIterator.hpp.

467  {
468  BOOST_ASSERT(m_Iterator);
469  ++m_Iterator;
470  m_AxisIndex = static_cast<unsigned int>(*m_Iterator) % m_AxisFactor;
471  return *this;
472  }

◆ operator+=()

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

Definition at line 474 of file BaseIterator.hpp.

475  {
476  BOOST_ASSERT(m_Iterator);
477  m_Iterator += increment;
478  m_AxisIndex = static_cast<unsigned int>(*m_Iterator) % m_AxisFactor;
479  return *this;
480  }

◆ operator-=()

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

Definition at line 482 of file BaseIterator.hpp.

483  {
484  BOOST_ASSERT(m_Iterator);
485  m_Iterator -= decrement;
486  m_AxisIndex = static_cast<unsigned int>(*m_Iterator) % m_AxisFactor;
487  return *this;
488  }

◆ operator[]()

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

Definition at line 490 of file BaseIterator.hpp.

491  {
492  BOOST_ASSERT(m_Iterator);
493  m_Iterator = m_Start + index;
494  m_AxisIndex = static_cast<unsigned int>(*m_Iterator) % m_AxisFactor;
495  return *this;
496  }

◆ Reset()

void Reset ( void *  data)
inlineoverride

Definition at line 459 of file BaseIterator.hpp.

460  {
461  m_Iterator = reinterpret_cast<T*>(data);
463  m_AxisIndex = 0;
464  }

◆ SetIndex()

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

Definition at line 451 of file BaseIterator.hpp.

452  {
453  BOOST_ASSERT(m_Iterator);
454  m_Iterator = m_Start + index;
455  m_AxisIndex = axisIndex;
456  return *this;
457  }

Member Data Documentation

◆ m_AxisFactor

unsigned int m_AxisFactor
protected

Definition at line 502 of file BaseIterator.hpp.

◆ m_AxisIndex

unsigned int m_AxisIndex
protected

Definition at line 501 of file BaseIterator.hpp.

◆ m_Iterator

T* m_Iterator
protected

Definition at line 499 of file BaseIterator.hpp.

◆ m_Start

T* m_Start
protected

Definition at line 500 of file BaseIterator.hpp.


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