armnnSerializer Namespace Reference

Classes
class	ISerializer
class	Serializer
class	SerializerVisitor

Typedefs
using	ISerializerPtr = std::unique_ptr< ISerializer, void(*)(ISerializer *serializer)>

Functions
serializer::ActivationFunction	GetFlatBufferActivationFunction (armnn::ActivationFunction function)
serializer::ArgMinMaxFunction	GetFlatBufferArgMinMaxFunction (armnn::ArgMinMaxFunction function)
armnnSerializer::ComparisonOperation	GetFlatBufferComparisonOperation (armnn::ComparisonOperation comparisonOperation)
armnnSerializer::ConstTensorData	GetFlatBufferConstTensorData (armnn::DataType dataType)
armnnSerializer::DataType	GetFlatBufferDataType (armnn::DataType dataType)
armnnSerializer::DataLayout	GetFlatBufferDataLayout (armnn::DataLayout dataLayout)
armnnSerializer::UnaryOperation	GetFlatBufferUnaryOperation (armnn::UnaryOperation comparisonOperation)
armnnSerializer::PoolingAlgorithm	GetFlatBufferPoolingAlgorithm (armnn::PoolingAlgorithm poolingAlgorithm)
armnnSerializer::OutputShapeRounding	GetFlatBufferOutputShapeRounding (armnn::OutputShapeRounding outputShapeRounding)
armnnSerializer::PaddingMethod	GetFlatBufferPaddingMethod (armnn::PaddingMethod paddingMethod)
armnnSerializer::NormalizationAlgorithmChannel	GetFlatBufferNormalizationAlgorithmChannel (armnn::NormalizationAlgorithmChannel normalizationAlgorithmChannel)
armnnSerializer::NormalizationAlgorithmMethod	GetFlatBufferNormalizationAlgorithmMethod (armnn::NormalizationAlgorithmMethod normalizationAlgorithmMethod)
armnnSerializer::ResizeMethod	GetFlatBufferResizeMethod (armnn::ResizeMethod method)

Typedef Documentation

ISerializerPtr

using ISerializerPtr = std::unique_ptr<ISerializer, void(*)(ISerializer* serializer)>

Definition at line 15 of file ISerializer.hpp.

Function Documentation

GetFlatBufferActivationFunction()

serializer::ActivationFunction armnnSerializer::GetFlatBufferActivationFunction

(

armnn::ActivationFunction

function

)

Definition at line 27 of file Serializer.cpp.

References armnn::Abs, armnn::BoundedReLu, armnn::LeakyReLu, armnn::Linear, armnn::ReLu, armnn::Sigmoid, armnn::Sqrt, armnn::Square, and armnn::TanH.

Referenced by SerializerVisitor::VisitActivationLayer().

{
    switch (function)
    {
        case armnn::ActivationFunction::Sigmoid:
            return serializer::ActivationFunction::ActivationFunction_Sigmoid;
        case armnn::ActivationFunction::TanH:
            return serializer::ActivationFunction::ActivationFunction_TanH;
        case armnn::ActivationFunction::Linear:
            return serializer::ActivationFunction::ActivationFunction_Linear;
        case armnn::ActivationFunction::ReLu:
            return serializer::ActivationFunction::ActivationFunction_ReLu;
        case armnn::ActivationFunction::BoundedReLu:
            return serializer::ActivationFunction::ActivationFunction_BoundedReLu;
        case armnn::ActivationFunction::LeakyReLu:
            return serializer::ActivationFunction::ActivationFunction_LeakyReLu;
        case armnn::ActivationFunction::Abs:
            return serializer::ActivationFunction::ActivationFunction_Abs;
        case armnn::ActivationFunction::Sqrt:
            return serializer::ActivationFunction::ActivationFunction_Sqrt;
        case armnn::ActivationFunction::Square:
            return serializer::ActivationFunction::ActivationFunction_Square;
        default:
            return serializer::ActivationFunction::ActivationFunction_Sigmoid;
    }
}

