TosaOperatorUtils.hpp File Reference

#include <armnn/Tensor.hpp>
#include <armnn/Types.hpp>
#include <tosa_generated.h>

Go to the source code of this file.

Functions
DType	ArmNNToDType (const DataType &type)
std::vector< int32_t >	GetTosaTensorShape (const TensorShape &shape)
std::string	GetUniqueTosaMappingID ()

Function Documentation

ArmNNToDType()

DType ArmNNToDType ( const DataType &  type )

inline

Definition at line 17 of file TosaOperatorUtils.hpp.

References armnn::BFloat16, armnn::Boolean, armnn::Float16, armnn::Float32, armnn::QAsymmS8, armnn::QAsymmU8, armnn::QSymmS16, armnn::QSymmS8, armnn::Signed32, and armnn::Signed64.

{
    switch (type)
    {
        case DataType::Float16:
        case DataType::BFloat16:
            return DType_FP16;
        case DataType::Float32:
            return DType_FP32;
        case DataType::QAsymmU8:
            return DType_UINT8;
        case DataType::QSymmS8:
        case DataType::QAsymmS8:
            return DType_INT8;
        case DataType::QSymmS16:
            return DType_INT16;
        case DataType::Signed32:
            return DType_INT32;
        case DataType::Signed64:
            // No signed 64, only DType_INT48.
            return DType_UNKNOWN;
        case DataType::Boolean:
            return DType_BOOL;
        default:
            return DType_UNKNOWN;
    }
}

GetTosaTensorShape()

std::vector<int32_t> GetTosaTensorShape ( const TensorShape &  shape )

inline

Definition at line 46 of file TosaOperatorUtils.hpp.

References TensorShape::GetNumDimensions().

Referenced by ConvertAdditionToTosaOperator().

{
    std::vector<int32_t> returnShape;
    for (u_int32_t i = 0; i < shape.GetNumDimensions(); i++)
    {
        returnShape.push_back(static_cast<int32_t>(shape[i]));
    }
    return returnShape;
}

GetUniqueTosaMappingID()

std::string GetUniqueTosaMappingID ( )

inline

Definition at line 58 of file TosaOperatorUtils.hpp.

Referenced by ConvertAdditionToTosaOperator().

{
    return std::to_string(++uniqueTosaMappingID);
}