22.08
|
PerAxisIterator for per-axis quantization. More...
#include <BaseIterator.hpp>
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 |
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.
|
inline |
Definition at line 757 of file BaseIterator.hpp.
|
inline |
Definition at line 768 of file BaseIterator.hpp.
|
inlineoverride |
Definition at line 788 of file BaseIterator.hpp.
|
inlineoverride |
Definition at line 795 of file BaseIterator.hpp.
|
inlineoverride |
Definition at line 802 of file BaseIterator.hpp.
|
inlineoverride |
Definition at line 826 of file BaseIterator.hpp.
|
inlineoverride |
Definition at line 780 of file BaseIterator.hpp.
|
inline |
Definition at line 810 of file BaseIterator.hpp.
|
protected |
Definition at line 836 of file BaseIterator.hpp.
|
protected |
Definition at line 837 of file BaseIterator.hpp.
|
protected |
Definition at line 835 of file BaseIterator.hpp.
|
protected |
Definition at line 838 of file BaseIterator.hpp.
|
protected |
Definition at line 833 of file BaseIterator.hpp.
|
protected |
Definition at line 834 of file BaseIterator.hpp.