diff options
| author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-10-17 16:11:54 +0100 |
|---|---|---|
| committer | Áron Virginás-Tar <aron.virginas-tar@arm.com> | 2019-10-21 10:54:45 +0000 |
| commit | e80ebd101b516751a798aa1b1d669e9117a32266 (patch) | |
| tree | 9f113060aaa2157941f726af6ec93ea8e9357df8 /src/armnnDeserializer/Deserializer.hpp | |
| parent | 77bfb5e32faadb1383d48364a6f54adbff84ad80 (diff) | |
| download | armnn-e80ebd101b516751a798aa1b1d669e9117a32266.tar.gz | |
IVGCVSW-3992 Add serialization support for ComparisonLayer
* Added serialization support and serialization-deserialization test
for Comparison
* Added backward compatibility tests for Equal and Greater, to make
sure they are serialized and deserialized as Comparison
* Refactored serialization tests and reduced code duplication by
taking advantage of operator==() recently added for Descriptors
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: Id0fd94ef4b17e4e51c8005e585ea3c47f9c1bd8d
Diffstat (limited to 'src/armnnDeserializer/Deserializer.hpp')
| -rw-r--r-- | src/armnnDeserializer/Deserializer.hpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/armnnDeserializer/Deserializer.hpp b/src/armnnDeserializer/Deserializer.hpp index dda588d5e3..b951483926 100644 --- a/src/armnnDeserializer/Deserializer.hpp +++ b/src/armnnDeserializer/Deserializer.hpp @@ -83,6 +83,7 @@ private: void ParseArgMinMax(GraphPtr graph, unsigned int layerIndex); void ParseBatchToSpaceNd(GraphPtr graph, unsigned int layerIndex); void ParseBatchNormalization(GraphPtr graph, unsigned int layerIndex); + void ParseComparison(GraphPtr graph, unsigned int layerIndex); void ParseConcat(GraphPtr graph, unsigned int layerIndex); void ParseConstant(GraphPtr graph, unsigned int layerIndex); void ParseConvolution2d(GraphPtr graph, unsigned int layerIndex); @@ -166,4 +167,4 @@ private: std::unordered_map<unsigned int, Connections> m_GraphConnections; }; -} //namespace armnnDeserializer +} // namespace armnnDeserializer |