GetFlatBufferArgMinMaxFunction()

serializer::ArgMinMaxFunction armnnSerializer::GetFlatBufferArgMinMaxFunction

(

armnn::ArgMinMaxFunction

function

)

Definition at line 54 of file Serializer.cpp.

References armnn::Max, and armnn::Min.

Referenced by SerializerVisitor::VisitArgMinMaxLayer().

{
    switch (function)
    {
        case armnn::ArgMinMaxFunction::Max:
            return serializer::ArgMinMaxFunction::ArgMinMaxFunction_Max;
        case armnn::ArgMinMaxFunction::Min:
        default:
            return serializer::ArgMinMaxFunction::ArgMinMaxFunction_Min;
    }
}

GetFlatBufferComparisonOperation()

armnnSerializer::ComparisonOperation GetFlatBufferComparisonOperation

(

armnn::ComparisonOperation

comparisonOperation

)

Definition at line 11 of file SerializerUtils.cpp.

References armnn::Equal, armnn::Greater, armnn::GreaterOrEqual, armnn::Less, armnn::LessOrEqual, and armnn::NotEqual.

Referenced by SerializerVisitor::VisitComparisonLayer().

{
    switch (comparisonOperation)
    {
        case armnn::ComparisonOperation::Equal:
            return armnnSerializer::ComparisonOperation::ComparisonOperation_Equal;
        case armnn::ComparisonOperation::Greater:
            return armnnSerializer::ComparisonOperation::ComparisonOperation_Greater;
        case armnn::ComparisonOperation::GreaterOrEqual:
            return armnnSerializer::ComparisonOperation::ComparisonOperation_GreaterOrEqual;
        case armnn::ComparisonOperation::Less:
            return armnnSerializer::ComparisonOperation::ComparisonOperation_Less;
        case armnn::ComparisonOperation::LessOrEqual:
            return armnnSerializer::ComparisonOperation::ComparisonOperation_LessOrEqual;
        case armnn::ComparisonOperation::NotEqual:
        default:
            return armnnSerializer::ComparisonOperation::ComparisonOperation_NotEqual;
    }
}

GetFlatBufferConstTensorData()

armnnSerializer::ConstTensorData GetFlatBufferConstTensorData

(

armnn::DataType

dataType

)

Definition at line 31 of file SerializerUtils.cpp.

References armnn::Boolean, armnn::Float16, armnn::Float32, armnn::QAsymmU8, armnn::QSymmS16, and armnn::Signed32.

Referenced by SerializerVisitor::VisitQuantizedLstmLayer().

{
    switch (dataType)
    {
        case armnn::DataType::Float32:
        case armnn::DataType::Signed32:
            return armnnSerializer::ConstTensorData::ConstTensorData_IntData;
        case armnn::DataType::Float16:
        case armnn::DataType::QSymmS16:
            return armnnSerializer::ConstTensorData::ConstTensorData_ShortData;
        case armnn::DataType::QAsymmU8:
        case armnn::DataType::Boolean:
            return armnnSerializer::ConstTensorData::ConstTensorData_ByteData;
        default:
            return armnnSerializer::ConstTensorData::ConstTensorData_NONE;
    }
}

GetFlatBufferDataLayout()

armnnSerializer::DataLayout GetFlatBufferDataLayout

(

armnn::DataLayout

dataLayout

)

Definition at line 70 of file SerializerUtils.cpp.

References armnn::NCHW, and armnn::NHWC.

Referenced by SerializerVisitor::VisitBatchNormalizationLayer(), SerializerVisitor::VisitBatchToSpaceNdLayer(), SerializerVisitor::VisitConvolution2dLayer(), SerializerVisitor::VisitDepthToSpaceLayer(), SerializerVisitor::VisitDepthwiseConvolution2dLayer(), SerializerVisitor::VisitInstanceNormalizationLayer(), SerializerVisitor::VisitL2NormalizationLayer(), SerializerVisitor::VisitNormalizationLayer(), SerializerVisitor::VisitPooling2dLayer(), SerializerVisitor::VisitResizeBilinearLayer(), SerializerVisitor::VisitResizeLayer(), SerializerVisitor::VisitSpaceToBatchNdLayer(), SerializerVisitor::VisitSpaceToDepthLayer(), SerializerVisitor::VisitStridedSliceLayer(), and SerializerVisitor::VisitTransposeConvolution2dLayer().

