| /* |
| * Copyright (C) 2017 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #define LOG_TAG "ValidateHal" |
| |
| #include "ValidateHal.h" |
| |
| #include <android-base/logging.h> |
| |
| #include <algorithm> |
| #include <vector> |
| |
| #include "NeuralNetworks.h" |
| #include "OperationsUtils.h" |
| #include "Tracing.h" |
| #include "Utils.h" |
| |
| namespace android { |
| namespace nn { |
| |
| using namespace hal; |
| |
| template <class T_Model> |
| struct ModelToHalVersion; |
| template <> |
| struct ModelToHalVersion<V1_0::Model> { |
| static constexpr HalVersion version = HalVersion::V1_0; |
| }; |
| template <> |
| struct ModelToHalVersion<V1_1::Model> { |
| static constexpr HalVersion version = HalVersion::V1_1; |
| }; |
| template <> |
| struct ModelToHalVersion<V1_2::Model> { |
| static constexpr HalVersion version = HalVersion::V1_2; |
| }; |
| template <> |
| struct ModelToHalVersion<V1_3::Model> { |
| static constexpr HalVersion version = HalVersion::V1_3; |
| }; |
| |
| class MemoryAccessVerifier { |
| public: |
| MemoryAccessVerifier(const hidl_vec<hidl_memory>& pools) |
| : mPoolCount(pools.size()), mPoolSizes(mPoolCount) { |
| for (size_t i = 0; i < mPoolCount; i++) { |
| mPoolSizes[i] = pools[i].size(); |
| } |
| } |
| bool validate(const DataLocation& location) { |
| if (location.poolIndex >= mPoolCount) { |
| LOG(ERROR) << "Invalid poolIndex " << location.poolIndex << "/" << mPoolCount; |
| return false; |
| } |
| const size_t size = mPoolSizes[location.poolIndex]; |
| // Do the addition using size_t to avoid potential wrap-around problems. |
| if (static_cast<size_t>(location.offset) + location.length > size) { |
| LOG(ERROR) << "Reference to pool " << location.poolIndex << " with offset " |
| << location.offset << " and length " << location.length |
| << " exceeds pool size of " << size; |
| return false; |
| } |
| return true; |
| } |
| |
| private: |
| size_t mPoolCount; |
| std::vector<size_t> mPoolSizes; |
| }; |
| |
| static bool validateOperandExtraParams(const V1_3::Operand& operand, uint32_t index) { |
| switch (operand.type) { |
| case OperandType::FLOAT32: |
| case OperandType::INT32: |
| case OperandType::UINT32: |
| case OperandType::BOOL: |
| case OperandType::TENSOR_FLOAT32: |
| case OperandType::TENSOR_FLOAT16: |
| case OperandType::TENSOR_INT32: |
| case OperandType::TENSOR_QUANT8_ASYMM: |
| case OperandType::TENSOR_QUANT8_ASYMM_SIGNED: |
| case OperandType::TENSOR_QUANT8_SYMM: |
| case OperandType::TENSOR_QUANT16_ASYMM: |
| case OperandType::TENSOR_QUANT16_SYMM: |
| case OperandType::TENSOR_BOOL8: { |
| NN_RET_CHECK(operand.extraParams.getDiscriminator() == |
| V1_3::Operand::ExtraParams::hidl_discriminator::none) |
| << "Operand " << index << ": Operand of type " |
| << getOperandTypeName(operand.type) |
| << " has incorrect extraParams: " << toString(operand.extraParams); |
| } break; |
| case OperandType::TENSOR_QUANT8_SYMM_PER_CHANNEL: { |
| NN_RET_CHECK(operand.extraParams.getDiscriminator() == |
| V1_3::Operand::ExtraParams::hidl_discriminator::channelQuant) |
| << "Operand " << index << ": Operand of type " |
| << getOperandTypeName(operand.type) << " without a Channel Quantization params"; |
| auto& channelQuant = operand.extraParams.channelQuant(); |
| |
| size_t count = operand.dimensions.size(); |
| NN_RET_CHECK_LT(channelQuant.channelDim, count) |
| << "Operand " << index << ": Operand of type " |
| << getOperandTypeName(operand.type) |
| << " with an invalid channelQuant.channelDim " << channelQuant.channelDim |
| << ", must be valid dimension index in range [0, " << count << ")"; |
| uint32_t expected = operand.dimensions[channelQuant.channelDim]; |
| NN_RET_CHECK_EQ(channelQuant.scales.size(), expected) |
| << "Operand " << index << ": Operand of type " |
| << getOperandTypeName(operand.type) << " with a wrong-sized scales, " |
| << "expected " << expected << " was " << channelQuant.scales.size(); |
| NN_RET_CHECK_NE(expected, 0) |
| << "Operand " << index << ": Operand of type " |
| << getOperandTypeName(operand.type) << " channel dimension " |
| << channelQuant.channelDim << " is underspecified (can't be 0)"; |
| for (uint32_t i = 0; i < expected; ++i) { |
| NN_RET_CHECK_GT(channelQuant.scales[i], .0f) |
| << "Operand " << index << ": Operand of type " |
| << getOperandTypeName(operand.type) << " with a negative value in scales[" |
| << i << "]=" << channelQuant.scales[i]; |
| } |
| } break; |
| default: { |
| if (isExtensionOperandType(operand.type)) { |
| NN_RET_CHECK(operand.extraParams.getDiscriminator() == |
| V1_3::Operand::ExtraParams::hidl_discriminator::extension || |
| operand.extraParams.getDiscriminator() == |
| V1_3::Operand::ExtraParams::hidl_discriminator::none) |
| << "Operand " << index << ": Extension operand of type " |
| << getOperandTypeName(operand.type) |
| << " has incorrect extraParams: " << toString(operand.extraParams); |
| } |
| // No validation for OEM types. |
| } break; |
| } |
| return true; |
| } |
| |
| template <typename VersionedOperand> |
| static bool validateOperands(const hidl_vec<VersionedOperand>& operands, |
| const hidl_vec<uint8_t>& operandValues, |
| const hidl_vec<hidl_memory>& pools, bool allowUnspecifiedRank) { |
| uint32_t index = 0; |
| MemoryAccessVerifier poolVerifier(pools); |
| for (auto& versionedOperand : operands) { |
| if (!validOperandType(versionedOperand.type)) { |
| LOG(ERROR) << "Operand is not supported by this version: " |
| << toString(versionedOperand.type); |
| return false; |
| } |
| // Once we are sure the operand is supported by its version, it is safe |
| // to convert it to the latest version for the rest of the validations. |
| V1_3::Operand operand = convertToV1_3(versionedOperand); |
| // Validate type and dimensions. |
| switch (operand.type) { |
| case OperandType::FLOAT16: |
| case OperandType::FLOAT32: |
| case OperandType::INT32: |
| case OperandType::UINT32: |
| case OperandType::BOOL: |
| case OperandType::OEM: { |
| size_t count = operand.dimensions.size(); |
| if (count != 0) { |
| LOG(ERROR) << "Operand " << index << ": Scalar data has dimensions of rank " |
| << count; |
| return false; |
| } |
| break; |
| } |
| case OperandType::TENSOR_FLOAT16: |
| case OperandType::TENSOR_FLOAT32: |
| case OperandType::TENSOR_INT32: |
| case OperandType::TENSOR_QUANT8_ASYMM: |
| case OperandType::TENSOR_QUANT8_ASYMM_SIGNED: |
| case OperandType::TENSOR_QUANT8_SYMM: |
| case OperandType::TENSOR_QUANT16_ASYMM: |
| case OperandType::TENSOR_QUANT16_SYMM: |
| case OperandType::TENSOR_BOOL8: |
| case OperandType::TENSOR_QUANT8_SYMM_PER_CHANNEL: |
| case OperandType::TENSOR_OEM_BYTE: { |
| if ((!allowUnspecifiedRank || operand.lifetime == OperandLifeTime::CONSTANT_COPY || |
| operand.lifetime == OperandLifeTime::CONSTANT_REFERENCE) && |
| operand.dimensions.size() == 0) { |
| LOG(ERROR) << "Operand " << index << ": Tensor has dimensions of rank 0"; |
| return false; |
| } |
| break; |
| } |
| default: { |
| if (!isExtensionOperandType(operand.type)) { |
| LOG(ERROR) << "Operand " << index << ": Invalid operand type " |
| << toString(operand.type); |
| return false; |
| } |
| } break; |
| } |
| |
| // TODO Validate the numberOfConsumers. |
| // TODO Since we have to validate it, there was no point in including it. For the next |
| // release, consider removing unless we have an additional process in system space |
| // that creates this value. In that case, it would not have to be validated. |
| |
| // Validate the scale. |
| switch (operand.type) { |
| case OperandType::FLOAT16: |
| case OperandType::FLOAT32: |
| case OperandType::INT32: |
| case OperandType::UINT32: |
| case OperandType::BOOL: |
| case OperandType::TENSOR_FLOAT16: |
| case OperandType::TENSOR_FLOAT32: |
| case OperandType::TENSOR_BOOL8: |
| case OperandType::TENSOR_QUANT8_SYMM_PER_CHANNEL: |
| if (operand.scale != 0.f) { |
| LOG(ERROR) << "Operand " << index << ": Operand of type " |
| << getOperandTypeName(operand.type) << " with a non-zero scale (" |
| << operand.scale << ")"; |
| return false; |
| } |
| break; |
| case OperandType::TENSOR_INT32: |
| // TENSOR_INT32 may be used with or without scale, depending on the operation. |
| if (operand.scale < 0.f) { |
| LOG(ERROR) << "Operand " << index << ": Operand of type " |
| << getOperandTypeName(operand.type) << " with a negative scale"; |
| return false; |
| } |
| break; |
| case OperandType::TENSOR_QUANT8_ASYMM: |
| case OperandType::TENSOR_QUANT8_ASYMM_SIGNED: |
| case OperandType::TENSOR_QUANT8_SYMM: |
| case OperandType::TENSOR_QUANT16_ASYMM: |
| case OperandType::TENSOR_QUANT16_SYMM: |
| if (operand.scale <= 0.f) { |
| LOG(ERROR) << "Operand " << index << ": Operand of type " |
| << getOperandTypeName(operand.type) << " with a non-positive scale"; |
| return false; |
| } |
| break; |
| default: |
| if (isExtensionOperandType(operand.type) && operand.scale != 0.f) { |
| LOG(ERROR) << "Operand " << index << ": Operand of type " |
| << getOperandTypeName(operand.type) << " with a non-zero scale (" |
| << operand.scale << ")"; |
| return false; |
| } |
| // No validation for OEM types. |
| // TODO(b/119869082) We should have a separate type for TENSOR_INT32 with a scale. |
| break; |
| } |
| |
| // Validate the zeroPoint. |
| switch (operand.type) { |
| case OperandType::FLOAT16: |
| case OperandType::FLOAT32: |
| case OperandType::INT32: |
| case OperandType::UINT32: |
| case OperandType::BOOL: |
| case OperandType::TENSOR_FLOAT16: |
| case OperandType::TENSOR_FLOAT32: |
| case OperandType::TENSOR_INT32: |
| case OperandType::TENSOR_BOOL8: |
| case OperandType::TENSOR_QUANT8_SYMM: |
| case OperandType::TENSOR_QUANT8_SYMM_PER_CHANNEL: |
| if (operand.zeroPoint != 0) { |
| LOG(ERROR) << "Operand " << index << ": Operand of type " |
| << getOperandTypeName(operand.type) << " with a non-zero zeroPoint " |
| << operand.zeroPoint; |
| return false; |
| } |
| break; |
| case OperandType::TENSOR_QUANT8_ASYMM: |
| if (operand.zeroPoint < 0 || operand.zeroPoint > 255) { |
| LOG(ERROR) << "Operand " << index << ": Operand of type " |
| << getOperandTypeName(operand.type) << " with an invalid zeroPoint " |
| << operand.zeroPoint << ", must be in range [0, 255]"; |
| return false; |
| } |
| break; |
| case OperandType::TENSOR_QUANT8_ASYMM_SIGNED: |
| if (operand.zeroPoint < -128 || operand.zeroPoint > 127) { |
| LOG(ERROR) << "Operand " << index << ": Operand of type " |
| << getOperandTypeName(operand.type) << " with an invalid zeroPoint " |
| << operand.zeroPoint << ", must be in range [-128, 127]"; |
| return false; |
| } |
| break; |
| case OperandType::TENSOR_QUANT16_ASYMM: |
| if (operand.zeroPoint < 0 || operand.zeroPoint > 65535) { |
| LOG(ERROR) << "Operand " << index << ": Operand of type " |
| << getOperandTypeName(operand.type) << " with an invalid zeroPoint " |
| << operand.zeroPoint << ", must be in range [0, 65535]"; |
| return false; |
| } |
| break; |
| case OperandType::TENSOR_QUANT16_SYMM: |
| if (operand.zeroPoint != 0) { |
| LOG(ERROR) << "Operand " << index << ": Operand of type " |
| << getOperandTypeName(operand.type) << " with a non-zero zeroPoint " |
| << operand.zeroPoint; |
| return false; |
| } |
| break; |
| default: |
| if (isExtensionOperandType(operand.type) && operand.zeroPoint != 0) { |
| LOG(ERROR) << "Operand " << index << ": Operand of type " |
| << getOperandTypeName(operand.type) << " with a non-zero zeroPoint " |
| << operand.zeroPoint; |
| return false; |
| } |
| // No validation for OEM types. |
| break; |
| } |
| |
| NN_RET_CHECK(validateOperandExtraParams(operand, index)); |
| |
| // Validate the lifetime and the location. |
| const DataLocation& location = operand.location; |
| switch (operand.lifetime) { |
| case OperandLifeTime::CONSTANT_COPY: |
| if (location.poolIndex != 0) { |
| LOG(ERROR) << "Operand " << index |
| << ": CONSTANT_COPY with a non-zero poolIndex " |
| << location.poolIndex; |
| return false; |
| } |
| // Do the addition using size_t to avoid potential wrap-around problems. |
| if (static_cast<size_t>(location.offset) + location.length > operandValues.size()) { |
| LOG(ERROR) << "Operand " << index |
| << ": OperandValue location out of range. Starts at " |
| << location.offset << ", length " << location.length << ", max " |
| << operandValues.size(); |
| return false; |
| } |
| break; |
| case OperandLifeTime::CONSTANT_REFERENCE: |
| if (!poolVerifier.validate(location)) { |
| return false; |
| } |
| break; |
| case OperandLifeTime::TEMPORARY_VARIABLE: |
| case OperandLifeTime::MODEL_INPUT: |
| case OperandLifeTime::MODEL_OUTPUT: |
| case OperandLifeTime::NO_VALUE: |
| if (location.poolIndex != 0 || location.offset != 0 || location.length != 0) { |
| LOG(ERROR) << "Operand " << index << ": Unexpected poolIndex " |
| << location.poolIndex << ", offset " << location.offset |
| << ", or length " << location.length << " for operand of lifetime " |
| << toString(operand.lifetime); |
| return false; |
| } |
| break; |
| default: |
| LOG(ERROR) << "Operand " << index << ": Invalid lifetime " |
| << toString(operand.lifetime); |
| return false; |
| } |
| |
| // For constants, validate that the length is as expected. The other lifetimes |
| // expect the length to be 0. Don't validate for OEM types. |
| if (operand.lifetime == OperandLifeTime::CONSTANT_REFERENCE || |
| operand.lifetime == OperandLifeTime::CONSTANT_COPY) { |
| if (!isExtensionOperandType(operand.type) && operand.type != OperandType::OEM && |
| operand.type != OperandType::TENSOR_OEM_BYTE) { |
| uint32_t expectedLength = nonExtensionOperandSizeOfData(operand); |
| if (location.length != expectedLength) { |
| LOG(ERROR) << "Operand " << index << ": For operand " << toString(operand) |
| << " expected a size of " << expectedLength << " but got " |
| << location.length; |
| return false; |
| } |
| } |
| } |
| |
| index++; |
| } |
| return true; |
| } |
| |
| static HalVersion getHalVersion(const V1_0::Operation&) { |
| return HalVersion::V1_0; |
| } |
| |
| static HalVersion getHalVersion(const V1_1::Operation&) { |
| return HalVersion::V1_1; |
| } |
| |
| static HalVersion getHalVersion(const V1_2::Operation&) { |
| return HalVersion::V1_2; |
| } |
| |
| static HalVersion getHalVersion(const V1_3::Operation&) { |
| return HalVersion::V1_3; |
| } |
| |
| template <typename VersionedOperation> |
| static bool validateOperations(const hidl_vec<VersionedOperation>& operations, |
| const hidl_vec<Operand>& operands) { |
| const size_t operandCount = operands.size(); |
| // This vector keeps track of whether there's an operation that writes to |
| // each operand. It is used to validate that temporary variables and |
| // model outputs will be written to. |
| std::vector<bool> writtenTo(operandCount, false); |
| for (auto& op : operations) { |
| // TODO Validate the shapes and any known values. This is currently |
| // done in CpuExecutor but should be done here for all drivers. |
| int error = validateOperation( |
| static_cast<int32_t>(op.type), op.inputs.size(), |
| op.inputs.size() > 0 ? op.inputs.data() : nullptr, op.outputs.size(), |
| op.outputs.size() > 0 ? op.outputs.data() : nullptr, operands, getHalVersion(op)); |
| if (error != ANEURALNETWORKS_NO_ERROR) { |
| LOG(ERROR) << "Invalid operation " << toString(op.type); |
| return false; |
| } |
| |
| for (uint32_t i : op.outputs) { |
| const Operand& operand = operands[i]; |
| if (operand.lifetime != OperandLifeTime::TEMPORARY_VARIABLE && |
| operand.lifetime != OperandLifeTime::MODEL_OUTPUT) { |
| LOG(ERROR) << "Writing to an operand with incompatible lifetime " |
| << toString(operand.lifetime); |
| return false; |
| } |
| |
| // Check that we only write once to an operand. |
| if (writtenTo[i]) { |
| LOG(ERROR) << "Operand " << i << " written a second time"; |
| return false; |
| } |
| writtenTo[i] = true; |
| } |
| } |
| for (size_t i = 0; i < operandCount; i++) { |
| if (!writtenTo[i]) { |
| const Operand& operand = operands[i]; |
| if (operand.lifetime == OperandLifeTime::TEMPORARY_VARIABLE || |
| operand.lifetime == OperandLifeTime::MODEL_OUTPUT) { |
| LOG(ERROR) << "Operand " << i << " with lifetime " << toString(operand.lifetime) |
| << " is not being written to."; |
| return false; |
| } |
| } |
| } |
| // TODO More whole graph verifications are possible, for example that an |
| // operand is not use as input & output for the same op, and more |
| // generally that it is acyclic. |
| return true; |
| } |
| |
| bool validatePool(const hidl_memory& pool, HalVersion ver) { |
| const auto& name = pool.name(); |
| if (name != "ashmem" && name != "mmap_fd" && |
| ((ver < HalVersion::V1_2) || |
| (name != "hardware_buffer_blob" && name != "hardware_buffer"))) { |
| LOG(ERROR) << "Unsupported memory type " << name; |
| return false; |
| } |
| if (pool.handle() == nullptr) { |
| LOG(ERROR) << "Memory of type " << name << " is null"; |
| return false; |
| } |
| return true; |
| } |
| |
| static bool validatePools(const hidl_vec<hidl_memory>& pools, HalVersion ver) { |
| return std::all_of(pools.begin(), pools.end(), |
| [ver](const hidl_memory& pool) { return validatePool(pool, ver); }); |
| } |
| |
| static bool validateModelInputOutputs(const hidl_vec<uint32_t> indexes, |
| const hidl_vec<Operand>& operands, OperandLifeTime lifetime) { |
| const size_t operandCount = operands.size(); |
| for (uint32_t i : indexes) { |
| if (i >= operandCount) { |
| LOG(ERROR) << "Model input or output index out of range: " << i << "/" << operandCount; |
| return false; |
| } |
| const Operand& operand = operands[i]; |
| if (operand.lifetime != lifetime) { |
| LOG(ERROR) << "Model input or output has lifetime of " << toString(operand.lifetime) |
| << " instead of the expected " << toString(lifetime); |
| return false; |
| } |
| } |
| |
| std::vector<uint32_t> sortedIndexes = indexes; |
| std::sort(sortedIndexes.begin(), sortedIndexes.end()); |
| auto adjacentI = std::adjacent_find(sortedIndexes.begin(), sortedIndexes.end()); |
| if (adjacentI != sortedIndexes.end()) { |
| LOG(ERROR) << "Model input or output occurs multiple times: " << *adjacentI; |
| return false; |
| } |
| return true; |
| } |
| |
| template <class T_Model> |
| bool validateModel(const T_Model& model) { |
| NNTRACE_FULL(NNTRACE_LAYER_UTILITY, NNTRACE_PHASE_UNSPECIFIED, "validateModel"); |
| HalVersion version = ModelToHalVersion<T_Model>::version; |
| if (model.operations.size() == 0 || model.operands.size() == 0) { |
| LOG(ERROR) << "Invalid empty model."; |
| return false; |
| } |
| // We only need versioned operands for their validation. For all the other |
| // validations we can use operands upcasted to the latest version. |
| const hidl_vec<Operand> latestVersionOperands = convertToV1_3(model.operands); |
| return (validateOperands(model.operands, model.operandValues, model.pools, |
| /*allowUnspecifiedRank=*/version >= HalVersion::V1_2) && |
| validateOperations(model.operations, latestVersionOperands) && |
| validateModelInputOutputs(model.inputIndexes, latestVersionOperands, |
| OperandLifeTime::MODEL_INPUT) && |
| validateModelInputOutputs(model.outputIndexes, latestVersionOperands, |
| OperandLifeTime::MODEL_OUTPUT) && |
| validatePools(model.pools, version)); |
| } |
| |
| template bool validateModel<V1_0::Model>(const V1_0::Model& model); |
| template bool validateModel<V1_1::Model>(const V1_1::Model& model); |
| template bool validateModel<V1_2::Model>(const V1_2::Model& model); |
| template bool validateModel<V1_3::Model>(const V1_3::Model& model); |
| |
| // Validates the arguments of a request. type is either "input" or "output" and is used |
| // for printing error messages. The operandIndexes is the appropriate array of input |
| // or output operand indexes that was passed to the ANeuralNetworksModel_identifyInputsAndOutputs. |
| static bool validateRequestArguments(const hidl_vec<RequestArgument>& requestArguments, |
| const hidl_vec<uint32_t>& operandIndexes, |
| const hidl_vec<Operand>& operands, |
| const hidl_vec<hidl_memory>& pools, bool allowUnspecified, |
| const char* type) { |
| MemoryAccessVerifier poolVerifier(pools); |
| // The request should specify as many arguments as were described in the model. |
| const size_t requestArgumentCount = requestArguments.size(); |
| if (requestArgumentCount != operandIndexes.size()) { |
| LOG(ERROR) << "Request specifies " << requestArgumentCount << " " << type |
| << "s but the model has " << operandIndexes.size(); |
| return false; |
| } |
| for (size_t requestArgumentIndex = 0; requestArgumentIndex < requestArgumentCount; |
| requestArgumentIndex++) { |
| const RequestArgument& requestArgument = requestArguments[requestArgumentIndex]; |
| const DataLocation& location = requestArgument.location; |
| // Get the operand index for this argument. We extract it from the list |
| // that was provided in the call to ANeuralNetworksModel_identifyInputsAndOutputs. |
| // We assume in this function that the model has been validated already. |
| const uint32_t operandIndex = operandIndexes[requestArgumentIndex]; |
| const Operand& operand = operands[operandIndex]; |
| if (requestArgument.hasNoValue) { |
| if (location.poolIndex != 0 || location.offset != 0 || location.length != 0 || |
| requestArgument.dimensions.size() != 0) { |
| LOG(ERROR) << "Request " << type << " " << requestArgumentIndex |
| << " has no value yet has details."; |
| return false; |
| } |
| } else { |
| // Validate the location. |
| if (!poolVerifier.validate(location)) { |
| return false; |
| } |
| // If the argument specified a dimension, validate it. |
| uint32_t rank = requestArgument.dimensions.size(); |
| if (rank == 0) { |
| if (!allowUnspecified) { |
| // Validate that all the dimensions are specified in the model. |
| for (size_t i = 0; i < operand.dimensions.size(); i++) { |
| if (operand.dimensions[i] == 0) { |
| LOG(ERROR) << "Model has dimension " << i |
| << " set to 0 but the request does specify the dimension."; |
| return false; |
| } |
| } |
| } |
| } else { |
| if (rank != operand.dimensions.size()) { |
| LOG(ERROR) << "Request " << type << " " << requestArgumentIndex |
| << " has number of dimensions (" << rank |
| << ") different than the model's (" << operand.dimensions.size() |
| << ")"; |
| return false; |
| } |
| for (size_t i = 0; i < rank; i++) { |
| if (requestArgument.dimensions[i] != operand.dimensions[i] && |
| operand.dimensions[i] != 0) { |
| LOG(ERROR) |
| << "Request " << type << " " << requestArgumentIndex |
| << " has dimension " << i << " of " << requestArgument.dimensions[i] |
| << " different than the model's " << operand.dimensions[i]; |
| return false; |
| } |
| if (requestArgument.dimensions[i] == 0 && !allowUnspecified) { |
| LOG(ERROR) << "Request " << type << " " << requestArgumentIndex |
| << " has dimension " << i << " of zero"; |
| return false; |
| } |
| } |
| } |
| } |
| } |
| return true; |
| } |
| |
| template <class T_Model> |
| bool validateRequest(const Request& request, const T_Model& model) { |
| HalVersion version = ModelToHalVersion<T_Model>::version; |
| return (validateRequestArguments(request.inputs, model.inputIndexes, |
| convertToV1_3(model.operands), request.pools, |
| /*allowUnspecified=*/false, "input") && |
| validateRequestArguments(request.outputs, model.outputIndexes, |
| convertToV1_3(model.operands), request.pools, |
| /*allowUnspecified=*/version >= HalVersion::V1_2, "output") && |
| validatePools(request.pools, version)); |
| } |
| |
| template bool validateRequest<V1_0::Model>(const Request& request, const V1_0::Model& model); |
| template bool validateRequest<V1_1::Model>(const Request& request, const V1_1::Model& model); |
| template bool validateRequest<V1_2::Model>(const Request& request, const V1_2::Model& model); |
| template bool validateRequest<V1_3::Model>(const Request& request, const V1_3::Model& model); |
| |
| bool validateExecutionPreference(ExecutionPreference preference) { |
| return preference == ExecutionPreference::LOW_POWER || |
| preference == ExecutionPreference::FAST_SINGLE_ANSWER || |
| preference == ExecutionPreference::SUSTAINED_SPEED; |
| } |
| |
| bool validOperandType(V1_0::OperandType operandType) { |
| switch (operandType) { |
| case V1_0::OperandType::FLOAT32: |
| case V1_0::OperandType::INT32: |
| case V1_0::OperandType::UINT32: |
| case V1_0::OperandType::TENSOR_FLOAT32: |
| case V1_0::OperandType::TENSOR_INT32: |
| case V1_0::OperandType::TENSOR_QUANT8_ASYMM: |
| case V1_0::OperandType::OEM: |
| case V1_0::OperandType::TENSOR_OEM_BYTE: |
| return true; |
| default: |
| return false; |
| } |
| } |
| |
| bool validOperandType(V1_2::OperandType operandType) { |
| switch (operandType) { |
| case V1_2::OperandType::FLOAT16: |
| case V1_2::OperandType::FLOAT32: |
| case V1_2::OperandType::INT32: |
| case V1_2::OperandType::UINT32: |
| case V1_2::OperandType::BOOL: |
| case V1_2::OperandType::TENSOR_FLOAT16: |
| case V1_2::OperandType::TENSOR_FLOAT32: |
| case V1_2::OperandType::TENSOR_INT32: |
| case V1_2::OperandType::TENSOR_QUANT8_ASYMM: |
| case V1_2::OperandType::TENSOR_QUANT8_SYMM: |
| case V1_2::OperandType::TENSOR_QUANT16_ASYMM: |
| case V1_2::OperandType::TENSOR_QUANT16_SYMM: |
| case V1_2::OperandType::TENSOR_BOOL8: |
| case V1_2::OperandType::TENSOR_QUANT8_SYMM_PER_CHANNEL: |
| case V1_2::OperandType::OEM: |
| case V1_2::OperandType::TENSOR_OEM_BYTE: |
| return true; |
| default: |
| return isExtensionOperandType(static_cast<V1_3::OperandType>(operandType)); |
| } |
| } |
| |
| bool validOperandType(V1_3::OperandType operandType) { |
| switch (operandType) { |
| case V1_3::OperandType::FLOAT16: |
| case V1_3::OperandType::FLOAT32: |
| case V1_3::OperandType::INT32: |
| case V1_3::OperandType::UINT32: |
| case V1_3::OperandType::BOOL: |
| case V1_3::OperandType::TENSOR_FLOAT16: |
| case V1_3::OperandType::TENSOR_FLOAT32: |
| case V1_3::OperandType::TENSOR_INT32: |
| case V1_3::OperandType::TENSOR_QUANT8_ASYMM: |
| case V1_3::OperandType::TENSOR_QUANT8_SYMM: |
| case V1_3::OperandType::TENSOR_QUANT16_ASYMM: |
| case V1_3::OperandType::TENSOR_QUANT16_SYMM: |
| case V1_3::OperandType::TENSOR_BOOL8: |
| case V1_3::OperandType::TENSOR_QUANT8_SYMM_PER_CHANNEL: |
| case V1_3::OperandType::TENSOR_QUANT8_ASYMM_SIGNED: |
| case V1_3::OperandType::OEM: |
| case V1_3::OperandType::TENSOR_OEM_BYTE: |
| return true; |
| default: |
| return isExtensionOperandType(operandType); |
| } |
| } |
| |
| } // namespace nn |
| } // namespace android |