DataTypeUtils.hpp File Reference

#include <ResolveType.hpp>
#include <reference/workloads/Encoders.hpp>
#include <vector>

Include dependency graph for DataTypeUtils.hpp:

Go to the source code of this file.

Functions
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
std::vector< T >	ConvertToDataType (const std::vector< float > &input, const armnn::TensorInfo &inputTensorInfo)
template<typename T >
T	ConvertToDataType (const float &value, const armnn::TensorInfo &tensorInfo)

Function Documentation

ConvertToDataType() [1/2]

template<typename T >

T ConvertToDataType	(	const float &	value,
		const armnn::TensorInfo &	tensorInfo )

Definition at line 52 of file DataTypeUtils.hpp.

{
    std::vector<T> output(1);
 
    if(sizeof(T) > 4)
    {
        std::unique_ptr<armnn::Encoder<double>> pEncoder = armnn::MakeEncoder<double>(tensorInfo, output.data());
        armnn::Encoder<double>& rEncoder = *pEncoder;
        rEncoder.Set(value);
    }
    else
    {
        std::unique_ptr<armnn::Encoder<float>> pEncoder = armnn::MakeEncoder<float>(tensorInfo, output.data());
        armnn::Encoder<float>& rEncoder = *pEncoder;
        rEncoder.Set(value);
    }
 
    return output[0];
}

References Encoder< IType >::Set().

ConvertToDataType() [2/2]

template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>

std::vector< T > ConvertToDataType	(	const std::vector< float > &	input,
		const armnn::TensorInfo &	inputTensorInfo )

Definition at line 17 of file DataTypeUtils.hpp.

{
    std::vector<T> output(input.size());
    auto outputTensorInfo = inputTensorInfo;
    outputTensorInfo.SetDataType(ArmnnType);
 
    if(sizeof(T) > 4)
    {
        std::unique_ptr<armnn::Encoder<double>> pOutputEncoder = armnn::MakeEncoder<double>(outputTensorInfo,
                                                                                            output.data());
        armnn::Encoder<double>& rOutputEncoder = *pOutputEncoder;
        for (auto it = input.begin(); it != input.end(); ++it)
        {
            rOutputEncoder.Set(*it);
            ++rOutputEncoder;
        }
    }
    else
    {
        std::unique_ptr<armnn::Encoder<float>> pOutputEncoder = armnn::MakeEncoder<float>(outputTensorInfo,
                                                                                          output.data());
        armnn::Encoder<float>& rOutputEncoder = *pOutputEncoder;
        for (auto it = input.begin(); it != input.end(); ++it)
        {
            rOutputEncoder.Set(*it);
            ++rOutputEncoder;
        }
    }
 
    return output;
}

References Encoder< IType >::Set(), and TensorInfo::SetDataType().