From a96479a46c87099d1a55a75233286e61a5ae02d5 Mon Sep 17 00:00:00 2001
From: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Date: Fri, 20 Sep 2019 16:38:01 +0100
Subject: IVGCVSW-3887 Add Quantizer support for DepthToSpace

Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: I144c04f920ba3329886367f5e73fe758505ab96e
---
 src/armnn/QuantizerVisitor.hpp | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'src/armnn/QuantizerVisitor.hpp')

diff --git a/src/armnn/QuantizerVisitor.hpp b/src/armnn/QuantizerVisitor.hpp
index db1a669ba8..b2acc7c56a 100644
--- a/src/armnn/QuantizerVisitor.hpp
+++ b/src/armnn/QuantizerVisitor.hpp
@@ -70,6 +70,11 @@ public:
                                  const Optional<ConstTensor>& biases,
                                  const char* name = nullptr) override;
 
+    void VisitDepthToSpaceLayer(const IConnectableLayer* layer,
+                                const DepthToSpaceDescriptor& depthToSpaceDescriptor,
+                                const char* name = nullptr) override;
+
+
     void VisitDepthwiseConvolution2dLayer(const IConnectableLayer* layer,
                                           const DepthwiseConvolution2dDescriptor& desc,
                                           const ConstTensor& weights,
-- 
cgit v1.2.1