{
    switch (dataLayout)
    {
        case armnn::DataLayout::NHWC:
            return armnnSerializer::DataLayout::DataLayout_NHWC;
        case armnn::DataLayout::NCHW:
        default:
            return armnnSerializer::DataLayout::DataLayout_NCHW;
    }
}

GetFlatBufferDataType()

armnnSerializer::DataType GetFlatBufferDataType

(

armnn::DataType

dataType

)

Definition at line 49 of file SerializerUtils.cpp.

References armnn::Boolean, armnn::Float16, armnn::Float32, armnn::QAsymmU8, armnn::QSymmS16, and armnn::Signed32.

Referenced by SerializerVisitor::GetVersionTable(), and SerializerVisitor::VisitQuantizedLstmLayer().

{
    switch (dataType)
    {
        case armnn::DataType::Float32:
            return armnnSerializer::DataType::DataType_Float32;
        case armnn::DataType::Float16:
            return armnnSerializer::DataType::DataType_Float16;
        case armnn::DataType::Signed32:
            return armnnSerializer::DataType::DataType_Signed32;
        case armnn::DataType::QSymmS16:
            return armnnSerializer::DataType::DataType_QSymmS16;
        case armnn::DataType::QAsymmU8:
            return armnnSerializer::DataType::DataType_QAsymmU8;
        case armnn::DataType::Boolean:
            return armnnSerializer::DataType::DataType_Boolean;
        default:
            return armnnSerializer::DataType::DataType_Float16;
    }
}

GetFlatBufferNormalizationAlgorithmChannel()

armnnSerializer::NormalizationAlgorithmChannel GetFlatBufferNormalizationAlgorithmChannel

(

armnn::NormalizationAlgorithmChannel

normalizationAlgorithmChannel

)

Definition at line 139 of file SerializerUtils.cpp.

References armnn::Across, and armnn::Within.

Referenced by SerializerVisitor::VisitNormalizationLayer().

{
    switch (normalizationAlgorithmChannel)
    {
        case armnn::NormalizationAlgorithmChannel::Across:
            return armnnSerializer::NormalizationAlgorithmChannel::NormalizationAlgorithmChannel_Across;
        case armnn::NormalizationAlgorithmChannel::Within:
            return armnnSerializer::NormalizationAlgorithmChannel::NormalizationAlgorithmChannel_Within;
        default:
            return armnnSerializer::NormalizationAlgorithmChannel::NormalizationAlgorithmChannel_Across;
    }
}

GetFlatBufferNormalizationAlgorithmMethod()

armnnSerializer::NormalizationAlgorithmMethod GetFlatBufferNormalizationAlgorithmMethod

(

armnn::NormalizationAlgorithmMethod

normalizationAlgorithmMethod

)

Definition at line 153 of file SerializerUtils.cpp.

References armnn::LocalBrightness, and armnn::LocalContrast.

Referenced by SerializerVisitor::VisitNormalizationLayer().

{
    switch (normalizationAlgorithmMethod)
    {
        case armnn::NormalizationAlgorithmMethod::LocalBrightness:
            return armnnSerializer::NormalizationAlgorithmMethod::NormalizationAlgorithmMethod_LocalBrightness;
        case armnn::NormalizationAlgorithmMethod::LocalContrast:
            return armnnSerializer::NormalizationAlgorithmMethod::NormalizationAlgorithmMethod_LocalContrast;
        default:
            return armnnSerializer::NormalizationAlgorithmMethod::NormalizationAlgorithmMethod_LocalBrightness;
    }
}

GetFlatBufferOutputShapeRounding()

armnnSerializer::OutputShapeRounding GetFlatBufferOutputShapeRounding

