diff options
Diffstat (limited to 'nn/common/operations/RoiAlign.cpp')
-rw-r--r-- | nn/common/operations/RoiAlign.cpp | 15 |
1 files changed, 8 insertions, 7 deletions
diff --git a/nn/common/operations/RoiAlign.cpp b/nn/common/operations/RoiAlign.cpp index 01008cc9d..78049b8bb 100644 --- a/nn/common/operations/RoiAlign.cpp +++ b/nn/common/operations/RoiAlign.cpp @@ -16,17 +16,18 @@ #define LOG_TAG "Operations" -#include "CpuOperationUtils.h" -#include "OperationResolver.h" -#include "OperationsUtils.h" -#include "Tracing.h" - #include <tensorflow/lite/kernels/internal/common.h> + #include <algorithm> #include <cfloat> #include <cmath> #include <vector> +#include "CpuOperationUtils.h" +#include "OperationResolver.h" +#include "OperationsUtils.h" +#include "Tracing.h" + namespace android { namespace nn { namespace roi_align { @@ -372,9 +373,9 @@ bool validate(const IOperationValidationContext* context) { NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); NN_RET_CHECK(validateOutputTypes(context, {inputType})); if (inputType == OperandType::TENSOR_QUANT8_ASYMM_SIGNED) { - return validateHalVersion(context, HalVersion::V1_3); + return validateVersion(context, Version::ANDROID_R); } else { - return validateHalVersion(context, HalVersion::V1_2); + return validateVersion(context, Version::ANDROID_Q); } } |