aboutsummaryrefslogtreecommitdiff
path: root/src/backends/tosaCommon/operatorMappings/TransposeConv2dOperator.hpp
diff options
context:
space:
mode:
authorMatthew Sloyan <matthew.sloyan@arm.com>2022-12-08 13:38:23 +0000
committerCathal Corbett <cathal.corbett@arm.com>2022-12-15 10:00:12 +0000
commitfc9d5e7d1e0c1a4d7fed4ebc363832e03c3e2543 (patch)
tree5c1b7efd146256c2de9781ae5a60e22cf2d6142a /src/backends/tosaCommon/operatorMappings/TransposeConv2dOperator.hpp
parent67fd526f8f7c79803d514a6045454049104eced9 (diff)
downloadarmnn-fc9d5e7d1e0c1a4d7fed4ebc363832e03c3e2543.tar.gz
IVGCVSW-7204 Add TransposeConv2d support to TOSA Reference Backend
Signed-off-by: Matthew Sloyan <matthew.sloyan@arm.com> Change-Id: I9bfd597afd41468f304edfbe5d7141378ce60d4f
Diffstat (limited to 'src/backends/tosaCommon/operatorMappings/TransposeConv2dOperator.hpp')
-rw-r--r--src/backends/tosaCommon/operatorMappings/TransposeConv2dOperator.hpp20
1 files changed, 20 insertions, 0 deletions
diff --git a/src/backends/tosaCommon/operatorMappings/TransposeConv2dOperator.hpp b/src/backends/tosaCommon/operatorMappings/TransposeConv2dOperator.hpp
new file mode 100644
index 0000000000..eb911a1195
--- /dev/null
+++ b/src/backends/tosaCommon/operatorMappings/TransposeConv2dOperator.hpp
@@ -0,0 +1,20 @@
+//
+// Copyright © 2022 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* ConvertTransposeConv2dToTosaOperator(const Layer* layer,
+ const std::vector<const TensorInfo*>& inputs,
+ const std::vector<const TensorInfo*>& outputs,
+ const TransposeConvolution2dDescriptor* descriptor);