ArmNN
 25.11
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) {}

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

Referenced by ScaledInt32Decoder().

◆ ScaledInt32Decoder() [2/2]

ScaledInt32Decoder ( const float scale)
inline

Definition at line 313 of file BaseIterator.hpp.

314 : ScaledInt32Decoder(nullptr, scale) {}

References ScaledInt32Decoder().

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
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: