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/armnnDeserializeParser/DeserializeParser.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/armnnDeserializeParser/DeserializeParser.hpp')
-rw-r--r-- | src/armnnDeserializeParser/DeserializeParser.hpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/armnnDeserializeParser/DeserializeParser.hpp b/src/armnnDeserializeParser/DeserializeParser.hpp index aee647c636..c295ee8b5f 100644 --- a/src/armnnDeserializeParser/DeserializeParser.hpp +++ b/src/armnnDeserializeParser/DeserializeParser.hpp @@ -15,6 +15,7 @@ class DeserializeParser : public IDeserializeParser { public: // Shorthands for deserializer types + using ConstTensorRawPtr = const armnn::armnnSerializer::ConstTensor *; using GraphPtr = const armnn::armnnSerializer::SerializedGraph *; using TensorRawPtr = const armnn::armnnSerializer::TensorInfo *; using PoolingDescriptor = const armnn::armnnSerializer::Pooling2dDescriptor *; @@ -68,6 +69,7 @@ private: void ParseUnsupportedLayer(unsigned int layerIndex); void ParseAdd(unsigned int layerIndex); + void ParseConvolution2d(unsigned int layerIndex); void ParseMultiplication(unsigned int layerIndex); void ParsePooling2d(unsigned int layerIndex); void ParseReshape(unsigned int layerIndex); |