diff options
Diffstat (limited to 'src/backends/tosaCommon/operatorMappings/Pooling2DOperator.hpp')
-rw-r--r-- | src/backends/tosaCommon/operatorMappings/Pooling2DOperator.hpp | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/src/backends/tosaCommon/operatorMappings/Pooling2DOperator.hpp b/src/backends/tosaCommon/operatorMappings/Pooling2DOperator.hpp index 22d2a3ae29..cc9ec097f9 100644 --- a/src/backends/tosaCommon/operatorMappings/Pooling2DOperator.hpp +++ b/src/backends/tosaCommon/operatorMappings/Pooling2DOperator.hpp @@ -5,15 +5,16 @@ #pragma once +#include "TosaOperatorUtils.hpp" + #include <Layer.hpp> #include <tosa_serialization_handler.h> -#include "TosaOperatorUtils.hpp" using namespace armnn; using namespace tosa; -TosaSerializationBasicBlock* ConvertPooling2DToTosaOperator(const std::vector<const TensorInfo*>& inputs, +TosaSerializationBasicBlock* ConvertPooling2DToTosaOperator(const Layer* layer, + const std::vector<const TensorInfo*>& inputs, const std::vector<const TensorInfo*>& outputs, - bool isMain, const Pooling2dDescriptor* poolDescriptor); |