aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/WorkloadFactory.hpp
diff options
context:
space:
mode:
authorAron Virginas-Tar <Aron.Virginas-Tar@arm.com>2019-06-20 14:28:19 +0100
committerMatteo Martincigh <matteo.martincigh@arm.com>2019-06-21 09:54:30 +0000
commit639fb0437d1a5a8a6ea737fed5a16b554dfffead (patch)
tree5b89adc18c1a071d23747a28dcddcfca41e4d815 /src/backends/backendsCommon/WorkloadFactory.hpp
parent713e95c8c531c5cecd804a7cecc8af745917315c (diff)
downloadarmnn-639fb0437d1a5a8a6ea737fed5a16b554dfffead.tar.gz
IVGCVSW-3319 Add frontend support for TransposeConvolution2d Layer
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> Change-Id: Ic06f63f1eff255e697facf319e2ac4c83d782e7c
Diffstat (limited to 'src/backends/backendsCommon/WorkloadFactory.hpp')
-rw-r--r--src/backends/backendsCommon/WorkloadFactory.hpp6
1 files changed, 5 insertions, 1 deletions
diff --git a/src/backends/backendsCommon/WorkloadFactory.hpp b/src/backends/backendsCommon/WorkloadFactory.hpp
index c9fbe71f96..978d3a3a98 100644
--- a/src/backends/backendsCommon/WorkloadFactory.hpp
+++ b/src/backends/backendsCommon/WorkloadFactory.hpp
@@ -190,6 +190,10 @@ public:
virtual std::unique_ptr<IWorkload> CreateSwitch(const SwitchQueueDescriptor& descriptor,
const WorkloadInfo& Info) const;
+
+ virtual std::unique_ptr<IWorkload> CreateTransposeConvolution2d(
+ const TransposeConvolution2dQueueDescriptor& descriptor,
+ const WorkloadInfo& info) const;
};
-} //namespace armnn
+} // namespace armnn