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

Commit d08e4d8d authored by Przemyslaw Szczepaniak's avatar Przemyslaw Szczepaniak Committed by android-build-merger
Browse files

Add TENSOR_QUANT8_ASYMM_SIGNED support for ARG(MIN|MAX) 

am: 2326dd1a

Change-Id: I71380df2974e42d6b18af5aac13edec01254907d
parents 58a95666 2326dd1a

neuralnetworks/1.3/vts/functional/ValidateModel.cpp

+2 −1
Original line number Diff line number Diff line
@@ -340,7 +340,8 @@ static bool mutateOperationOperandTypeSkip(size_t operand, OperandType type, con 
            case OperationType::ARGMAX:

            case OperationType::ARGMIN: {

                if (type == OperandType::TENSOR_FLOAT16 || type == OperandType::TENSOR_FLOAT32 ||

                    type == OperandType::TENSOR_INT32 || type == OperandType::TENSOR_QUANT8_ASYMM) {

                    type == OperandType::TENSOR_INT32 || type == OperandType::TENSOR_QUANT8_ASYMM ||

                    type == OperandType::TENSOR_QUANT8_ASYMM_SIGNED) {

                    return true;

                }

            } break;