(

armnn::OutputShapeRounding

outputShapeRounding

)

Definition at line 115 of file SerializerUtils.cpp.

References armnn::Ceiling, and armnn::Floor.

Referenced by SerializerVisitor::VisitPooling2dLayer().

{
    switch (outputShapeRounding)
    {
        case armnn::OutputShapeRounding::Ceiling:
            return armnnSerializer::OutputShapeRounding::OutputShapeRounding_Ceiling;
        case armnn::OutputShapeRounding::Floor:
        default:
            return armnnSerializer::OutputShapeRounding::OutputShapeRounding_Floor;
    }
}

GetFlatBufferPaddingMethod()

armnnSerializer::PaddingMethod GetFlatBufferPaddingMethod

(

armnn::PaddingMethod

paddingMethod

)

Definition at line 127 of file SerializerUtils.cpp.

References armnn::Exclude, and armnn::IgnoreValue.

Referenced by SerializerVisitor::VisitPooling2dLayer().

{
    switch (paddingMethod)
    {
        case armnn::PaddingMethod::IgnoreValue:
            return armnnSerializer::PaddingMethod::PaddingMethod_IgnoreValue;
        case armnn::PaddingMethod::Exclude:
        default:
            return armnnSerializer::PaddingMethod::PaddingMethod_Exclude;
    }
}

GetFlatBufferPoolingAlgorithm()

armnnSerializer::PoolingAlgorithm GetFlatBufferPoolingAlgorithm

(

armnn::PoolingAlgorithm

poolingAlgorithm

)

Definition at line 101 of file SerializerUtils.cpp.

References armnn::Average, armnn::L2, and armnn::Max.

Referenced by SerializerVisitor::VisitPooling2dLayer().

{
    switch (poolingAlgorithm)
    {
        case armnn::PoolingAlgorithm::Average:
            return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_Average;
        case armnn::PoolingAlgorithm::L2:
            return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_L2;
        case armnn::PoolingAlgorithm::Max:
        default:
            return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_Max;
    }
}

GetFlatBufferResizeMethod()

armnnSerializer::ResizeMethod GetFlatBufferResizeMethod

(

armnn::ResizeMethod

method

)

Definition at line 167 of file SerializerUtils.cpp.

References armnn::Bilinear, and armnn::NearestNeighbor.

Referenced by SerializerVisitor::VisitResizeLayer().

{
    switch (method)
    {
        case armnn::ResizeMethod::NearestNeighbor:
            return armnnSerializer::ResizeMethod_NearestNeighbor;
        case armnn::ResizeMethod::Bilinear:
            return armnnSerializer::ResizeMethod_Bilinear;
        default:
            return armnnSerializer::ResizeMethod_NearestNeighbor;
    }
}

GetFlatBufferUnaryOperation()

armnnSerializer::UnaryOperation GetFlatBufferUnaryOperation

(

armnn::UnaryOperation

comparisonOperation

)

Definition at line 82 of file SerializerUtils.cpp.

References armnn::Abs, armnn::Exp, armnn::Neg, armnn::Rsqrt, and armnn::Sqrt.

Referenced by SerializerVisitor::VisitElementwiseUnaryLayer().

{
    switch (comparisonOperation)
    {
        case armnn::UnaryOperation::Abs:
            return armnnSerializer::UnaryOperation::UnaryOperation_Abs;
        case armnn::UnaryOperation::Rsqrt:
            return armnnSerializer::UnaryOperation::UnaryOperation_Rsqrt;
        case armnn::UnaryOperation::Sqrt:
            return armnnSerializer::UnaryOperation::UnaryOperation_Sqrt;
        case armnn::UnaryOperation::Exp:
            return armnnSerializer::UnaryOperation::UnaryOperation_Exp;
        case armnn::UnaryOperation::Neg:
            return armnnSerializer::UnaryOperation::UnaryOperation_Neg;
        default:
            throw armnn::InvalidArgumentException("Unary operation unknown");
    }
}