commit 8081536d24291794b4e189e6d5532d913a4525cb
author Narumol Prangnawarat <narumol.prangnawarat@arm.com> Wed Nov 11 11:33:03 2020 +0000
committer Narumol Prangnawarat <narumol.prangnawarat@arm.com> Wed Nov 11 19:20:08 2020 +0000
tree 7b30bfc8f3f619377f0e1fa4f0f81943e62386ba
parent 3dda41daba5210ebd842115471ecadeb2fb3ae3b

IVGCVSW-5387 TfLiteDelegate: Implement the Pooling operators

 * Add support for AveragePool2d and L2Pool2d operators
 * Unit tests

Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Ida3c2e80120bce2991035f143e9eb5b9480b0e4b

delegate/src/Pooling.hpp

diff --git a/delegate/src/Pooling.hpp b/delegate/src/Pooling.hpp
index 28e26f6..b3a2af8 100644
--- a/delegate/src/Pooling.hpp
+++ b/delegate/src/Pooling.hpp
@@ -72,6 +72,12 @@
     armnn::PoolingAlgorithm poolingAlgorithm;
     switch(tfLitePoolingOperatorCode)
     {
+        case kTfLiteBuiltinAveragePool2d:
+            poolingAlgorithm = armnn::PoolingAlgorithm::Average;
+            break;
+        case kTfLiteBuiltinL2Pool2d:
+            poolingAlgorithm = armnn::PoolingAlgorithm::L2;
+            break;
         case kTfLiteBuiltinMaxPool2d:
             poolingAlgorithm = armnn::PoolingAlgorithm::Max;
             break;