aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/Serializer.hpp
diff options
context:
space:
mode:
authorNattapat Chaimanowong <nattapat.chaimanowong@arm.com>2019-03-18 12:37:06 +0000
committerNattapat Chaimanowong <nattapat.chaimanowong@arm.com>2019-03-18 12:37:06 +0000
commit3e14a9d2033530df49546ab0da63ad4b6470f551 (patch)
tree030433752c31b42eeecc1ed094c39c2c508dd57f /src/armnnSerializer/Serializer.hpp
parent65d30965eef7e8534fc16ea4ded413c42a81c362 (diff)
downloadarmnn-3e14a9d2033530df49546ab0da63ad4b6470f551.tar.gz
IVGCVSW-2686 Add Serializer and Deserializer for DetectionPostProcess
Change-Id: Ife48db5fdb005ebca0a6f21862b0ce971ccf58b7 Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Diffstat (limited to 'src/armnnSerializer/Serializer.hpp')
-rw-r--r--src/armnnSerializer/Serializer.hpp5
1 files changed, 5 insertions, 0 deletions
diff --git a/src/armnnSerializer/Serializer.hpp b/src/armnnSerializer/Serializer.hpp
index cb05792863..71066d2699 100644
--- a/src/armnnSerializer/Serializer.hpp
+++ b/src/armnnSerializer/Serializer.hpp
@@ -77,6 +77,11 @@ public:
const armnn::Optional<armnn::ConstTensor>& biases,
const char* name = nullptr) override;
+ void VisitDetectionPostProcessLayer(const armnn::IConnectableLayer* layer,
+ const armnn::DetectionPostProcessDescriptor& descriptor,
+ const armnn::ConstTensor& anchors,
+ const char* name = nullptr) override;
+
void VisitDivisionLayer(const armnn::IConnectableLayer* layer,
const char* name = nullptr) override;