diff options
Diffstat (limited to 'src/backends/tosaCommon/operatorMappings/AdditionOperator.hpp')
-rw-r--r-- | src/backends/tosaCommon/operatorMappings/AdditionOperator.hpp | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/src/backends/tosaCommon/operatorMappings/AdditionOperator.hpp b/src/backends/tosaCommon/operatorMappings/AdditionOperator.hpp index f467bb6d10..5eb7441531 100644 --- a/src/backends/tosaCommon/operatorMappings/AdditionOperator.hpp +++ b/src/backends/tosaCommon/operatorMappings/AdditionOperator.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* ConvertAdditionToTosaOperator(const std::vector<const TensorInfo*>& inputs, - const std::vector<const TensorInfo*>& outputs, - bool isMain); +TosaSerializationBasicBlock* ConvertAdditionToTosaOperator(const Layer* layer, + const std::vector<const TensorInfo*>& inputs, + const std::vector<const TensorInfo*>& outputs); |