aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/Serializer.hpp
diff options
context:
space:
mode:
authorSadik Armagan <sadik.armagan@arm.com>2019-02-21 09:01:41 +0000
committerSadik Armagan <sadik.armagan@arm.com>2019-02-21 09:57:44 +0000
commitdbb0c0ca0c8425886ee3a2095e0ced07099134f9 (patch)
tree2d85dd36b0c23c2fda9f6898a7ab8a9f74f4520c /src/armnnSerializer/Serializer.hpp
parent5324782821dce525cf7c1636c659f998fae2fb85 (diff)
downloadarmnn-dbb0c0ca0c8425886ee3a2095e0ced07099134f9.tar.gz
IVGCVSW-2639 Add Serializer & Deserializer for Fully Connected
* Added FullyConnectedLayer to Serializer Schema Schema.fbs * Added FullyConnected serialization and deserialization support * Added FullyConnected serialization and deserialization unit tests Change-Id: I8ef14f9728158f849fa4d1a8d05a1a4170cd5b41 Signed-off-by: Sadik Armagan <sadik.armagan@arm.com> Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Diffstat (limited to 'src/armnnSerializer/Serializer.hpp')
-rw-r--r--src/armnnSerializer/Serializer.hpp6
1 files changed, 6 insertions, 0 deletions
diff --git a/src/armnnSerializer/Serializer.hpp b/src/armnnSerializer/Serializer.hpp
index 329b005624..0a62732ef2 100644
--- a/src/armnnSerializer/Serializer.hpp
+++ b/src/armnnSerializer/Serializer.hpp
@@ -61,6 +61,12 @@ public:
const armnn::Optional<armnn::ConstTensor>& biases,
const char* name = nullptr) override;
+ void VisitFullyConnectedLayer(const armnn::IConnectableLayer* layer,
+ const armnn::FullyConnectedDescriptor& fullyConnectedDescriptor,
+ const armnn::ConstTensor& weights,
+ const armnn::Optional<armnn::ConstTensor>& biases,
+ const char* name = nullptr) override;
+
void VisitInputLayer(const armnn::IConnectableLayer* layer,
armnn::LayerBindingId id,
const char* name = nullptr) override;