ArmNN
 26.01
Loading...
Searching...
No Matches
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 ()
 
- 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) {}
ScaledInt32Decoder(const int32_t *data, const float 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 }
TypedIterator & operator[](const unsigned int index) override

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

◆ Get()

float Get ( ) const
inlineoverridevirtual

Implements Decoder< float >.

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: