aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/Serializer.hpp
diff options
context:
space:
mode:
authorSamuel Yap <samuel.yap@arm.com>2022-08-19 11:14:38 +0100
committerSam Yap <samuel.yap@arm.com>2022-08-30 09:52:48 +0000
commitb9e6b5c3b96792f40201315c831db0aa257f286c (patch)
tree5f9ec80ee2a2f941c475115a274a2ba18e3965ce /src/armnnSerializer/Serializer.hpp
parent75d2cb12de43ad308161ed49a8d87762c0ff873e (diff)
downloadarmnn-b9e6b5c3b96792f40201315c831db0aa257f286c.tar.gz
IVGCVSW-7104: BatchMatMul Serializer/Deserializer Support
* Updated FlatBuffers schema for BatchMatMul layer type * Added Serializer and Deserializer implementations for BatchMatMul * Added unit tests for BatchMatMul serialization and deserialization * Updated CMakeLists and docs Signed-off-by: Samuel Yap <samuel.yap@arm.com> Change-Id: Iad63afbd036a3eb648683eb7416a475561aa20cb
Diffstat (limited to 'src/armnnSerializer/Serializer.hpp')
-rw-r--r--src/armnnSerializer/Serializer.hpp4
1 files changed, 4 insertions, 0 deletions
diff --git a/src/armnnSerializer/Serializer.hpp b/src/armnnSerializer/Serializer.hpp
index 216f4dc016..60fed4f6df 100644
--- a/src/armnnSerializer/Serializer.hpp
+++ b/src/armnnSerializer/Serializer.hpp
@@ -113,6 +113,10 @@ private:
const armnn::ArgMinMaxDescriptor& argMinMaxDescriptor,
const char* name = nullptr);
+ void SerializeBatchMatMulLayer(const armnn::IConnectableLayer* layer,
+ const armnn::BatchMatMulDescriptor& descriptor,
+ const char* name = nullptr);
+
void SerializeBatchToSpaceNdLayer(const armnn::IConnectableLayer* layer,
const armnn::BatchToSpaceNdDescriptor& descriptor,
const char* name = nullptr);