diff options
| author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-11-11 11:33:03 +0000 |
|---|---|---|
| committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-11-11 19:20:08 +0000 |
| commit | 8081536d24291794b4e189e6d5532d913a4525cb (patch) | |
| tree | 7b30bfc8f3f619377f0e1fa4f0f81943e62386ba /delegate/src/Pooling.hpp | |
| parent | 3dda41daba5210ebd842115471ecadeb2fb3ae3b (diff) | |
| download | armnn-8081536d24291794b4e189e6d5532d913a4525cb.tar.gz | |
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
Diffstat (limited to 'delegate/src/Pooling.hpp')
| -rw-r--r-- | delegate/src/Pooling.hpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/delegate/src/Pooling.hpp b/delegate/src/Pooling.hpp index 28e26f6504..b3a2af8900 100644 --- a/delegate/src/Pooling.hpp +++ b/delegate/src/Pooling.hpp @@ -72,6 +72,12 @@ TfLiteStatus VisitPoolingOperator(DelegateData& delegateData, 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; |