| commit | f8e358463b4b612eb958fa2dd7588dbefe9b3e2a | [log] [tgz] |
|---|---|---|
| author | thecha01 <theo.charalambous@arm.com> | Tue Jul 28 17:28:17 2020 +0100 |
| committer | Michele Di Giorgio <michele.digiorgio@arm.com> | Mon Aug 24 09:32:49 2020 +0000 |
| tree | e8754177347fe911799cb6c9815d3e7c4a85f2c8 | |
| parent | ebd8fb479aa978c428c54c12be6457a0e635c363 [diff] |
Add support ElementwiseMax operator in graph API Signed-off-by: thecha01 <theo.charalambous@arm.com> Change-Id: I764f1eabb6412350eb719cc755b8777efc7d70a1 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3736 Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
diff --git a/src/graph/backends/NEON/NEFunctionFactory.cpp b/src/graph/backends/NEON/NEFunctionFactory.cpp index 4fee630..1a1519b 100644 --- a/src/graph/backends/NEON/NEFunctionFactory.cpp +++ b/src/graph/backends/NEON/NEFunctionFactory.cpp
@@ -69,6 +69,7 @@ using Addition = NEArithmeticAddition; using Subtraction = NEArithmeticSubtraction; using Multiplication = NEPixelWiseMultiplication; + using Maximum = NEElementwiseMax; }; /** Collection of NEON unary element-wise functions */
diff --git a/src/graph/backends/NEON/NENodeValidator.cpp b/src/graph/backends/NEON/NENodeValidator.cpp index a5d22fb..74c959a 100644 --- a/src/graph/backends/NEON/NENodeValidator.cpp +++ b/src/graph/backends/NEON/NENodeValidator.cpp
@@ -44,6 +44,7 @@ using ArithmeticAddition = NEArithmeticAddition; using ArithmeticSubtraction = NEArithmeticSubtraction; using PixelWiseMultiplication = NEPixelWiseMultiplication; + using ElementwiseMax = NEElementwiseMax; }; /** Collection of NEON unary element-wise functions */