#include <BaseIterator.hpp>

Inheritance diagram for ScaledInt32PerAxisDecoder:

Public Member Functions
	ScaledInt32PerAxisDecoder (const int32_t *data, const std::vector< float > &scales, unsigned int axisFactor)

float	Get () const override

float	GetScale () const

std::vector< float >	DecodeTensor (const TensorShape &tensorShape, const unsigned int channelMultiplier, bool isDepthwise) override

Public Member Functions inherited from PerAxisIterator< const int32_t, Decoder< float > >
	PerAxisIterator (const int32_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

Public Member Functions inherited from Decoder< float >
	Decoder ()

virtual	~Decoder ()

Public Member Functions inherited from BaseIterator
	BaseIterator ()

virtual	~BaseIterator ()

Additional Inherited Members
Protected Attributes inherited from PerAxisIterator< const int32_t, Decoder< float > >
const int32_t *	m_Iterator

const int32_t *	m_Start

unsigned int	m_AxisIndex

unsigned int	m_AxisFactor

Detailed Description

Definition at line 920 of file BaseIterator.hpp.

Constructor & Destructor Documentation

◆ ScaledInt32PerAxisDecoder()

ScaledInt32PerAxisDecoder	(	const int32_t *	data,
		const std::vector< float > &	scales,
		unsigned int	axisFactor
	)

inline

Definition at line 923 of file BaseIterator.hpp.

924 : PerAxisIterator(data, axisFactor), m_Scales(scales) {}

armnn::PerAxisIterator< const int32_t, Decoder< float > >::PerAxisIterator

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

Definition: BaseIterator.hpp:778

Member Function Documentation

◆ DecodeTensor()

std::vector<float> DecodeTensor	(	const TensorShape &	tensorShape,
		const unsigned int	channelMultiplier,
		bool	isDepthwise
	)

inlineoverridevirtual

Implements Decoder< float >.

Definition at line 937 of file BaseIterator.hpp.

References armnn::Dequantize, TensorShape::GetNumElements(), and BaseIterator::operator[]().

     {
         const uint32_t size = tensorShape.GetNumElements();
         const uint32_t scaleSize = static_cast<uint32_t>(m_Scales.size());
 
         const uint32_t stepSize = isDepthwise ?
                                   tensorShape[2] * tensorShape[3] : tensorShape.GetNumElements() / tensorShape[0];
 
         const uint32_t stepNum = size / (stepSize * channelMultiplier);
         uint32_t scale;
 
         std::vector<float> decodedTensor;
         decodedTensor.reserve(size);
 
         // channelMultiplier is only used in depthwise convolutions and in other cases will have no effect
         // stepSize is the length of a contiguous area sharing a quantization scale within a tensor
         // stepNum is the number of those steps/blocks in the tensor
         for (uint32_t mult = 0; mult < channelMultiplier; ++mult)
         {
             for (uint32_t step = 0; step < stepNum; ++step)
             {
                 scale = (channelMultiplier * step + mult) % scaleSize;
                 for (uint32_t i = 0; i < stepSize; ++i)
                 {
                     unsigned int index = mult * stepSize * channelMultiplier +
                                          step * stepSize + i;
                     this->operator[](index);
                     decodedTensor.emplace_back(armnn::Dequantize(*m_Iterator, m_Scales[scale], 0));
                 }
             }
         }
         return decodedTensor;
     }

◆ Get()

float Get ( ) const

inlineoverridevirtual

Implements Decoder< float >.

Definition at line 926 of file BaseIterator.hpp.

References armnn::Dequantize.

     {
         return armnn::Dequantize(*m_Iterator, m_Scales[m_AxisIndex], 0);
     }

◆ GetScale()

float GetScale ( ) const

inline

Definition at line 932 of file BaseIterator.hpp.

     {
         return m_Scales[m_AxisIndex];
     }

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

src/backends/reference/workloads/BaseIterator.hpp

Public Member Functions