IVGCVSW-4375 Add support for Transpose
* Added TransposeLayer
* Added CL, Neon and Ref Workloads
* Added Transpose utilities
* Added Serializer and Deserializer support
* Added Quantizer support
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: I04c755ba7cb5b1edf72b3c9f3c0314878032e3c7
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp
index 7edc624..b405a77 100644
--- a/src/armnn/Network.cpp
+++ b/src/armnn/Network.cpp
@@ -1526,6 +1526,12 @@
return layer;
}
+IConnectableLayer* Network::AddTransposeLayer(const TransposeDescriptor& transposeDescriptor,
+ const char* name)
+{
+ return m_Graph->AddLayer<TransposeLayer>(transposeDescriptor, name);
+}
+
IConnectableLayer* Network::AddStackLayer(const StackDescriptor& stackDescriptor,
const char* name)
{