diff options
| author | Tamas Nyiri <tamas.nyiri@arm.com> | 2021-11-05 14:55:33 +0000 |
|---|---|---|
| committer | Tamas Nyiri <tamas.nyiri@arm.com> | 2021-11-17 11:32:55 +0000 |
| commit | d998a1cfbcc7b4ed9922532ffe19c51283fcae68 (patch) | |
| tree | 53317946bb5845f4f253ebfd53b668c6698fcfee /src/armnnSerializer/Serializer.hpp | |
| parent | 7b885b3cce70154596b1994b013ea91527117c26 (diff) | |
| download | armnn-d998a1cfbcc7b4ed9922532ffe19c51283fcae68.tar.gz | |
IVGCVSW-6510 Serialization + Deserialization implementation
Subtask of story: IVGCVSW-6164 Add a Pooling3d FrontEnd and Ref Implementation
* Add serialization support
* Add deserialization support
* Add corresponding unit tests
Change-Id: I7cce5421f0a9b7c47a03524e733f3315131ba125
Signed-off-by: Tamas Nyiri <tamas.nyiri@arm.com>
Diffstat (limited to 'src/armnnSerializer/Serializer.hpp')
| -rw-r--r-- | src/armnnSerializer/Serializer.hpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/armnnSerializer/Serializer.hpp b/src/armnnSerializer/Serializer.hpp index 1c0a9a619f..bb6455f842 100644 --- a/src/armnnSerializer/Serializer.hpp +++ b/src/armnnSerializer/Serializer.hpp @@ -248,6 +248,10 @@ private: const armnn::Pooling2dDescriptor& pooling2dDescriptor, const char* name = nullptr); + void SerializePooling3dLayer(const armnn::IConnectableLayer* layer, + const armnn::Pooling3dDescriptor& pooling3dDescriptor, + const char* name = nullptr); + void SerializePreluLayer(const armnn::IConnectableLayer* layer, const char* name = nullptr); |