Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 19d63453 authored by Michael K. Sanders's avatar Michael K. Sanders
Browse files

Adds TENSOR_FLOAT16 operand type. 

Bug: 113563458
Test: VtsHalNeuralnetworksV1_2TargetTest --hal_service_instance=android.hardware.neuralnetworks@1.2::IDevice/sample-all
Change-Id: If12ceff428e1b1a90ef99b7353f0df60d4ef8010
parent 5d7c9952

neuralnetworks/1.2/types.hal

+2 −0
Original line number Diff line number Diff line
@@ -42,6 +42,8 @@ enum OperandType : @1.0::OperandType { 
     * realValue = (integerValue - zeroPoint) * scale.

     */

    TENSOR_QUANT16_ASYMM = 7,

    /** A tensor of 16 bit floating point values. */

    TENSOR_FLOAT16 = 8,

};



/**

neuralnetworks/1.2/vts/functional/ValidateModel.cpp

+4 −0
Original line number Diff line number Diff line
@@ -157,6 +157,7 @@ static uint32_t getInvalidRank(OperandType type) { 
        case OperandType::UINT32:

        case OperandType::BOOL:

            return 1;

        case OperandType::TENSOR_FLOAT16:

        case OperandType::TENSOR_FLOAT32:

        case OperandType::TENSOR_INT32:

        case OperandType::TENSOR_QUANT8_ASYMM:
@@ -186,6 +187,7 @@ static float getInvalidScale(OperandType type) { 
        case OperandType::INT32:

        case OperandType::UINT32:

        case OperandType::BOOL:

        case OperandType::TENSOR_FLOAT16:

        case OperandType::TENSOR_FLOAT32:

            return 1.0f;

        case OperandType::TENSOR_INT32:
@@ -217,6 +219,7 @@ static std::vector<int32_t> getInvalidZeroPoints(OperandType type) { 
        case OperandType::INT32:

        case OperandType::UINT32:

        case OperandType::BOOL:

        case OperandType::TENSOR_FLOAT16:

        case OperandType::TENSOR_FLOAT32:

        case OperandType::TENSOR_INT32:

            return {1};
@@ -262,6 +265,7 @@ static void mutateOperand(Operand* operand, OperandType type) { 
            newOperand.scale = 0.0f;

            newOperand.zeroPoint = 0;

            break;

        case OperandType::TENSOR_FLOAT16:

        case OperandType::TENSOR_FLOAT32:

            newOperand.dimensions =

                operand->dimensions.size() > 0 ? operand->dimensions : hidl_vec<uint32_t>({1});