IVGCVSW-3805 Add Comparison support to the android-nn-driver
* Added support for the following HAL1.2 operations: EQUAL,
GREATER, GREATER_EQUAL, LESS, LESS_EQUAL and NOT_EQUAL
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: I71b68db70232da4aaad28caa7b0b5f9a1d7778d0
diff --git a/1.2/HalPolicy.hpp b/1.2/HalPolicy.hpp
index 743ac11..d611102 100644
--- a/1.2/HalPolicy.hpp
+++ b/1.2/HalPolicy.hpp
@@ -39,6 +39,11 @@
static bool ConvertBatchToSpaceNd(const Operation& operation, const Model& model, ConversionData& data);
+ static bool ConvertComparison(const Operation& operation,
+ const Model& model,
+ ConversionData& data,
+ armnn::ComparisonOperation comparisonOperation);
+
static bool ConvertConcatenation(const Operation& operation, const Model& model, ConversionData& data);
static bool ConvertConv2d(const Operation& operation, const Model& model, ConversionData& data);