diff options
author | Mike Kelly <mike.kelly@arm.com> | 2019-02-19 17:22:07 +0000 |
---|---|---|
committer | mike.kelly <mike.kelly@arm.com> | 2019-02-19 18:21:11 +0000 |
commit | a0766c3d461f6a635ef0c41c83c3ba868f2fd21a (patch) | |
tree | 3565ace677fd6de80dc30476c00face7e885d46a /src/armnnSerializer/Serializer.hpp | |
parent | 2b183fb359774cbac5d628579ec2b4a7b6b41def (diff) | |
download | armnn-a0766c3d461f6a635ef0c41c83c3ba868f2fd21a.tar.gz |
IVGCVSW-2646 Add Serializer & Deserializer for Conv2D
* Added Convolution2dLayer to Schema.fbs
* Added ConstTensorData serialization and deserialization helper functions
* Added Convolution2d serialization and deserialization support
* Added serialization and deserialization unit tests
Change-Id: Id376c08410ae01511972a2b0abdce9cfab907462
Signed-off-by: Mike Kelly <mike.kelly@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.hpp | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/src/armnnSerializer/Serializer.hpp b/src/armnnSerializer/Serializer.hpp index e4485f5856..fd1a792fb0 100644 --- a/src/armnnSerializer/Serializer.hpp +++ b/src/armnnSerializer/Serializer.hpp @@ -45,6 +45,12 @@ public: void VisitAdditionLayer(const armnn::IConnectableLayer* layer, const char* name = nullptr) override; + void VisitConvolution2dLayer(const armnn::IConnectableLayer* layer, + const armnn::Convolution2dDescriptor& descriptor, + const armnn::ConstTensor& weights, + const armnn::Optional<armnn::ConstTensor>& biases, + const char* = nullptr) override; + void VisitInputLayer(const armnn::IConnectableLayer* layer, armnn::LayerBindingId id, const char* name = nullptr) override; @@ -78,6 +84,13 @@ private: /// Creates the serializer AnyLayer for the layer and adds it to m_serializedLayers. void CreateAnyLayer(const flatbuffers::Offset<void>& layer, const armnn::armnnSerializer::Layer serializerLayer); + /// Creates the serializer ConstTensor for the armnn ConstTensor. + flatbuffers::Offset<armnn::armnnSerializer::ConstTensor> CreateConstTensorInfo( + const armnn::ConstTensor& constTensor); + + template <typename T> + flatbuffers::Offset<flatbuffers::Vector<T>> CreateDataVector(const void* memory, unsigned int size); + ///Function which maps Guid to an index uint32_t GetSerializedId(unsigned int guid); |