diff options
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; |