summaryrefslogtreecommitdiff
path: root/nn/common/operations/RoiAlign.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'nn/common/operations/RoiAlign.cpp')
-rw-r--r--nn/common/operations/RoiAlign.cpp15
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);
}
}