aboutsummaryrefslogtreecommitdiff
path: root/src/backends/tosaCommon/operatorMappings/ElementwiseUnaryOperator.hpp
diff options
context:
space:
mode:
authorDavid Monahan <david.monahan@arm.com>2023-01-12 14:53:34 +0000
committerDavid Monahan <david.monahan@arm.com>2023-01-13 11:28:52 +0000
commitd7fca093be53b314e100e15a8fa080cb506b60a3 (patch)
treefaf4adbdd3154ffa6101bcbbf8c4839179ad180b /src/backends/tosaCommon/operatorMappings/ElementwiseUnaryOperator.hpp
parentc17a35f4b016c223add511a148ed3bb741770593 (diff)
downloadarmnn-d7fca093be53b314e100e15a8fa080cb506b60a3.tar.gz
IVGCVSW-7173 Add Rsqrt to Tosa Ref Backend
* Added ElementwiseUnary support with a mapping for Rsqrt * Added unittests * Added Rsqrt EndtoEnd tests for all backends * Changed TosaRefLayerSupport to default to false on unsupported layers Signed-off-by: David Monahan <david.monahan@arm.com> Change-Id: I3eaa9c684647ead61520a563815581aa68bee51b
Diffstat (limited to 'src/backends/tosaCommon/operatorMappings/ElementwiseUnaryOperator.hpp')
-rw-r--r--src/backends/tosaCommon/operatorMappings/ElementwiseUnaryOperator.hpp19
1 files changed, 19 insertions, 0 deletions
diff --git a/src/backends/tosaCommon/operatorMappings/ElementwiseUnaryOperator.hpp b/src/backends/tosaCommon/operatorMappings/ElementwiseUnaryOperator.hpp
new file mode 100644
index 0000000000..d13428ca5a
--- /dev/null
+++ b/src/backends/tosaCommon/operatorMappings/ElementwiseUnaryOperator.hpp
@@ -0,0 +1,19 @@
+//
+// Copyright © 2023 Arm Ltd and Contributors. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#pragma once
+
+#include "TosaOperatorUtils.hpp"
+#include <Layer.hpp>
+
+#include <tosa_serialization_handler.h>
+
+using namespace armnn;
+using namespace tosa;
+
+TosaSerializationBasicBlock* ConvertElementwiseUnaryOperator(const Layer* layer,
+ const std::vector<const TensorInfo*>& inputs,
+ const std::vector<const TensorInfo*>& outputs,
+ const ElementwiseUnaryDescriptor* unaryDescriptor);