diff options
Diffstat (limited to 'nn/common/operations/PRelu.cpp')
-rw-r--r-- | nn/common/operations/PRelu.cpp | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/nn/common/operations/PRelu.cpp b/nn/common/operations/PRelu.cpp index 7e0c8c371..db2f6d416 100644 --- a/nn/common/operations/PRelu.cpp +++ b/nn/common/operations/PRelu.cpp @@ -16,6 +16,8 @@ #define LOG_TAG "Operations" +#include <tensorflow/lite/kernels/internal/optimized/legacy_optimized_ops.h> + #include <algorithm> #include <vector> @@ -24,8 +26,6 @@ #include "OperationsUtils.h" #include "Tracing.h" -#include <tensorflow/lite/kernels/internal/optimized/legacy_optimized_ops.h> - namespace android { namespace nn { namespace prelu { @@ -107,9 +107,9 @@ bool validate(const IOperationValidationContext* context) { NN_RET_CHECK(validateInputTypes(context, {inputType, inputType})); 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); } } |