ArmNN
 24.08
ScaledInt32Decoder Class Reference

#include <BaseIterator.hpp>

Inheritance diagram for ScaledInt32Decoder:
[legend]
Collaboration diagram for ScaledInt32Decoder:
[legend]

Public Member Functions

 ScaledInt32Decoder (const int32_t *data, const float scale)
 
 ScaledInt32Decoder (const float scale)
 
float Get () const override
 
std::vector< float > DecodeTensor (const TensorShape &tensorShape, const bool) override
 
- Public Member Functions inherited from TypedIterator< const int32_t, Decoder< float > >
 TypedIterator (const int32_t *data=nullptr)
 
void Reset (void *data) override
 
TypedIteratoroperator++ () override
 
TypedIteratoroperator+= (const unsigned int increment) override
 
TypedIteratoroperator-= (const unsigned int increment) override
 
TypedIteratoroperator[] (const unsigned int index) override
 
- Public Member Functions inherited from Decoder< float >
 Decoder ()
 
virtual ~Decoder ()
 
virtual float Get () const=0
 
- Public Member Functions inherited from BaseIterator
 BaseIterator ()
 
virtual ~BaseIterator ()
 

Additional Inherited Members

- Protected Attributes inherited from TypedIterator< const int32_t, Decoder< float > >
const int32_t * m_Iterator
 
const int32_t * m_Start
 

Detailed Description

Definition at line 307 of file BaseIterator.hpp.

Constructor & Destructor Documentation

◆ ScaledInt32Decoder() [1/2]

ScaledInt32Decoder ( const int32_t *  data,
const float  scale 
)
inline

Definition at line 310 of file BaseIterator.hpp.

311  : TypedIterator(data), m_Scale(scale) {}

◆ ScaledInt32Decoder() [2/2]

ScaledInt32Decoder ( const float  scale)
inline

Definition at line 313 of file BaseIterator.hpp.

314  : ScaledInt32Decoder(nullptr, scale) {}

Member Function Documentation

◆ DecodeTensor()

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

Implements Decoder< float >.

Definition at line 320 of file BaseIterator.hpp.

321  {
322  const unsigned int size = tensorShape.GetNumElements();
323  std::vector<float> decodedTensor;
324  decodedTensor.reserve(size);
325 
326  for (uint32_t i = 0; i < size; ++i)
327  {
328  this->operator[](i);
329  decodedTensor.emplace_back(static_cast<float>(*m_Iterator) * m_Scale);
330  }
331 
332  return decodedTensor;
333  }

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

◆ Get()

float Get ( ) const
inlineoverride

Definition at line 316 of file BaseIterator.hpp.

317  {
318  return static_cast<float>(*m_Iterator) * m_Scale;
319  }

The documentation for this class was generated from the following file:
armnn::TypedIterator< const int32_t, Decoder< float > >::m_Iterator
const int32_t * m_Iterator
Definition: BaseIterator.hpp:107
armnn::ScaledInt32Decoder::ScaledInt32Decoder
ScaledInt32Decoder(const int32_t *data, const float scale)
Definition: BaseIterator.hpp:310
armnn::TypedIterator< const int32_t, Decoder< float > >::TypedIterator
TypedIterator(const int32_t *data=nullptr)
Definition: BaseIterator.hpp:68
armnn::TypedIterator< const int32_t, Decoder< float > >::operator[]
TypedIterator & operator[](const unsigned int index) override
Definition: BaseIterator.hpp:99