aboutsummaryrefslogtreecommitdiff
path: root/src/backends/tosaCommon/operatorMappings/ReluOperator.hpp
diff options
context:
space:
mode:
authorCathal Corbett <cathal.corbett@arm.com>2024-06-27 14:53:50 +0100
committerCathal Corbett <cathal.corbett@arm.com>2024-06-28 12:28:22 +0100
commit66d0e7d5c6d3c271f72848a180616ebc2e3403cc (patch)
treedab3b4433d4e2145583b21084df415fcfd277340 /src/backends/tosaCommon/operatorMappings/ReluOperator.hpp
parent7177b72cdd36c0a5b0b52a5b3034b1c2b0f8a3ec (diff)
downloadarmnn-66d0e7d5c6d3c271f72848a180616ebc2e3403cc.tar.gz
IVGCVSW-8443 Implement Activation:ReLu in TosaRef
Signed-off-by: Cathal Corbett <cathal.corbett@arm.com> Change-Id: I2a1f7be2b0b4558b0e0e8dbd2350ccf21b5babf8
Diffstat (limited to 'src/backends/tosaCommon/operatorMappings/ReluOperator.hpp')
-rw-r--r--src/backends/tosaCommon/operatorMappings/ReluOperator.hpp20
1 files changed, 20 insertions, 0 deletions
diff --git a/src/backends/tosaCommon/operatorMappings/ReluOperator.hpp b/src/backends/tosaCommon/operatorMappings/ReluOperator.hpp
new file mode 100644
index 0000000000..9c8f999949
--- /dev/null
+++ b/src/backends/tosaCommon/operatorMappings/ReluOperator.hpp
@@ -0,0 +1,20 @@
+//
+// Copyright © 2024 Arm Ltd and Contributors. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#pragma once
+
+#include <Layer.hpp>
+
+#include <tosa_serialization_handler.h>
+
+#include "TosaOperatorUtils.hpp"
+
+using namespace armnn;
+using namespace tosa;
+
+TosaSerializationBasicBlock* ConvertReluToTosaOperator(const Layer* layer,
+ const std::vector<const TensorInfo*>& inputs,
+ const std::vector<const TensorInfo*>& outputs,
+ const ActivationDescriptor* activationDescriptor);