Float32ToFloat16 Struct Reference

#include <ConvertConstants.hpp>

Static Public Member Functions
static void	Func (std::unique_ptr< ScopedCpuTensorHandle > &handle)

Detailed Description

Definition at line 87 of file ConvertConstants.hpp.

Member Function Documentation

Func()

static void Func ( std::unique_ptr< ScopedCpuTensorHandle > &  handle )

inlinestatic

Definition at line 89 of file ConvertConstants.hpp.

References FloatingPointConverter::ConvertFloat32To16(), armnn::Float16, armnn::Float32, TensorInfo::GetDataType(), TensorInfo::GetNumElements(), TensorInfo::GetShape(), and armnn::info.

    {
        const TensorInfo& info = handle->GetTensorInfo();

        if (info.GetDataType() == DataType::Float32)
        {
            std::vector<Half> newValues(info.GetNumElements());

            armnnUtils::FloatingPointConverter::ConvertFloat32To16(handle->GetTensor<float>(),
                                                                   info.GetNumElements(),
                                                                   newValues.data());

            TensorInfo newInfo(info.GetShape(), DataType::Float16);
            ConstTensor newInput(newInfo, newValues);
            handle.reset(new ScopedCpuTensorHandle(newInput));
        }
    }

---

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

• src/armnn/optimizations/ConvertConstants.hpp