ScaledInt32PerAxisDecoder Class Reference

#include <BaseIterator.hpp>

Inheritance diagram for ScaledInt32PerAxisDecoder:

Collaboration diagram for ScaledInt32PerAxisDecoder:

Public Member Functions
	ScaledInt32PerAxisDecoder (const int32_t *data, const armnn::TensorInfo tensorInfo)
float	Get () const override
float	GetScale () const
std::vector< float >	DecodeTensor (const TensorShape &tensorShape, bool isDepthwise) override
Public Member Functions inherited from PerAxisIterator< const int32_t, Decoder< float > >
	PerAxisIterator (const int32_t *data=nullptr, unsigned int axisFactor=0, unsigned int axisDimensionality=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
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_AxisDimensionality
unsigned int	m_AxisFactor
unsigned int	m_Index

Detailed Description

Definition at line 861 of file BaseIterator.hpp.

Constructor & Destructor Documentation

ScaledInt32PerAxisDecoder()

ScaledInt32PerAxisDecoder ( const int32_t * data, const armnn::TensorInfo tensorInfo )

inline

Definition at line 864 of file BaseIterator.hpp.

    : PerAxisIterator(data, tensorInfo.GetShape(), tensorInfo.GetQuantizationDim().value()),
      m_Scales(tensorInfo.GetQuantizationScales())
    {}

References PerAxisIterator< const int32_t, Decoder< float > >::PerAxisIterator().

Member Function Documentation

DecodeTensor()

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

inlineoverridevirtual

Implements Decoder< float >.

Definition at line 880 of file BaseIterator.hpp.

    {
        const uint32_t size = tensorShape.GetNumElements();
 
        const uint32_t stepSize = isDepthwise ?
                                  tensorShape[2] * tensorShape[3] : tensorShape.GetNumElements() / tensorShape[0];
 
        const uint32_t stepNum = size / stepSize;
 
        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 step = 0; step < stepNum; ++step)
        {
            //scale = (channelMultiplier * step + mult) % scaleSize;
            for (uint32_t i = 0; i < stepSize; ++i)
            {
                unsigned int index = step * stepSize + i;
                this->operator[](index);
                decodedTensor.emplace_back(armnn::Dequantize(*m_Iterator, m_Scales[step], 0));
            }
        }
        return decodedTensor;
    }

References armnn::Dequantize(), TensorShape::GetNumElements(), PerAxisIterator< const int32_t, Decoder< float > >::m_Iterator, and PerAxisIterator< const int32_t, Decoder< float > >::operator[]().

Get()

float Get ( ) const

inlineoverridevirtual

Implements Decoder< float >.

Definition at line 869 of file BaseIterator.hpp.

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

References armnn::Dequantize, PerAxisIterator< const int32_t, Decoder< float > >::m_AxisIndex, and PerAxisIterator< const int32_t, Decoder< float > >::m_Iterator.

GetScale()

float GetScale ( ) const

inline

Definition at line 875 of file BaseIterator.hpp.

    {
        return m_Scales[m_AxisIndex];
    }

References PerAxisIterator< const int32_t, Decoder< float > >::m_AxisIndex.

---

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

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